mirror of
https://github.com/logos-storage/logos-storage-nim.git
synced 2026-05-22 03:09:43 +00:00
4740ffc144
* rework discovery with async queues * misc style changes * increase max message size for large manifests * use upraises and avoid exceptions on key access * increase sleep time to 100 millis * pass config * make list blocks trigger a callback on each block * check for nil on start/stop * fix tests and split out discovery tests * don't auto mount network * add list block tests * add discovery tests * rework moc discovery * move discovery moc to disc dir * don't force logging syncs * don't force moc discovery on all tests * rework discovery with methods * add top level utils file * don't use asyncCheck * don't pass entire blocks to list blocks calback * spelling
45 lines
1.4 KiB
Nim
45 lines
1.4 KiB
Nim
import std/sequtils
|
|
import std/tables
|
|
import pkg/libp2p
|
|
import pkg/chronos
|
|
import pkg/nitro
|
|
import pkg/questionable
|
|
import ./protobuf/blockexc
|
|
import ./protobuf/payments
|
|
import ./protobuf/presence
|
|
|
|
export payments, nitro
|
|
|
|
type
|
|
BlockExcPeerCtx* = ref object of RootObj
|
|
id*: PeerID
|
|
peerPrices*: Table[Cid, UInt256] # remote peer have list including price
|
|
peerWants*: seq[Entry] # remote peers want lists
|
|
exchanged*: int # times peer has exchanged with us
|
|
lastExchange*: Moment # last time peer has exchanged with us
|
|
account*: ?Account # ethereum account of this peer
|
|
paymentChannel*: ?ChannelId # payment channel id
|
|
|
|
proc peerHave*(context: BlockExcPeerCtx): seq[Cid] =
|
|
toSeq(context.peerPrices.keys)
|
|
|
|
proc contains*(a: openArray[BlockExcPeerCtx], b: PeerID): bool =
|
|
## Convenience method to check for peer prepense
|
|
##
|
|
|
|
a.anyIt( it.id == b )
|
|
|
|
func updatePresence*(context: BlockExcPeerCtx, presence: Presence) =
|
|
let cid = presence.cid
|
|
let price = presence.price
|
|
|
|
if cid notin context.peerHave and presence.have:
|
|
context.peerPrices[cid] = price
|
|
elif cid in context.peerHave and not presence.have:
|
|
context.peerPrices.del(cid)
|
|
|
|
func price*(context: BlockExcPeerCtx, cids: seq[Cid]): UInt256 =
|
|
for cid in cids:
|
|
if price =? context.peerPrices.?[cid]:
|
|
result += price
|