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

Ping protocol: move with statement out of try

Browse Source
This commit is contained in:
mhchia
2020-02-04 15:09:42 +08:00
parent 113696dce2
commit 5da102d1c9
1 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
libp2p/host/ping.py
View File

@ -17,23 +17,23 @@ logger = logging.getLogger("libp2p.host.ping")
async def _handle_ping(stream: INetStream, peer_id: PeerID) -> bool: async def _handle_ping(stream: INetStream, peer_id: PeerID) -> bool:
"""Return a boolean indicating if we expect more pings from the peer at """Return a boolean indicating if we expect more pings from the peer at
``peer_id``.""" ``peer_id``."""
with trio.fail_after(RESP_TIMEOUT): try:
try: with trio.fail_after(RESP_TIMEOUT):
payload = await stream.read(PING_LENGTH) payload = await stream.read(PING_LENGTH)
except trio.TooSlowError as error: except trio.TooSlowError as error:
logger.debug("Timed out waiting for ping from %s: %s", peer_id, error) logger.debug("Timed out waiting for ping from %s: %s", peer_id, error)
raise raise
except StreamEOF: except StreamEOF:
logger.debug("Other side closed while waiting for ping from %s", peer_id) logger.debug("Other side closed while waiting for ping from %s", peer_id)
return False return False
except StreamReset as error: except StreamReset as error:
logger.debug( logger.debug(
"Other side reset while waiting for ping from %s: %s", peer_id, error "Other side reset while waiting for ping from %s: %s", peer_id, error
) )
raise raise
except Exception as error: except Exception as error:
logger.debug("Error while waiting to read ping for %s: %s", peer_id, error) logger.debug("Error while waiting to read ping for %s: %s", peer_id, error)
raise raise
logger.debug("Received ping from %s with data: 0x%s", peer_id, payload.hex()) logger.debug("Received ping from %s with data: 0x%s", peer_id, payload.hex())