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

removed redundant function and added try catch block

Browse Source
This commit is contained in:
sumanjeet0012@gmail.com
2025-06-24 14:15:42 +05:30
parent 9adf9aa499
commit 28d0e5759a
2 changed files with 29 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
libp2p/discovery/mdns/broadcaster.py
View File

@ -2,6 +2,7 @@ import logging
import socket
from zeroconf import (
EventLoopBlocked,
ServiceInfo,
Zeroconf,
)
@ -57,8 +58,34 @@ class PeerBroadcaster:
def register(self) -> None:
"""Register the peer's mDNS service on the network."""
self.zeroconf.register_service(self.service_info)
try:
self.zeroconf.register_service(self.service_info)
logger.debug("mDNS service registered: %s", self.service_name)
except EventLoopBlocked as e:
logger.warning(
"EventLoopBlocked while registering mDNS '%s': %s", self.service_name, e
)
except Exception as e:
logger.error(
"Unexpected error during mDNS registration for '%s': %r",
self.service_name,
e,
)
def unregister(self) -> None:
"""Unregister the peer's mDNS service from the network."""
self.zeroconf.unregister_service(self.service_info)
try:
self.zeroconf.unregister_service(self.service_info)
logger.debug("mDNS service unregistered: %s", self.service_name)
except EventLoopBlocked as e:
logger.warning(
"EventLoopBlocked while unregistering mDNS '%s': %s",
self.service_name,
e,
)
except Exception as e:
logger.error(
"Unexpected error during mDNS unregistration for '%s': %r",
self.service_name,
e,
)

11
libp2p/discovery/mdns/listener.py
View File

@ -61,17 +61,6 @@ class PeerListener(ServiceListener):
self.peerstore.add_addrs(peer_info.peer_id, peer_info.addrs, 10)
logger.debug("Updated Peer", peer_info.peer_id)
def _process_discovered_service(
self, zeroconf: Zeroconf, type_: str, name: str
) -> None:
info = zeroconf.get_service_info(type_, name, timeout=5000)
if not info:
return
peer_info = self._extract_peer_info(info)
if peer_info:
self.peerstore.add_addrs(peer_info.peer_id, peer_info.addrs, 10)
logger.debug("Discovered:", peer_info.peer_id)
def _extract_peer_info(self, info: ServiceInfo) -> PeerInfo | None:
try:
addrs = [