varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-04-05 21:41:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: log cleanup

Browse Source
This commit is contained in:
Akash Mondal
2025-07-04 06:40:22 +00:00
committed by lla-dane
parent 03bf071739
commit 0f64bb49b5
7 changed files with 56 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
libp2p/network/stream/net_stream.py
View File

@ -1,7 +1,6 @@
from enum import (
Enum,
)
import inspect
import trio
@ -165,25 +164,20 @@ class NetStream(INetStream):
data = await self.muxed_stream.read(n)
return data
except MuxedStreamEOF as error:
print("NETSTREAM: READ ERROR, RECEIVED EOF")
async with self._state_lock:
if self.__stream_state == StreamState.CLOSE_WRITE:
self.__stream_state = StreamState.CLOSE_BOTH
print("NETSTREAM: READ ERROR, REMOVING STREAM")
await self._remove()
elif self.__stream_state == StreamState.OPEN:
print("NETSTREAM: READ ERROR, NEW STATE -> CLOSE_READ")
self.__stream_state = StreamState.CLOSE_READ
raise StreamEOF() from error
except (MuxedStreamReset, QUICStreamClosedError, QUICStreamResetError) as error:
print("NETSTREAM: READ ERROR, MUXED STREAM RESET")
async with self._state_lock:
if self.__stream_state in [
StreamState.OPEN,
StreamState.CLOSE_READ,
StreamState.CLOSE_WRITE,
]:
print("NETSTREAM: READ ERROR, NEW STATE -> RESET")
self.__stream_state = StreamState.RESET
await self._remove()
raise StreamReset() from error
@ -222,8 +216,6 @@ class NetStream(INetStream):
async def close(self) -> None:
"""Close stream for writing."""
print("NETSTREAM: CLOSING STREAM, CURRENT STATE: ", self.__stream_state)
print("CALLED BY: ", inspect.stack()[1].function)
async with self._state_lock:
if self.__stream_state in [
StreamState.CLOSE_BOTH,
@ -243,7 +235,6 @@ class NetStream(INetStream):
async def reset(self) -> None:
"""Reset stream, closing both ends."""
print("NETSTREAM: RESETING STREAM")
async with self._state_lock:
if self.__stream_state == StreamState.RESET:
return