varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor bootstrap discovery to use async methods and update bootstrap peers list

Browse Source
This commit is contained in:
sumanjeet0012@gmail.com
2025-07-10 19:24:09 +05:30
parent 198208aef3
commit 2dfee68f20
4 changed files with 28 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
libp2p/discovery/bootstrap/bootstrap.py
View File

@ -22,7 +22,7 @@ class BootstrapDiscovery:
self.bootstrap_addrs = bootstrap_addrs or []
self.discovered_peers: set[str] = set()
def start(self) -> None:
async def start(self) -> None:
"""Process bootstrap addresses and emit peer discovery events."""
logger.debug(
f"Starting bootstrap discovery with "
@ -34,7 +34,7 @@ class BootstrapDiscovery:
for addr_str in self.bootstrap_addrs:
try:
self._process_bootstrap_addr(addr_str)
await self._process_bootstrap_addr(addr_str)
except Exception as e:
logger.debug(f"Failed to process bootstrap address {addr_str}: {e}")
@ -43,19 +43,33 @@ class BootstrapDiscovery:
logger.debug("Stopping bootstrap discovery")
self.discovered_peers.clear()
def _process_bootstrap_addr(self, addr_str: str) -> None:
async def _process_bootstrap_addr(self, addr_str: str) -> None:
"""Convert string address to PeerInfo and add to peerstore."""
try:
multiaddr = Multiaddr(addr_str)
except Exception as e:
logger.debug(f"Invalid multiaddr format '{addr_str}': {e}")
return
if (self.is_dns_addr(multiaddr)):
resolved_addrs = await multiaddr.resolve()
for resolved_addr in resolved_addrs:
if resolved_addr == multiaddr:
return
self.add_addr(Multiaddr(resolved_addr))
self.add_addr(multiaddr)
def is_dns_addr(self, addr: Multiaddr) -> bool:
"""Check if the address is a DNS address."""
return any(protocol.name == "dnsaddr" for protocol in addr.protocols())
def add_addr(self, addr: Multiaddr) -> None:
"""Add a peer to the peerstore and emit discovery event."""
# Extract peer info from multiaddr
try:
peer_info = info_from_p2p_addr(multiaddr)
peer_info = info_from_p2p_addr(addr)
except Exception as e:
logger.debug(f"Failed to extract peer info from '{addr_str}': {e}")
logger.debug(f"Failed to extract peer info from '{addr}': {e}")
return
# Skip if it's our own peer
@ -75,4 +89,4 @@ class BootstrapDiscovery:
self.discovered_peers.add(str(peer_info.peer_id))
# Emit peer discovery event
peerDiscovery.emit_peer_discovered(peer_info)
peerDiscovery.emit_peer_discovered(peer_info)