33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
from abc import ABC, abstractmethod
|
|
from typing import Dict, Tuple, TypeVar
|
|
|
|
from libp2p.stream_muxer.abc import IMuxedStream
|
|
from libp2p.typing import StreamHandlerFn, TProtocol
|
|
|
|
|
|
class IMultiselectMuxer(ABC):
|
|
"""
|
|
Multiselect module that is responsible for responding to
|
|
a multiselect client and deciding on
|
|
a specific protocol and handler pair to use for communication
|
|
"""
|
|
|
|
handlers: Dict[TProtocol, StreamHandlerFn]
|
|
|
|
@abstractmethod
|
|
def add_handler(self, protocol: TProtocol, handler: StreamHandlerFn) -> None:
|
|
"""
|
|
Store the handler with the given protocol
|
|
:param protocol: protocol name
|
|
:param handler: handler function
|
|
"""
|
|
|
|
@abstractmethod
|
|
async def negotiate(self, stream: IMuxedStream) -> Tuple[TProtocol, StreamHandlerFn]:
|
|
"""
|
|
Negotiate performs protocol selection
|
|
:param stream: stream to negotiate on
|
|
:return: selected protocol name, handler function
|
|
:raise Exception: negotiation failed exception
|
|
"""
|