mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
e1137e5801
* start sequence for node to connect to bootstrap nodes * lint * maybe warning is more appropiate? * fix gazelle * preston's feedback * add go-ipfs-addr to workspace * move flag to shared * removed routed host
71 lines
1.8 KiB
Go
71 lines
1.8 KiB
Go
package p2p
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
iaddr "github.com/ipfs/go-ipfs-addr"
|
|
"github.com/libp2p/go-libp2p-host"
|
|
ps "github.com/libp2p/go-libp2p-peerstore"
|
|
mdns "github.com/libp2p/go-libp2p/p2p/discovery"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var log = logrus.WithField("prefix", "p2p")
|
|
|
|
// Discovery interval for multicast DNS querying.
|
|
var discoveryInterval = 1 * time.Minute
|
|
|
|
// mDNSTag is the name of the mDNS service.
|
|
var mDNSTag = mdns.ServiceTag
|
|
|
|
// TODO(287): add other discovery protocols such as DHT, etc.
|
|
// startmDNSDiscovery supports discovery via multicast DNS peer discovery.
|
|
func startmDNSDiscovery(ctx context.Context, host host.Host) error {
|
|
mdnsService, err := mdns.NewMdnsService(ctx, host, discoveryInterval, mDNSTag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mdnsService.RegisterNotifee(&discovery{ctx, host})
|
|
return nil
|
|
}
|
|
|
|
// startDHTDiscovery supports discovery via DHT.
|
|
func startDHTDiscovery(ctx context.Context, host host.Host, bootstrapAddr string) error {
|
|
addr, err := iaddr.ParseString(bootstrapAddr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
peerinfo, err := ps.InfoFromP2pAddr(addr.Multiaddr())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = host.Connect(ctx, *peerinfo)
|
|
return err
|
|
}
|
|
|
|
// Discovery implements mDNS notifee interface.
|
|
type discovery struct {
|
|
ctx context.Context
|
|
host host.Host
|
|
}
|
|
|
|
// HandlePeerFound registers the peer with the host.
|
|
func (d *discovery) HandlePeerFound(pi ps.PeerInfo) {
|
|
log.WithFields(logrus.Fields{
|
|
"peer addrs": pi.Addrs,
|
|
"peer id": pi.ID,
|
|
}).Debug("Attempting to connect to a peer")
|
|
|
|
d.host.Peerstore().AddAddrs(pi.ID, pi.Addrs, ps.PermanentAddrTTL)
|
|
if err := d.host.Connect(d.ctx, pi); err != nil {
|
|
log.Warnf("Failed to connect to peer: %v", err)
|
|
}
|
|
|
|
log.WithFields(logrus.Fields{
|
|
"peers": d.host.Peerstore().Peers(),
|
|
}).Debug("Peers are now")
|
|
}
|