Hulk did something

[bytom/vapor.git] / p2p / discover / mdns / mdns.go

diff --git a/p2p/discover/mdns/mdns.go b/p2p/discover/mdns/mdns.go
new file mode 100644 (file)
index 0000000..c061be9
--- /dev/null
+++ b/p2p/discover/mdns/mdns.go
@@ -0,0 +1,66 @@
+package mdns
+
+import (
+       "context"
+
+       "github.com/grandcat/zeroconf"
+)
+
+const (
+       instanceName = "bytomd"
+       serviceName  = "lanDiscover"
+       domainName   = "local"
+)
+
+// Protocol decoration ZeroConf,which is a pure Golang library
+// that employs Multicast DNS-SD.
+type Protocol struct {
+       entries chan *zeroconf.ServiceEntry
+       server  *zeroconf.Server
+       quite   chan struct{}
+}
+
+// NewProtocol create a specific Protocol.
+func NewProtocol() *Protocol {
+       return &Protocol{
+               entries: make(chan *zeroconf.ServiceEntry),
+               quite:   make(chan struct{}),
+       }
+}
+
+func (m *Protocol) getLanPeerLoop(event chan LANPeerEvent) {
+       for {
+               select {
+               case entry := <-m.entries:
+                       event <- LANPeerEvent{IP: entry.AddrIPv4, Port: entry.Port}
+               case <-m.quite:
+                       return
+               }
+       }
+}
+
+func (m *Protocol) registerService(port int) error {
+       var err error
+       m.server, err = zeroconf.Register(instanceName, serviceName, domainName, port, nil, nil)
+       return err
+}
+
+func (m *Protocol) registerResolver(event chan LANPeerEvent) error {
+       go m.getLanPeerLoop(event)
+       resolver, err := zeroconf.NewResolver(nil)
+       if err != nil {
+               return err
+       }
+
+       return resolver.Browse(context.Background(), serviceName, domainName, m.entries)
+}
+
+func (m *Protocol) stopResolver() {
+       close(m.quite)
+}
+
+func (m *Protocol) stopService() {
+       if m.server != nil {
+               m.server.Shutdown()
+       }
+}