49 lines
1.5 KiB
Nim
49 lines
1.5 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
|
|
peerHave*: seq[Cid] # remote peers have lists
|
|
peerWants*: seq[Entry] # remote peers want lists
|
|
bytesSent*: int # bytes sent to remote
|
|
bytesRecv*: int # bytes received from remote
|
|
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 )
|
|
|
|
proc debtRatio*(b: BitswapPeerCtx): float =
|
|
b.bytesSent / (b.bytesRecv + 1)
|
|
|
|
proc `<`*(a, b: BitswapPeerCtx): bool =
|
|
a.debtRatio < b.debtRatio
|
|
|
|
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)
|