vacp2p
/
nim-libp2p-experimental
mirror of https://github.com/vacp2p/nim-libp2p-experimental.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

avoid triggering unsubscribeAll from unsubscribe (gossip)

This commit is contained in:
Giovanni Petrantoni 2020-12-20 17:08:03 +09:00
parent 4858e0ab15
commit 5e79d3ab9c
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1251,7 +1251,7 @@ method subscribe*(g: GossipSub,
g.rebalanceMesh(topic) g.rebalanceMesh(topic)
method unsubscribeAll*(g: GossipSub, topic: string) = proc unsubscribe*(g: GossipSub, topic: string) =
var var
msg = RPCMsg.withSubs(@[topic], subscribe = false) msg = RPCMsg.withSubs(@[topic], subscribe = false)
gpeers = g.gossipsub.getOrDefault(topic) gpeers = g.gossipsub.getOrDefault(topic)
@ -1281,6 +1281,8 @@ method unsubscribeAll*(g: GossipSub, topic: string) =
else: else:
g.broadcast(toSeq(gpeers), msg) g.broadcast(toSeq(gpeers), msg)
method unsubscribeAll*(g: GossipSub, topic: string) =
g.unsubscribe(topic)
# finally let's remove from g.topics, do that by calling PubSub # finally let's remove from g.topics, do that by calling PubSub
procCall PubSub(g).unsubscribeAll(topic) procCall PubSub(g).unsubscribeAll(topic)
@ -1292,7 +1294,7 @@ method unsubscribe*(g: GossipSub,
# delete from mesh only if no handlers are left # delete from mesh only if no handlers are left
# (handlers are removed in pubsub unsubscribe above) # (handlers are removed in pubsub unsubscribe above)
if topic notin g.topics: if topic notin g.topics:
g.unsubscribeAll(topic) g.unsubscribe(topic)
method publish*(g: GossipSub, method publish*(g: GossipSub,
topic: string, topic: string,