Default protocols are constructed using a reference to the host

This commit is contained in:
Alex Stokes 2019-10-28 18:48:00 +09:00
parent e0a94b6092
commit 693a8cf99a
No known key found for this signature in database
GPG Key ID: 51CE1721B245C086
2 changed files with 5 additions and 5 deletions

View File

@ -55,7 +55,7 @@ class BasicHost(IHost):
self._network.set_stream_handler(self._swarm_stream_handler)
self.peerstore = self._network.peerstore
# Protocol muxing
default_protocols = default_protocols or get_default_protocols()
default_protocols = default_protocols or get_default_protocols(self)
self.multiselect = Multiselect(default_protocols)
self.multiselect_client = MultiselectClient()

View File

@ -1,11 +1,11 @@
from collections import OrderedDict
from typing import TYPE_CHECKING
from libp2p.host.host_interface import IHost
if TYPE_CHECKING:
from libp2p.typing import TProtocol, StreamHandlerFn
DEFAULT_HOST_PROTOCOLS: "OrderedDict[TProtocol, StreamHandlerFn]" = OrderedDict()
def get_default_protocols() -> "OrderedDict[TProtocol, StreamHandlerFn]":
return DEFAULT_HOST_PROTOCOLS.copy()
def get_default_protocols(host: IHost) -> "OrderedDict[TProtocol, StreamHandlerFn]":
return OrderedDict()