Merge pull request #201 from Bytom/v0.1

[bytom/vapor.git] / vendor / github.com / grandcat / zeroconf / examples / register / server.go

diff --git a/vendor/github.com/grandcat/zeroconf/examples/register/server.go b/vendor/github.com/grandcat/zeroconf/examples/register/server.go
new file mode 100644 (file)
index 0000000..a2d15c5
--- /dev/null
+++ b/vendor/github.com/grandcat/zeroconf/examples/register/server.go
@@ -0,0 +1,54 @@
+package main
+
+import (
+       "flag"
+       "log"
+       "os"
+       "os/signal"
+       "syscall"
+
+       "time"
+
+       "github.com/grandcat/zeroconf"
+)
+
+var (
+       name     = flag.String("name", "GoZeroconfGo", "The name for the service.")
+       service  = flag.String("service", "_workstation._tcp", "Set the service type of the new service.")
+       domain   = flag.String("domain", "local.", "Set the network domain. Default should be fine.")
+       port     = flag.Int("port", 42424, "Set the port the service is listening to.")
+       waitTime = flag.Int("wait", 10, "Duration in [s] to publish service for.")
+)
+
+func main() {
+       flag.Parse()
+
+       server, err := zeroconf.Register(*name, *service, *domain, *port, []string{"txtv=0", "lo=1", "la=2"}, nil)
+       if err != nil {
+               panic(err)
+       }
+       defer server.Shutdown()
+       log.Println("Published service:")
+       log.Println("- Name:", *name)
+       log.Println("- Type:", *service)
+       log.Println("- Domain:", *domain)
+       log.Println("- Port:", *port)
+
+       // Clean exit.
+       sig := make(chan os.Signal, 1)
+       signal.Notify(sig, os.Interrupt, syscall.SIGTERM)
+       // Timeout timer.
+       var tc <-chan time.Time
+       if *waitTime > 0 {
+               tc = time.After(time.Second * time.Duration(*waitTime))
+       }
+
+       select {
+       case <-sig:
+               // Exit by user
+       case <-tc:
+               // Exit by timeout
+       }
+
+       log.Println("Shutting down.")
+}