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

Add stream.write error handling in gossipsub

Browse Source
This commit is contained in:
NIC619 2019-11-16 16:24:48 +08:00
parent 7d1f3d6000
commit ccc7879422
No known key found for this signature in database
GPG Key ID: 570C35F5C2D51B17
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libp2p/pubsub/gossipsub.py
View File

@ -518,7 +518,10 @@ class GossipSub(IPubsubRouter):
peer_stream = self.pubsub.peers[sender_peer_id] peer_stream = self.pubsub.peers[sender_peer_id]
# 4) And write the packet to the stream # 4) And write the packet to the stream
try:
await peer_stream.write(encode_varint_prefixed(rpc_msg)) await peer_stream.write(encode_varint_prefixed(rpc_msg))
except StreamClosed:
logger.debug("Fail to responed to iwant request from %s: stream closed", sender_peer_id)
async def handle_graft( async def handle_graft(
self, graft_msg: rpc_pb2.ControlGraft, sender_peer_id: ID self, graft_msg: rpc_pb2.ControlGraft, sender_peer_id: ID
@ -602,4 +605,7 @@ class GossipSub(IPubsubRouter):
peer_stream = self.pubsub.peers[to_peer] peer_stream = self.pubsub.peers[to_peer]
# Write rpc to stream # Write rpc to stream
try:
await peer_stream.write(encode_varint_prefixed(rpc_msg)) await peer_stream.write(encode_varint_prefixed(rpc_msg))
except StreamClosed:
logger.debug("Fail to emit control message to %s: stream closed", to_peer)