StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix list deletion and add list remove check

Browse Source
This commit is contained in:
NIC619 2019-12-03 22:37:49 +08:00
parent bb15c817b1
commit a9abf1e3dd
No known key found for this signature in database
GPG Key ID: 570C35F5C2D51B17
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
libp2p/pubsub/gossipsub.py
View File

@ -460,9 +460,12 @@ class GossipSub(IPubsubRouter):
else:
# Check whether our peers are still in the topic
# ref: https://github.com/libp2p/go-libp2p-pubsub/blob/01b9825fbee1848751d90a8469e3f5f43bac8466/gossipsub.go#L498-L504 # noqa: E501
for peer in self.fanout[topic]:
if peer not in self.pubsub.peer_topics[topic]:
self.fanout[topic].remove(peer)
in_topic_fanout_peers = [
peer
for peer in self.fanout[topic]
if peer in self.pubsub.peer_topics[topic]
]
self.fanout[topic] = in_topic_fanout_peers
num_fanout_peers_in_topic = len(self.fanout[topic])
# If |fanout[topic]| < D
@ -644,7 +647,10 @@ class GossipSub(IPubsubRouter):
# Remove peer from mesh for topic, if peer is in topic
if topic in self.mesh and sender_peer_id in self.mesh[topic]:
self.mesh[topic].remove(sender_peer_id)
if len(self.mesh[topic]) == 1:
del self.mesh[topic]
else:
self.mesh[topic].remove(sender_peer_id)
# RPC emitters