mirror of
https://github.com/status-im/nim-codex.git
synced 2025-01-11 03:16:16 +00:00
d3dbbc75fa
* don't force logging syncs * Add failing test * wip discovery engine * re-add chronicles sinks * wip * move network related stuff to own folder * move peer related stuff to own folder * extract discovery into it's own engine * update imports * move pending blocks into engine module * add top level exports * update imports * update import paths * update imports * support for inflight request filtering and tests * use `remove` instead of `del` * fix sorting in `selectCheapest` * re-org test file structure * fix to use discovery engine * file re-org * fix compilation * fixup discovery to use async handlers * more re-org * rework with support for discovery engine * add logging * use defaults * wip: reworking with discoveryengine * wip: more test fixes * more logging * use ordered table * use `bt` for blocktype Block * fix tests * make tests work with discovery engine * expose all node components * fix to work with discovery engine * wip * propagate cancellation in listBlocks * start/stop disc engine in blockexc engine * remove disc engine start/stop * wire up discovery engine * misc comments and imports * pass discovery to dagger node * set sleep timers * unused imports * misc * don't spawn a task, await it * don't await handlers * trace logging * reduce default sleep time Co-authored-by: Tanguy <tanguy@status.im>
83 lines
2.3 KiB
Nim
83 lines
2.3 KiB
Nim
## Nim-Dagger
|
|
## 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 std/sequtils
|
|
import std/tables
|
|
import std/algorithm
|
|
|
|
import pkg/upraises
|
|
|
|
push: {.upraises: [].}
|
|
|
|
import pkg/chronos
|
|
import pkg/chronicles
|
|
import pkg/libp2p
|
|
|
|
import ../protobuf/blockexc
|
|
|
|
import ../peercontext
|
|
export peercontext
|
|
|
|
logScope:
|
|
topics = "dagger blockexc peerctxstore"
|
|
|
|
type
|
|
PeerCtxStore* = ref object of RootObj
|
|
peers*: OrderedTable[PeerID, BlockExcPeerCtx]
|
|
|
|
iterator items*(self: PeerCtxStore): BlockExcPeerCtx =
|
|
for p in self.peers.values:
|
|
yield p
|
|
|
|
func contains*(self: PeerCtxStore, peerId: PeerID): bool =
|
|
peerId in self.peers
|
|
|
|
func add*(self: PeerCtxStore, peer: BlockExcPeerCtx) =
|
|
trace "Adding peer to peer context store", peer = peer.id
|
|
self.peers[peer.id] = peer
|
|
|
|
func remove*(self: PeerCtxStore, peerId: PeerID) =
|
|
trace "Removing peer from peer context store", peer = peerId
|
|
self.peers.del(peerId)
|
|
|
|
func get*(self: PeerCtxStore, peerId: PeerID): BlockExcPeerCtx =
|
|
trace "Retrieving peer from peer context store", peer = peerId
|
|
self.peers.getOrDefault(peerId, nil)
|
|
|
|
func len*(self: PeerCtxStore): int =
|
|
self.peers.len
|
|
|
|
func peersHave*(self: PeerCtxStore, cid: Cid): seq[BlockExcPeerCtx] =
|
|
toSeq(self.peers.values).filterIt( it.peerHave.anyIt( it == cid ) )
|
|
|
|
func peersWant*(self: PeerCtxStore, cid: Cid): seq[BlockExcPeerCtx] =
|
|
toSeq(self.peers.values).filterIt( it.peerWants.anyIt( it.cid == cid ) )
|
|
|
|
func selectCheapest*(self: PeerCtxStore, cid: Cid): seq[BlockExcPeerCtx] =
|
|
var
|
|
peers = self.peersHave(cid)
|
|
|
|
func cmp(a, b: BlockExcPeerCtx): int =
|
|
# Can't do (a - b) without cast[int](a - b)
|
|
if a.peerPrices.getOrDefault(cid, 0.u256) ==
|
|
b.peerPrices.getOrDefault(cid, 0.u256):
|
|
0
|
|
elif a.peerPrices.getOrDefault(cid, 0.u256) >
|
|
b.peerPrices.getOrDefault(cid, 0.u256):
|
|
1
|
|
else:
|
|
-1
|
|
|
|
peers.sort(cmp)
|
|
trace "Selected cheapest peers", peers = peers.len
|
|
return peers
|
|
|
|
proc new*(T: type PeerCtxStore): PeerCtxStore =
|
|
T(peers: initOrderedTable[PeerID, BlockExcPeerCtx]())
|