varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-09 22:50:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Modify NetStream to read n bytes

Browse Source
This commit is contained in:
mhchia
2019-08-07 15:23:20 +08:00
committed by Kevin Mai-Husan Chia
parent dbdbcf7440
commit 2485a00e24
4 changed files with 34 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -28,12 +28,13 @@ class NetStream(INetStream):
"""
self.protocol_id = protocol_id
async def read(self) -> bytes:
async def read(self, n: int = -1) -> bytes:
"""
read from stream
:return: bytes of input until EOF
reads from stream
:param n: number of bytes to read
:return: bytes of input
"""
return await self.muxed_stream.read()
return await self.muxed_stream.read(n)
async def write(self, data: bytes) -> int:
"""

3
libp2p/network/stream/net_stream_interface.py
View File

@ -22,9 +22,10 @@ class INetStream(ABC):
"""
@abstractmethod
async def read(self) -> bytes:
async def read(self, n: int = -1) -> bytes:
"""
reads from the underlying muxed_stream
:param n: number of bytes to read
:return: bytes of input
"""