Apply PR feedback:
Only use pop method if graceful failure handling is desired
This commit is contained in:
parent
19907e18ec
commit
501eef59de
|
@ -270,9 +270,11 @@ class Swarm(INetwork):
|
|||
"""Simply remove the connection from Swarm's records, without closing
|
||||
the connection."""
|
||||
peer_id = swarm_conn.muxed_conn.peer_id
|
||||
if peer_id not in self.connections:
|
||||
return
|
||||
# TODO: Should be changed to remove the exact connection,
|
||||
# if we have several connections per peer in the future.
|
||||
self.connections.pop(peer_id, None)
|
||||
del self.connections[peer_id]
|
||||
|
||||
# Notifee
|
||||
|
||||
|
|
|
@ -353,8 +353,8 @@ class GossipSub(IPubsubRouter):
|
|||
# TODO: there's no way time_since_last_publish gets set anywhere yet
|
||||
if self.time_since_last_publish[topic] > self.time_to_live:
|
||||
# Remove topic from fanout
|
||||
self.fanout.pop(topic, None)
|
||||
self.time_since_last_publish.pop(topic, None)
|
||||
del self.fanout[topic]
|
||||
del self.time_since_last_publish[topic]
|
||||
else:
|
||||
num_fanout_peers_in_topic = len(self.fanout[topic])
|
||||
|
||||
|
|
|
@ -298,7 +298,7 @@ class Pubsub:
|
|||
def _handle_dead_peer(self, peer_id: ID) -> None:
|
||||
if peer_id not in self.peers:
|
||||
return
|
||||
self.peers.pop(peer_id, None)
|
||||
del self.peers[peer_id]
|
||||
|
||||
for topic in self.peer_topics:
|
||||
if peer_id in self.peer_topics[topic]:
|
||||
|
|
Loading…
Reference in New Issue
Block a user