mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2026-02-12 16:10:57 +00:00
refactored routedhost into router passed to swarm
This commit is contained in:
0
libp2p/routing/kademlia/__init__.py
Normal file
0
libp2p/routing/kademlia/__init__.py
Normal file
21
libp2p/routing/kademlia/kademlia_content_router.py
Normal file
21
libp2p/routing/kademlia/kademlia_content_router.py
Normal file
@ -0,0 +1,21 @@
|
||||
from libp2p.routing.interfaces import IContentRouting
|
||||
|
||||
|
||||
class KadmeliaContentRouter(IContentRouting):
|
||||
|
||||
def provide(self, cid, announce=True):
|
||||
"""
|
||||
Provide adds the given cid to the content routing system. If announce is True,
|
||||
it also announces it, otherwise it is just kept in the local
|
||||
accounting of which objects are being provided.
|
||||
"""
|
||||
# the DHT finds the closest peers to `key` using the `FIND_NODE` RPC
|
||||
# then sends a `ADD_PROVIDER` RPC with its own `PeerInfo` to each of these peers.
|
||||
pass
|
||||
|
||||
def find_provider_iter(self, cid, count):
|
||||
"""
|
||||
Search for peers who are able to provide a given key
|
||||
returns an iterator of peer.PeerInfo
|
||||
"""
|
||||
pass
|
||||
33
libp2p/routing/kademlia/kademlia_peer_router.py
Normal file
33
libp2p/routing/kademlia/kademlia_peer_router.py
Normal file
@ -0,0 +1,33 @@
|
||||
from libp2p.routing.interfaces import IPeerRouting
|
||||
from libp2p.kademlia.utils import digest
|
||||
from libp2p.peer.peerinfo import PeerInfo
|
||||
from libp2p.peer.peerdata import PeerData
|
||||
|
||||
|
||||
class KadmeliaPeerRouter(IPeerRouting):
|
||||
# pylint: disable=too-few-public-methods
|
||||
|
||||
def __init__(self, dht_server):
|
||||
self.server = dht_server
|
||||
|
||||
def find_peer(self, peer_id):
|
||||
"""
|
||||
Find specific Peer
|
||||
FindPeer searches for a peer with given peer_id, returns a peer.PeerInfo
|
||||
with relevant addresses.
|
||||
"""
|
||||
# switching peer_id to xor_id used by kademlia as node_id
|
||||
xor_id = peer_id.get_xor_id()
|
||||
value = self.server.get(xor_id)
|
||||
return decode_peerinfo(value)
|
||||
|
||||
|
||||
def decode_peerinfo(encoded):
|
||||
if isinstance(encoded, bytes):
|
||||
encoded = encoded.decode()
|
||||
lines = encoded.splitlines()
|
||||
peer_id = lines[0]
|
||||
addrs = lines[1:]
|
||||
peer_data = PeerData()
|
||||
peer_data.add_addrs(addrs)
|
||||
return PeerInfo(peer_id, addrs)
|
||||
Reference in New Issue
Block a user