mirror of
https://github.com/logos-storage/logos-storage-nim.git
synced 2026-03-03 19:03:42 +00:00
03aa6c5890
Part of https://github.com/logos-storage/logos-storage-nim/issues/1366 Signed-off-by: Chrysostomos Nanakos <chris@include.gr>
118 lines
3.0 KiB
Nim
118 lines
3.0 KiB
Nim
## Logos Storage
|
|
## Copyright (c) 2021 Status Research & Development GmbH
|
|
## Licensed under either of
|
|
## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
## * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
## at your option.
|
|
## This file may not be copied, modified, or distributed except according to
|
|
## those terms.
|
|
|
|
{.push raises: [].}
|
|
|
|
import std/options
|
|
import std/sugar
|
|
import std/sequtils
|
|
|
|
import pkg/results
|
|
import pkg/chronos
|
|
import pkg/questionable/results
|
|
|
|
export results
|
|
|
|
type
|
|
StorageError* = object of CatchableError # base Storage error
|
|
StorageResult*[T] = Result[T, ref StorageError]
|
|
|
|
WantBlocksErrorKind* = enum
|
|
RequestTooShort
|
|
RequestTooLarge
|
|
RequestTruncated
|
|
InvalidCid
|
|
InvalidCodec
|
|
MetadataTooShort
|
|
MetadataTruncated
|
|
ResponseTooLarge
|
|
MetadataTooLarge
|
|
DataSizeMismatch
|
|
ProofTooShort
|
|
ProofTruncated
|
|
ProofCreationFailed
|
|
ProofPathTooLarge
|
|
ProofDecodeFailed
|
|
TooManyBlocks
|
|
NoConnection
|
|
ConnectionClosed
|
|
RequestFailed
|
|
|
|
WantBlocksError* = object of StorageError
|
|
kind*: WantBlocksErrorKind
|
|
|
|
WantBlocksResult*[T] = Result[T, ref WantBlocksError]
|
|
|
|
FinishedFailed*[T] = tuple[success: seq[Future[T]], failure: seq[Future[T]]]
|
|
|
|
proc wantBlocksError*(kind: WantBlocksErrorKind, msg: string): ref WantBlocksError =
|
|
(ref WantBlocksError)(kind: kind, msg: msg)
|
|
|
|
template mapFailure*[T, V, E](
|
|
exp: Result[T, V], exc: typedesc[E]
|
|
): Result[T, ref CatchableError] =
|
|
## Convert `Result[T, E]` to `Result[E, ref CatchableError]`
|
|
##
|
|
|
|
exp.mapErr(
|
|
proc(e: V): ref CatchableError =
|
|
(ref exc)(msg: $e)
|
|
)
|
|
|
|
template mapFailure*[T, V](exp: Result[T, V]): Result[T, ref CatchableError] =
|
|
mapFailure(exp, StorageError)
|
|
|
|
# TODO: using a template here, causes bad codegen
|
|
func toFailure*[T](exp: Option[T]): Result[T, ref CatchableError] {.inline.} =
|
|
if exp.isSome:
|
|
success exp.get
|
|
else:
|
|
T.failure("Option is None")
|
|
|
|
proc allFinishedFailed*[T](
|
|
futs: auto
|
|
): Future[FinishedFailed[T]] {.async: (raises: [CancelledError]).} =
|
|
## Check if all futures have finished or failed
|
|
##
|
|
## TODO: wip, not sure if we want this - at the minimum,
|
|
## we should probably avoid the async transform
|
|
|
|
var res: FinishedFailed[T] = (@[], @[])
|
|
await allFutures(futs)
|
|
for f in futs:
|
|
if f.failed:
|
|
res.failure.add f
|
|
else:
|
|
res.success.add f
|
|
|
|
return res
|
|
|
|
proc allFinishedValues*[T](
|
|
futs: auto
|
|
): Future[?!seq[T]] {.async: (raises: [CancelledError]).} =
|
|
## If all futures have finished, return corresponding values,
|
|
## otherwise return failure
|
|
##
|
|
|
|
# wait for all futures to be either completed, failed or canceled
|
|
await allFutures(futs)
|
|
|
|
let numOfFailed = futs.countIt(it.failed)
|
|
|
|
if numOfFailed > 0:
|
|
return failure "Some futures failed (" & $numOfFailed & "))"
|
|
|
|
# here, we know there are no failed futures in "futs"
|
|
# and we are only interested in those that completed successfully
|
|
let values = collect:
|
|
for b in futs:
|
|
if b.finished:
|
|
b.value
|
|
return success values
|