status-im/nim-libp2p
2
0
Fork 0
mirror of https://github.com/status-im/nim-libp2p.git synced 2025-01-20 18:00:04 +00:00
Code Issues Projects Releases Wiki Activity

make sure to not remove peers from gossip table

Browse Source
This commit is contained in:
Giovanni Petrantoni 2020-06-29 10:45:06 +09:00
parent d4ca9691d0
commit 2356f6ab68
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
libp2p/protocols/pubsub/gossipsub.nim
View File

@ -190,31 +190,29 @@ proc getGossipPeers(g: GossipSub): Table[string, ControlMessage] {.gcsafe.} =
let fanout: HashSet[string] = g.fanout.getOrDefault(topic)
let gossipPeers = mesh + fanout
var extraPeers = g.gossipsub # copy it!
let mids = g.mcache.window(topic)
if mids.len > 0:
let ihave = ControlIHave(topicID: topic,
messageIDs: toSeq(mids))
if topic notin g.gossipsub:
if topic notin extraPeers:
trace "topic not in gossip array, skipping", topicID = topic
continue
while result.len < GossipSubD:
if g.gossipsub.getOrDefault(topic).len == 0:
if extraPeers.getOrDefault(topic).len == 0:
trace "no peers for topic, skipping", topicID = topic
break
let id = toSeq(g.gossipsub.getOrDefault(topic)).sample()
if id in g.gossipsub.getOrDefault(topic):
g.gossipsub[topic].excl(id)
let id = toSeq(extraPeers.getOrDefault(topic)).sample()
if id in extraPeers.getOrDefault(topic):
extraPeers[topic].excl(id)
if id notin gossipPeers:
if id notin result:
result[id] = ControlMessage()
result[id].ihave.add(ihave)
libp2p_gossipsub_peers_per_topic_gossipsub
.set(g.gossipsub.getOrDefault(topic).len.int64, labelValues = [topic])
proc heartbeat(g: GossipSub) {.async.} =
while g.heartbeatRunning:
try: