status-im/nim-libp2p
2
0
Fork 0
mirror of https://github.com/status-im/nim-libp2p.git synced 2025-02-22 17:48:12 +00:00
Code Issues Projects Releases Wiki Activity

fix: limit

Browse Source
This commit is contained in:
Ludovic Chenut 2024-10-03 12:00:58 +02:00
parent 43d1ce2ef1
commit ea7d6c8b3a
No known key found for this signature in database
GPG Key ID: D9A59B1907F1D50C
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libp2p/protocols/rendezvous.nim
View File

@ -558,7 +558,7 @@ proc requestLocally*(rdv: RendezVous, ns: string): seq[PeerRecord] =
@[] @[]
proc request*( proc request*(
rdv: RendezVous, ns: string, limit: int = DiscoverLimit.int, peers: seq[PeerId] rdv: RendezVous, ns: string, limit: uint64 = DiscoverLimit, peers: seq[PeerId]
): Future[seq[PeerRecord]] {.async.} = ): Future[seq[PeerRecord]] {.async.} =
## This async procedure discovers and returns peers for a given namespace ## This async procedure discovers and returns peers for a given namespace
## by sending requests and processing responses. It limits the number of ## by sending requests and processing responses. It limits the number of
@ -566,14 +566,12 @@ proc request*(
## ##
var var
s: Table[PeerId, (PeerRecord, Register)] s: Table[PeerId, (PeerRecord, Register)]
limit: uint64
d = Discover(ns: ns) d = Discover(ns: ns)
if limit <= 0 or limit > DiscoverLimit.int: if limit > DiscoverLimit.int:
raise newException(RendezVousError, "Invalid limit") raise newException(RendezVousError, "Invalid limit")
if ns.len notin 0 .. 255: if ns.len notin 0 .. 255:
raise newException(RendezVousError, "Invalid namespace") raise newException(RendezVousError, "Invalid namespace")
limit = limit.uint64
proc requestPeer(peer: PeerId) {.async.} = proc requestPeer(peer: PeerId) {.async.} =
let conn = await rdv.switch.dial(peer, RendezVousCodec) let conn = await rdv.switch.dial(peer, RendezVousCodec)
defer: defer:
@ -641,7 +639,7 @@ proc request*(
return toSeq(s.values()).mapIt(it[0]) return toSeq(s.values()).mapIt(it[0])
proc request*( proc request*(
rdv: RendezVous, ns: string, limit: int = DiscoverLimit.int rdv: RendezVous, ns: string, limit: uint64 = DiscoverLimit
): Future[seq[PeerRecord]] {.async.} = ): Future[seq[PeerRecord]] {.async.} =
await rdv.request(ns, limit, rdv.peers) await rdv.request(ns, limit, rdv.peers)