Apply PR feedback

This commit is contained in:
NIC619 2019-07-22 23:22:07 +08:00
parent cdbeb63879
commit afc853a776
No known key found for this signature in database
GPG Key ID: 570C35F5C2D51B17

View File

@ -265,10 +265,12 @@ class GossipSub(IPubsubRouter):
async def mesh_heartbeat(self): async def mesh_heartbeat(self):
# Note: the comments here are the exact pseudocode from the spec # Note: the comments here are the exact pseudocode from the spec
for topic in self.mesh: for topic in self.mesh:
# Skip if no peers have subscribed to the topic
if topic not in self.pubsub.peer_topics:
continue
num_mesh_peers_in_topic = len(self.mesh[topic]) num_mesh_peers_in_topic = len(self.mesh[topic])
if num_mesh_peers_in_topic < self.degree_low: if num_mesh_peers_in_topic < self.degree_low:
if topic in self.pubsub.peer_topics:
gossipsub_peers_in_topic = [peer for peer in self.pubsub.peer_topics[topic] gossipsub_peers_in_topic = [peer for peer in self.pubsub.peer_topics[topic]
if peer in self.peers_gossipsub] if peer in self.peers_gossipsub]