forked from libp2p/go-libp2p
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mdns.go
31 lines (25 loc) · 840 Bytes
/
mdns.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
package main
import (
"github.com/thusharprakash/go-libp2p/core/host"
"github.com/thusharprakash/go-libp2p/core/peer"
"github.com/thusharprakash/go-libp2p/p2p/discovery/mdns"
)
type discoveryNotifee struct {
PeerChan chan peer.AddrInfo
}
// interface to be called when new peer is found
func (n *discoveryNotifee) HandlePeerFound(pi peer.AddrInfo) {
n.PeerChan <- pi
}
// Initialize the MDNS service
func initMDNS(peerhost host.Host, rendezvous string) chan peer.AddrInfo {
// register with service so that we get notified about peer discovery
n := &discoveryNotifee{}
n.PeerChan = make(chan peer.AddrInfo)
// An hour might be a long long period in practical applications. But this is fine for us
ser := mdns.NewMdnsService(peerhost, rendezvous, n)
if err := ser.Start(); err != nil {
panic(err)
}
return n.PeerChan
}