StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update peer store initialization in host factory

Browse Source
This commit is contained in:
NIC619 2019-11-25 17:17:09 +08:00
parent 8b4022328d
commit e28a974425
No known key found for this signature in database
GPG Key ID: 570C35F5C2D51B17
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
libp2p/tools/factories.py
View File

@ -10,6 +10,7 @@ from libp2p.host.basic_host import BasicHost
from libp2p.network.connection.swarm_connection import SwarmConn from libp2p.network.connection.swarm_connection import SwarmConn
from libp2p.network.stream.net_stream_interface import INetStream from libp2p.network.stream.net_stream_interface import INetStream
from libp2p.network.swarm import Swarm from libp2p.network.swarm import Swarm
from libp2p.peer.id import ID
from libp2p.peer.peerstore import PeerStore from libp2p.peer.peerstore import PeerStore
from libp2p.pubsub.floodsub import FloodSub from libp2p.pubsub.floodsub import FloodSub
from libp2p.pubsub.gossipsub import GossipSub from libp2p.pubsub.gossipsub import GossipSub
@ -33,6 +34,13 @@ from .constants import (
from .utils import connect, connect_swarm from .utils import connect, connect_swarm
def initialize_peerstore_with_our_keypair(self_id: ID, key_pair: KeyPair):
peer_store = PeerStore()
peer_store.add_pubkey(self_id, key_pair.public_key)
peer_store.add_privkey(self_id, key_pair.private_key)
return peer_store
def security_transport_factory( def security_transport_factory(
is_secure: bool, key_pair: KeyPair is_secure: bool, key_pair: KeyPair
) -> Dict[TProtocol, BaseSecureTransport]: ) -> Dict[TProtocol, BaseSecureTransport]:
@ -52,7 +60,9 @@ class SwarmFactory(factory.Factory):
muxer_opt = {MPLEX_PROTOCOL_ID: Mplex} muxer_opt = {MPLEX_PROTOCOL_ID: Mplex}
peer_id = factory.LazyAttribute(lambda o: generate_peer_id_from(o.key_pair)) peer_id = factory.LazyAttribute(lambda o: generate_peer_id_from(o.key_pair))
peerstore = factory.LazyFunction(PeerStore) peerstore = factory.LazyAttribute(
lambda o: initialize_peerstore_with_our_keypair(o.peer_id, o.key_pair)
)
upgrader = factory.LazyAttribute( upgrader = factory.LazyAttribute(
lambda o: TransportUpgrader( lambda o: TransportUpgrader(
security_transport_factory(o.is_secure, o.key_pair), o.muxer_opt security_transport_factory(o.is_secure, o.key_pair), o.muxer_opt
@ -97,7 +107,6 @@ class HostFactory(factory.Factory):
is_secure = False is_secure = False
key_pair = factory.LazyFunction(generate_new_rsa_identity) key_pair = factory.LazyFunction(generate_new_rsa_identity)
public_key = factory.LazyAttribute(lambda o: o.key_pair.public_key)
network = factory.LazyAttribute( network = factory.LazyAttribute(
lambda o: SwarmFactory(is_secure=o.is_secure, key_pair=o.key_pair) lambda o: SwarmFactory(is_secure=o.is_secure, key_pair=o.key_pair)
) )
@ -113,10 +122,7 @@ class HostFactory(factory.Factory):
for key_pair in key_pairs for key_pair in key_pairs
] ]
) )
return tuple( return tuple(BasicHost(swarm) for swarm in swarms)
BasicHost(key_pair.public_key, swarm)
for key_pair, swarm in zip(key_pairs, swarms)
)
class FloodsubFactory(factory.Factory): class FloodsubFactory(factory.Factory):