varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-12 16:10:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added tests for 'RoutedHost' and modified 'FindPeer'

Browse Source
This commit is contained in:
Aratz M. Lasa
2019-10-15 01:01:16 +02:00
parent 3f24b015ab
commit 8143563831
7 changed files with 105 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libp2p/host/routed_host.py
View File

@ -1,4 +1,5 @@
from libp2p.host.basic_host import BasicHost
from libp2p.host.exceptions import ConnectionFailure
from libp2p.network.network_interface import INetwork
from libp2p.peer.peerinfo import PeerInfo
from libp2p.routing.interfaces import IPeerRouting
@ -26,7 +27,10 @@ class RoutedHost(BasicHost):
"""
# check if we were given some addresses, otherwise, find some with the routing system.
if not peer_info.addrs:
peer_info.addrs = (await self._router.find_peer(peer_info.peer_id)).addrs
found_peer_info = await self._router.find_peer(peer_info.peer_id)
if not found_peer_info:
raise ConnectionFailure("Unable to find Peer address")
peer_info.addrs = found_peer_info.addrs
self.peerstore.add_addrs(peer_info.peer_id, peer_info.addrs, 10)
# there is already a connection to this peer