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

Conform stream_id to go-mplex

Browse Source
This commit is contained in:
mhchia
2019-08-28 21:43:34 +08:00
parent 9b60e1757d
commit d35b8ffc64
5 changed files with 55 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
libp2p/stream_muxer/abc.py
View File

@ -4,6 +4,7 @@ from typing import TYPE_CHECKING, Optional
from libp2p.peer.id import ID
from libp2p.security.secure_conn_interface import ISecureConn
from libp2p.stream_muxer.mplex.constants import HeaderTags
from libp2p.stream_muxer.mplex.datastructures import StreamID
if TYPE_CHECKING:
# Prevent GenericProtocolHandlerFn introducing circular dependencies
@ -51,7 +52,7 @@ class IMuxedConn(ABC):
"""
@abstractmethod
async def read_buffer(self, stream_id: int) -> bytes:
async def read_buffer(self, stream_id: StreamID) -> bytes:
"""
Read a message from stream_id's buffer, check raw connection for new messages
:param stream_id: stream id of stream to read from
@ -59,7 +60,7 @@ class IMuxedConn(ABC):
"""
@abstractmethod
async def read_buffer_nonblocking(self, stream_id: int) -> Optional[bytes]:
async def read_buffer_nonblocking(self, stream_id: StreamID) -> Optional[bytes]:
"""
Read a message from `stream_id`'s buffer, non-blockingly.
"""
@ -78,7 +79,9 @@ class IMuxedConn(ABC):
"""
@abstractmethod
async def send_message(self, flag: HeaderTags, data: bytes, stream_id: int) -> int:
async def send_message(
self, flag: HeaderTags, data: bytes, stream_id: StreamID
) -> int:
"""
sends a message over the connection
:param header: header to use