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

refactor add_router in swarm

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

9
libp2p/__init__.py
View File

@ -52,8 +52,8 @@ def initialize_default_kademlia_router(
def initialize_default_swarm(
id_opt=None, transport_opt=None,
muxer_opt=None, sec_opt=None, peerstore_opt=None):
id_opt=None, transport_opt=None, muxer_opt=None,
sec_opt=None, peerstore_opt=None, disc_opt=None):
"""
initialize swarm when no swarm is passed in
:param id_opt: optional id for host
@ -79,7 +79,8 @@ def initialize_default_swarm(
upgrader = TransportUpgrader(sec, muxer)
peerstore = peerstore_opt or PeerStore()
swarm_opt = Swarm(id_opt, peerstore, upgrader, transport)
swarm_opt = Swarm(id_opt, peerstore,\
upgrader, transport, disc_opt)
return swarm_opt
@ -106,7 +107,7 @@ async def new_node(
swarm_opt = initialize_default_swarm(
id_opt=id_opt, transport_opt=transport_opt,
muxer_opt=muxer_opt, sec_opt=sec_opt,
peerstore_opt=peerstore_opt)
peerstore_opt=peerstore_opt, disc_opt=disc_opt)
swarm_opt.add_router(disc_opt)