status-im
/
nim-libp2p
mirror of https://github.com/status-im/nim-libp2p.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

improve fanout replenish behavior further

This commit is contained in:
Giovanni Petrantoni 2020-06-30 11:36:47 +09:00
parent 509d84f03d
commit f49e59cb4e
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -218,6 +218,11 @@ proc heartbeat(g: GossipSub) {.async.} =
await g.rebalanceMesh(t) await g.rebalanceMesh(t)
await g.dropFanoutPeers() await g.dropFanoutPeers()
# replenish known topics to the fanout
for t in toSeq(g.fanout.keys):
g.replenishFanout(t)
let peers = g.getGossipPeers() let peers = g.getGossipPeers()
var sent: seq[Future[void]] var sent: seq[Future[void]]
for peer in peers.keys: for peer in peers.keys:
@ -470,8 +475,12 @@ method publish*(g: GossipSub,
if topic in g.topics: # if we're subscribed use the mesh if topic in g.topics: # if we're subscribed use the mesh
peers = g.mesh.getOrDefault(topic) peers = g.mesh.getOrDefault(topic)
else: # not subscribed, send to fanout peers else: # not subscribed, send to fanout peers
g.replenishFanout(topic) # try optimistically
peers = g.fanout.getOrDefault(topic) peers = g.fanout.getOrDefault(topic)
if peers.len == 0:
# ok we had nothing.. let's try replenish inline
g.replenishFanout(topic)
peers = g.fanout.getOrDefault(topic)
let let
msg = Message.init(g.peerInfo, data, topic, g.sign) msg = Message.init(g.peerInfo, data, topic, g.sign)