mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2025-12-31 20:36:24 +00:00
Added bootstrap module
This commit is contained in:
69
libp2p/discovery/bootstrap/bootstrap.py
Normal file
69
libp2p/discovery/bootstrap/bootstrap.py
Normal file
@ -0,0 +1,69 @@
|
||||
import logging
|
||||
|
||||
from multiaddr import Multiaddr
|
||||
|
||||
from libp2p.abc import INetworkService
|
||||
from libp2p.discovery.events.peerDiscovery import peerDiscovery
|
||||
from libp2p.peer.peerinfo import info_from_p2p_addr
|
||||
|
||||
logger = logging.getLogger("libp2p.discovery.bootstrap")
|
||||
|
||||
|
||||
class BootstrapDiscovery:
|
||||
"""
|
||||
Bootstrap-based peer discovery for py-libp2p.
|
||||
Connects to predefined bootstrap peers and adds them to peerstore.
|
||||
"""
|
||||
|
||||
def __init__(self, swarm: INetworkService, bootstrap_addrs: list[str]):
|
||||
self.swarm = swarm
|
||||
self.peerstore = swarm.peerstore
|
||||
self.bootstrap_addrs = bootstrap_addrs or []
|
||||
self.discovered_peers: set[str] = set()
|
||||
|
||||
def start(self) -> None:
|
||||
"""Process bootstrap addresses and emit peer discovery events."""
|
||||
logger.debug(
|
||||
f"Starting bootstrap discovery with "
|
||||
f"{len(self.bootstrap_addrs)} bootstrap addresses"
|
||||
)
|
||||
|
||||
for addr_str in self.bootstrap_addrs:
|
||||
try:
|
||||
self._process_bootstrap_addr(addr_str)
|
||||
except Exception as e:
|
||||
logger.warning(f"Failed to process bootstrap address {addr_str}: {e}")
|
||||
|
||||
def stop(self) -> None:
|
||||
"""Clean up bootstrap discovery resources."""
|
||||
logger.debug("Stopping bootstrap discovery")
|
||||
self.discovered_peers.clear()
|
||||
|
||||
def _process_bootstrap_addr(self, addr_str: str) -> None:
|
||||
"""Convert string address to PeerInfo and add to peerstore."""
|
||||
# Convert string to Multiaddr
|
||||
multiaddr = Multiaddr(addr_str)
|
||||
|
||||
# Extract peer info from multiaddr
|
||||
peer_info = info_from_p2p_addr(multiaddr)
|
||||
|
||||
# Skip if it's our own peer
|
||||
if peer_info.peer_id == self.swarm.get_peer_id():
|
||||
logger.debug(f"Skipping own peer ID: {peer_info.peer_id}")
|
||||
return
|
||||
|
||||
# Skip if already discovered
|
||||
if str(peer_info.peer_id) in self.discovered_peers:
|
||||
logger.debug(f"Peer already discovered: {peer_info.peer_id}")
|
||||
return
|
||||
|
||||
# Add to peerstore with TTL (using same pattern as mDNS)
|
||||
self.peerstore.add_addrs(peer_info.peer_id, peer_info.addrs, 10)
|
||||
|
||||
# Track discovered peer
|
||||
self.discovered_peers.add(str(peer_info.peer_id))
|
||||
|
||||
# Emit peer discovery event
|
||||
peerDiscovery.emit_peer_discovered(peer_info)
|
||||
|
||||
logger.info(f"Discovered bootstrap peer: {peer_info.peer_id}")
|
||||
Reference in New Issue
Block a user