mirror of
https://github.com/logos-storage/logos-storage-nim.git
synced 2026-02-20 13:33:09 +00:00
e3b68f96d8
Rationale: selecting peers based on debt ratio in bytes is no longer needed when the bandwidth for each chunk is compensated.
45 lines
1.3 KiB
Nim
45 lines
1.3 KiB
Nim
import std/sequtils
|
|
import std/tables
|
|
import pkg/libp2p
|
|
import pkg/chronos
|
|
import pkg/nitro
|
|
import pkg/questionable
|
|
import ./protobuf/bitswap
|
|
import ./protobuf/payments
|
|
import ./protobuf/presence
|
|
|
|
export payments
|
|
export nitro
|
|
|
|
type
|
|
BitswapPeerCtx* = 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
|
|
pricing*: ?Pricing # optional bandwidth price for this peer
|
|
paymentChannel*: ?ChannelId # payment channel id
|
|
|
|
proc contains*(a: openArray[BitswapPeerCtx], b: PeerID): bool =
|
|
## Convenience method to check for peer prepense
|
|
##
|
|
|
|
a.anyIt( it.id == b )
|
|
|
|
func updatePresence*(context: BitswapPeerCtx, presence: Presence) =
|
|
let cid = presence.cid
|
|
let price = presence.price
|
|
|
|
if cid notin context.peerHave and presence.have:
|
|
context.peerHave.add(cid)
|
|
context.peerPrices[cid] = price
|
|
elif cid in context.peerHave and not presence.have:
|
|
context.peerHave.keepItIf(it != cid)
|
|
context.peerPrices.del(cid)
|
|
|
|
func price*(context: BitswapPeerCtx, cids: seq[Cid]): UInt256 =
|
|
for cid in cids:
|
|
if price =? context.peerPrices.?[cid]:
|
|
result += price
|