82840b5e6c
* Add generic protocol handler * Add generic protocol handler to stream muxing pipeline * Modify conn_handler to only deal with connections * mplex accept stream architecture changes * Add create generic protocol handler * Fix minor bugs * who would win 4 devs or one not * Debugging * rearch with handle_incoming infinite loop, seems to work, needs cleanup" * passing linting, still needs cleanup * fixing linting again; code still needs cleanup * fixing tests; code still needs cleanup * adding test cleanup and task cleanup, removing prints * linting, and cleanup complete * storing connections based on peer id * remove dead code * remove unnecessary peer_id
47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
from abc import ABC, abstractmethod
|
|
|
|
|
|
class IMuxedConn(ABC):
|
|
"""
|
|
reference: https://github.com/libp2p/go-stream-muxer/blob/master/muxer.go
|
|
"""
|
|
|
|
@abstractmethod
|
|
def __init__(self, conn, generic_protocol_handler):
|
|
"""
|
|
create a new muxed connection
|
|
:param conn: an instance of raw connection
|
|
:param generic_protocol_handler: generic protocol handler
|
|
for new muxed streams
|
|
"""
|
|
|
|
@abstractmethod
|
|
def close(self):
|
|
"""
|
|
close connection
|
|
:return: true if successful
|
|
"""
|
|
|
|
@abstractmethod
|
|
def is_closed(self):
|
|
"""
|
|
check connection is fully closed
|
|
:return: true if successful
|
|
"""
|
|
|
|
@abstractmethod
|
|
def open_stream(self, protocol_id, multi_addr):
|
|
"""
|
|
creates a new muxed_stream
|
|
:param protocol_id: protocol_id of stream
|
|
:param multi_addr: multi_addr that stream connects to
|
|
:return: a new stream
|
|
"""
|
|
|
|
@abstractmethod
|
|
def accept_stream(self):
|
|
"""
|
|
accepts a muxed stream opened by the other end
|
|
:return: the accepted stream
|
|
"""
|