mirror of
https://github.com/logos-storage/logos-storage-nim.git
synced 2026-01-03 14:03:10 +00:00
1cac3e2a11
* cleanup imports and logs * add BlockHandle type * revert deps * refactor: async error handling and future tracking improvements - Update async procedures to use explicit raises annotation - Modify TrackedFutures to handle futures with no raised exceptions - Replace `asyncSpawn` with explicit future tracking - Update test suites to use `unittest2` - Standardize error handling across network and async components - Remove deprecated error handling patterns This commit introduces a more robust approach to async error handling and future management, improving type safety and reducing potential runtime errors. * bump nim-serde * remove asyncSpawn * rework background downloads and prefetch * imporove logging * refactor: enhance async procedures with error handling and raise annotations * misc cleanup * misc * refactor: implement allFinishedFailed to aggregate future results with success and failure tracking * refactor: update error handling in reader procedures to raise ChunkerError and CancelledError * refactor: improve error handling in wantListHandler and accountHandler procedures * refactor: simplify LPStreamReadError creation by consolidating parameters * refactor: enhance error handling in AsyncStreamWrapper to catch unexpected errors * refactor: enhance error handling in advertiser and discovery loops to improve resilience * misc * refactor: improve code structure and readability * remove cancellation from addSlotToQueue * refactor: add assertion for unexpected errors in local store checks * refactor: prevent tracking of finished futures and improve test assertions * refactor: improve error handling in local store checks * remove usage of msgDetail * feat: add initial implementation of discovery engine and related components * refactor: improve task scheduling logic by removing unnecessary break statement * break after scheduling a task * make taskHandler cancelable * refactor: update async handlers to raise CancelledError * refactor(advertiser): streamline error handling and improve task flow in advertise loops * fix: correct spelling of "divisible" in error messages and comments * refactor(discovery): simplify discovery task loop and improve error handling * refactor(engine): filter peers before processing in cancelBlocks procedure
73 lines
2.1 KiB
Nim
73 lines
2.1 KiB
Nim
## Nim-Codex
|
|
## Copyright (c) 2022 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.
|
|
|
|
import pkg/chronos
|
|
import pkg/libp2p
|
|
import pkg/questionable
|
|
import pkg/codex/discovery
|
|
import pkg/contractabi/address as ca
|
|
|
|
type MockDiscovery* = ref object of Discovery
|
|
findBlockProvidersHandler*: proc(
|
|
d: MockDiscovery, cid: Cid
|
|
): Future[seq[SignedPeerRecord]] {.async: (raises: [CancelledError]).}
|
|
|
|
publishBlockProvideHandler*:
|
|
proc(d: MockDiscovery, cid: Cid): Future[void] {.async: (raises: [CancelledError]).}
|
|
|
|
findHostProvidersHandler*: proc(
|
|
d: MockDiscovery, host: ca.Address
|
|
): Future[seq[SignedPeerRecord]] {.async: (raises: [CancelledError]).}
|
|
|
|
publishHostProvideHandler*: proc(d: MockDiscovery, host: ca.Address): Future[void] {.
|
|
async: (raises: [CancelledError])
|
|
.}
|
|
|
|
proc new*(T: type MockDiscovery): MockDiscovery =
|
|
MockDiscovery()
|
|
|
|
proc findPeer*(
|
|
d: Discovery, peerId: PeerId
|
|
): Future[?PeerRecord] {.async: (raises: [CancelledError]).} =
|
|
## mock find a peer - always return none
|
|
##
|
|
return none(PeerRecord)
|
|
|
|
method find*(
|
|
d: MockDiscovery, cid: Cid
|
|
): Future[seq[SignedPeerRecord]] {.async: (raises: [CancelledError]).} =
|
|
if isNil(d.findBlockProvidersHandler):
|
|
return
|
|
|
|
return await d.findBlockProvidersHandler(d, cid)
|
|
|
|
method provide*(
|
|
d: MockDiscovery, cid: Cid
|
|
): Future[void] {.async: (raises: [CancelledError]).} =
|
|
if isNil(d.publishBlockProvideHandler):
|
|
return
|
|
|
|
await d.publishBlockProvideHandler(d, cid)
|
|
|
|
method find*(
|
|
d: MockDiscovery, host: ca.Address
|
|
): Future[seq[SignedPeerRecord]] {.async: (raises: [CancelledError]).} =
|
|
if isNil(d.findHostProvidersHandler):
|
|
return
|
|
|
|
return await d.findHostProvidersHandler(d, host)
|
|
|
|
method provide*(
|
|
d: MockDiscovery, host: ca.Address
|
|
): Future[void] {.async: (raises: [CancelledError]).} =
|
|
if isNil(d.publishHostProvideHandler):
|
|
return
|
|
|
|
await d.publishHostProvideHandler(d, host)
|