mirror of
https://github.com/logos-storage/logos-storage-nim.git
synced 2026-01-02 13:33: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
94 lines
2.5 KiB
Nim
94 lines
2.5 KiB
Nim
## Nim-Codex
|
|
## Copyright (c) 2024 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/chronicles
|
|
import pkg/circomcompat
|
|
import pkg/poseidon2
|
|
import pkg/questionable/results
|
|
|
|
import pkg/libp2p/cid
|
|
|
|
import ../../manifest
|
|
import ../../merkletree
|
|
import ../../stores
|
|
import ../../market
|
|
import ../../utils/poseidon2digest
|
|
import ../../conf
|
|
|
|
import ../builder
|
|
import ../sampler
|
|
|
|
import ./backends
|
|
import ../types
|
|
|
|
export backends
|
|
|
|
logScope:
|
|
topics = "codex prover"
|
|
|
|
type
|
|
AnyProof* = CircomProof
|
|
|
|
AnySampler* = Poseidon2Sampler
|
|
# add any other generic type here, eg. Poseidon2Sampler | ReinforceConcreteSampler
|
|
AnyBuilder* = Poseidon2Builder
|
|
# add any other generic type here, eg. Poseidon2Builder | ReinforceConcreteBuilder
|
|
|
|
AnyProofInputs* = ProofInputs[Poseidon2Hash]
|
|
Prover* = ref object of RootObj
|
|
backend: AnyBackend
|
|
store: BlockStore
|
|
nSamples: int
|
|
|
|
proc prove*(
|
|
self: Prover, slotIdx: int, manifest: Manifest, challenge: ProofChallenge
|
|
): Future[?!(AnyProofInputs, AnyProof)] {.async.} =
|
|
## Prove a statement using backend.
|
|
## Returns a future that resolves to a proof.
|
|
|
|
logScope:
|
|
cid = manifest.treeCid
|
|
slot = slotIdx
|
|
challenge = challenge
|
|
|
|
trace "Received proof challenge"
|
|
|
|
without builder =? AnyBuilder.new(self.store, manifest), err:
|
|
error "Unable to create slots builder", err = err.msg
|
|
return failure(err)
|
|
|
|
without sampler =? AnySampler.new(slotIdx, self.store, builder), err:
|
|
error "Unable to create data sampler", err = err.msg
|
|
return failure(err)
|
|
|
|
without proofInput =? await sampler.getProofInput(challenge, self.nSamples), err:
|
|
error "Unable to get proof input for slot", err = err.msg
|
|
return failure(err)
|
|
|
|
# prove slot
|
|
without proof =? self.backend.prove(proofInput), err:
|
|
error "Unable to prove slot", err = err.msg
|
|
return failure(err)
|
|
|
|
success (proofInput, proof)
|
|
|
|
proc verify*(
|
|
self: Prover, proof: AnyProof, inputs: AnyProofInputs
|
|
): Future[?!bool] {.async.} =
|
|
## Prove a statement using backend.
|
|
## Returns a future that resolves to a proof.
|
|
self.backend.verify(proof, inputs)
|
|
|
|
proc new*(
|
|
_: type Prover, store: BlockStore, backend: AnyBackend, nSamples: int
|
|
): Prover =
|
|
Prover(store: store, backend: backend, nSamples: nSamples)
|