varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-09 22:50:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

added RoutedHost and updated new_node to support it

Browse Source
This commit is contained in:
Alex Haynes
2019-04-20 17:35:05 -04:00
parent 367a939087
commit e7424d3673
6 changed files with 85 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
libp2p/__init__.py
View File

@ -6,8 +6,10 @@ from .peer.peerstore import PeerStore
from .peer.id import id_from_public_key
from .network.swarm import Swarm
from .host.basic_host import BasicHost
from .kademlia.routed_host import RoutedHost
from .transport.upgrader import TransportUpgrader
from .transport.tcp.tcp import TCP
from .kademlia.network import KademliaServer
async def cleanup_done_tasks():
@ -23,6 +25,30 @@ async def cleanup_done_tasks():
# Some sleep necessary to context switch
await asyncio.sleep(3)
def generate_id():
new_key = RSA.generate(2048, e=65537)
new_id = id_from_public_key(new_key.publickey())
# private_key = new_key.exportKey("PEM")
return new_id
def initialize_default_kademlia(
ksize=20, alpha=3, id_opt=None, storage=None):
"""
initialize swam when no swarm is passed in
:param ksize: The k parameter from the paper
:param alpha: The alpha parameter from the paper
:param id_opt: optional id for host
:param storage: An instance that implements
:interface:`~kademlia.storage.IStorage`
:return: return a default kademlia instance
"""
if not id_opt:
id_opt = generate_id()
node_id = id_opt.get_raw_id()
return KademliaServer(ksize=ksize, alpha=alpha,
node_id=node_id, storage=storage)
def initialize_default_swarm(
id_opt=None, transport_opt=None,
@ -37,10 +63,9 @@ def initialize_default_swarm(
:return: return a default swarm instance
"""
# pylint: disable=too-many-arguments, unused-argument
if not id_opt:
new_key = RSA.generate(2048, e=65537)
id_opt = id_from_public_key(new_key.publickey())
# private_key = new_key.exportKey("PEM")
id_opt = generate_id()
transport_opt = transport_opt or ["/ip4/127.0.0.1/tcp/8001"]
transport = [multiaddr.Multiaddr(t) for t in transport_opt]
@ -58,7 +83,8 @@ def initialize_default_swarm(
async def new_node(
swarm_opt=None, id_opt=None, transport_opt=None,
muxer_opt=None, sec_opt=None, peerstore_opt=None):
muxer_opt=None, sec_opt=None, peerstore_opt=None,
disc_opt=None):
"""
create new libp2p node
:param id_opt: optional id for host
@ -69,15 +95,22 @@ async def new_node(
:return: return a default swarm instance
"""
# pylint: disable=too-many-arguments
if not id_opt:
id_opt = generate_id()
if not swarm_opt:
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)
if not disc_opt:
disc_opt = initialize_default_kademlia(ksize=20, alpha=3,
id_opt=id_opt, storage=None)
# TODO enable support for other host type
# TODO routing unimplemented
host = BasicHost(swarm_opt)
host = RoutedHost(swarm_opt, disc_opt)
# Kick off cleanup job
asyncio.ensure_future(cleanup_done_tasks())