py-libp2p/network/stream_interface.py
2018-10-15 01:52:25 -04:00

47 lines
911 B
Python

from abc import ABC, abstractmethod
class IStream(ABC):
def __init__(self, context, peer_id):
self.context = context
self.peer_id = peer_id
@abstractmethod
def protocol(self):
"""
:return: protocol id that stream runs on
"""
pass
@abstractmethod
def set_protocol(self, protocol_id):
"""
:param protocol_id: protocol id that stream runs on
:return: true if successful
"""
pass
@abstractmethod
def read(self):
"""
read from stream
:return: bytes of input
"""
pass
@abstractmethod
def write(self, _bytes):
"""
write to stream
:return: number of bytes written
"""
pass
@abstractmethod
def close(self):
"""
close stream
:return: true if successful
"""
pass