refactor add_router in swarm

This commit is contained in:
zixuanzh
2019-04-28 15:00:43 -04:00
parent 21a99287b1
commit 8ab387ac05
2 changed files with 8 additions and 7 deletions

View File

@ -10,17 +10,17 @@ from .stream.net_stream import NetStream
from .connection.raw_connection import RawConnection
class Swarm(INetwork):
# pylint: disable=too-many-instance-attributes, cell-var-from-loop
# pylint: disable=too-many-instance-attributes,cell-var-from-loop,too-many-arguments
def __init__(self, peer_id, peerstore, upgrader, transport):
def __init__(self, peer_id, peerstore, upgrader, transport, router):
self.self_id = peer_id
self.peerstore = peerstore
self.upgrader = upgrader
self.transport = transport
self.router = router
self.connections = dict()
self.listeners = dict()
self.stream_handlers = dict()
self.router = None
# Protocol muxing
self.multiselect = Multiselect()