varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

add typing to protocol_muxer

Browse Source
This commit is contained in:
Chih Cheng Liang
2019-08-07 18:02:30 +08:00
committed by Kevin Mai-Husan Chia
parent 0d709364f8
commit 5903012e0e
13 changed files with 72 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libp2p/protocol_muxer/multiselect_client_interface.py
View File

@ -1,4 +1,8 @@
from abc import ABC, abstractmethod
from typing import Sequence
from libp2p.stream_muxer.abc import IMuxedStream
from libp2p.typing import TProtocol
class IMultiselectClient(ABC):
@ -8,7 +12,7 @@ class IMultiselectClient(ABC):
"""
@abstractmethod
def select_protocol_or_fail(self, protocol, stream):
async def select_protocol_or_fail(self, protocol: TProtocol, stream: IMuxedStream) -> TProtocol:
"""
Send message to multiselect selecting protocol
and fail if multiselect does not return same protocol
@ -18,7 +22,9 @@ class IMultiselectClient(ABC):
"""
@abstractmethod
def select_one_of(self, protocols, stream):
async def select_one_of(
self, protocols: Sequence[TProtocol], stream: IMuxedStream
) -> TProtocol:
"""
For each protocol, send message to multiselect selecting protocol
and fail if multiselect does not return same protocol. Returns first