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

Remove unneccessary filter and check in gossipsub

Browse Source
This commit is contained in:
NIC619
2019-12-02 22:40:35 +08:00
parent c2d88962c7
commit 357341e0d8
1 changed files with 3 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
libp2p/pubsub/gossipsub.py
View File

@ -320,10 +320,7 @@ class GossipSub(IPubsubRouter):
topic, self.degree - num_mesh_peers_in_topic, self.mesh[topic] topic, self.degree - num_mesh_peers_in_topic, self.mesh[topic]
) )
fanout_peers_not_in_mesh: List[ID] = [ for peer in selected_peers:
peer for peer in selected_peers if peer not in self.mesh[topic]
]
for peer in fanout_peers_not_in_mesh:
# Add peer to mesh[topic] # Add peer to mesh[topic]
self.mesh[topic].append(peer) self.mesh[topic].append(peer)
@ -377,12 +374,7 @@ class GossipSub(IPubsubRouter):
) )
for peer in peers_to_emit_ihave_to: for peer in peers_to_emit_ihave_to:
# TODO: this line is a monster, can hopefully be simplified if peer not in self.mesh[topic]:
if (
topic not in self.mesh or (peer not in self.mesh[topic])
) and (
topic not in self.fanout or (peer not in self.fanout[topic])
):
msg_id_strs = [str(msg_id) for msg_id in msg_ids] msg_id_strs = [str(msg_id) for msg_id in msg_ids]
await self.emit_ihave(topic, msg_id_strs, peer) await self.emit_ihave(topic, msg_id_strs, peer)
@ -399,10 +391,7 @@ class GossipSub(IPubsubRouter):
topic, self.degree, [] topic, self.degree, []
) )
for peer in peers_to_emit_ihave_to: for peer in peers_to_emit_ihave_to:
if ( if peer not in self.fanout[topic]:
peer not in self.mesh[topic]
and peer not in self.fanout[topic]
):
msg_id_strs = [str(msg) for msg in msg_ids] msg_id_strs = [str(msg) for msg in msg_ids]
await self.emit_ihave(topic, msg_id_strs, peer) await self.emit_ihave(topic, msg_id_strs, peer)