mirror of https://github.com/docker/docs.git
47 lines
994 B
Go
47 lines
994 B
Go
package main
|
|
|
|
import (
|
|
"regexp"
|
|
"time"
|
|
|
|
log "github.com/Sirupsen/logrus"
|
|
"github.com/codegangsta/cli"
|
|
"github.com/docker/swarm/discovery"
|
|
)
|
|
|
|
func checkAddrFormat(addr string) bool {
|
|
m, _ := regexp.MatchString("^[0-9a-zA-Z._-]+:[0-9]{1,5}$", addr)
|
|
return m
|
|
}
|
|
|
|
func join(c *cli.Context) {
|
|
dflag := getDiscovery(c)
|
|
if dflag == "" {
|
|
log.Fatalf("discovery required to join a cluster. See '%s join --help'.", c.App.Name)
|
|
}
|
|
|
|
d, err := discovery.New(dflag, c.Int("heartbeat"))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
addr := c.String("addr")
|
|
|
|
if !checkAddrFormat(addr) {
|
|
log.Fatal("--addr should be of the form ip:port or hostname:port")
|
|
}
|
|
|
|
if err := d.Register(addr); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
hb := time.Duration(c.Int("heartbeat"))
|
|
for {
|
|
log.WithFields(log.Fields{"addr": addr, "discovery": dflag}).Infof("Registering on the discovery service every %d seconds...", hb)
|
|
time.Sleep(hb * time.Second)
|
|
if err := d.Register(addr); err != nil {
|
|
log.Error(err)
|
|
}
|
|
}
|
|
}
|