varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-11 07:30:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add check to prevent gossipsub re-join and re-leave

Browse Source
This commit is contained in:
NIC619
2019-07-19 19:56:25 +08:00
parent 183eee0e85
commit 36575e8c9b
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
libp2p/pubsub/gossipsub.py
View File

@ -179,9 +179,10 @@ class GossipSub(IPubsubRouter):
subscription announcement subscription announcement
:param topic: topic to join :param topic: topic to join
""" """
if topic in self.mesh:
return
# Create mesh[topic] if it does not yet exist # Create mesh[topic] if it does not yet exist
if topic not in self.mesh: self.mesh[topic] = []
self.mesh[topic] = []
if topic in self.fanout and len(self.fanout[topic]) == self.degree: if topic in self.fanout and len(self.fanout[topic]) == self.degree:
# If router already has D peers from the fanout peers of a topic # If router already has D peers from the fanout peers of a topic
@ -228,6 +229,8 @@ class GossipSub(IPubsubRouter):
It is invoked after the unsubscription announcement. It is invoked after the unsubscription announcement.
:param topic: topic to leave :param topic: topic to leave
""" """
if topic not in self.mesh:
return
# Notify the peers in mesh[topic] with a PRUNE(topic) message # Notify the peers in mesh[topic] with a PRUNE(topic) message
for peer in self.mesh[topic]: for peer in self.mesh[topic]:
await self.emit_prune(topic, peer) await self.emit_prune(topic, peer)