StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests 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
No known key found for this signature in database
GPG Key ID: 570C35F5C2D51B17
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libp2p/pubsub/gossipsub.py
View File

@ -179,8 +179,9 @@ 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:
@ -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)