This commit is contained in:
zixuanzh
2018-11-12 13:02:49 -05:00
parent b4272918d9
commit c5c9d3e5c9
17 changed files with 66 additions and 84 deletions

View File

@ -1,10 +1,11 @@
import asyncio
from .net_stream_interface import INetStream
class NetStream(INetStream):
def __init__(self, muxed_stream):
self.muxed_stream = muxed_stream
self.protocol_id = None
def get_protocol(self):
"""
@ -26,12 +27,12 @@ class NetStream(INetStream):
"""
return await self.muxed_stream.read()
async def write(self, bytes):
async def write(self, data):
"""
write to stream
:return: number of bytes written
"""
return await self.muxed_stream.write(bytes)
return await self.muxed_stream.write(data)
def close(self):
"""

View File

@ -1,11 +1,7 @@
from abc import ABC, abstractmethod
class INetStream(ABC):
def __init__(self, peer_id, multi_addr, connection):
self.peer_id = peer_id
self.multi_addr = multi_addr
self.connection = connection
class INetStream(ABC):
@abstractmethod
def get_protocol(self):