py-libp2p/network/stream_interface.py
2018-10-21 15:11:27 -04:00

47 lines
920 B
Python

from abc import ABC, abstractmethod
class IStream(ABC):
def __init__(self, peer_id, multi_addr):
self.peer_id = peer_id
self.multi_addr = multi_addr
@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