48 lines
976 B
Python
48 lines
976 B
Python
|
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
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def get_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
|