mirror of
https://github.com/DeBrosOfficial/network.git
synced 2025-10-06 10:19:07 +00:00
anonpenguin
6301ed9182
The changes reorganize the network client code by splitting it into focused modules for better maintainability, including
39 lines
1.3 KiB
Go
39 lines
1.3 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
|
"github.com/libp2p/go-libp2p/p2p/discovery/mdns"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// startMDNSDiscovery enables mDNS peer discovery for local network
|
|
func (c *Client) startMDNSDiscovery() {
|
|
mdnsService := mdns.NewMdnsService(c.host, "anchat-p2p", &discoveryNotifee{ client: c, logger: c.logger })
|
|
if err := mdnsService.Start(); err != nil {
|
|
c.logger.Warn("Failed to start mDNS discovery", zap.Error(err))
|
|
return
|
|
}
|
|
c.logger.Info("Started mDNS discovery for Anchat")
|
|
}
|
|
|
|
// discoveryNotifee handles mDNS peer discovery notifications
|
|
type discoveryNotifee struct {
|
|
client *Client
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func (n *discoveryNotifee) HandlePeerFound(pi peer.AddrInfo) {
|
|
n.logger.Info("mDNS discovered Anchat peer", zap.String("peer", pi.ID.String()[:8]+"..."), zap.Int("addrs", len(pi.Addrs)))
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
if err := n.client.host.Connect(ctx, pi); err != nil {
|
|
n.logger.Debug("Failed to connect to mDNS discovered peer", zap.String("peer", pi.ID.String()[:8]+"..."), zap.Error(err))
|
|
} else {
|
|
n.logger.Info("Successfully connected to mDNS discovered peer", zap.String("peer", pi.ID.String()[:8]+"..."))
|
|
if n.client.libp2pPS != nil { _ = n.client.libp2pPS.ListPeers("") }
|
|
}
|
|
}
|