speed up refresh

This commit is contained in:
Victor Vieux 2014-11-21 19:11:03 +00:00
parent 1f74cb5809
commit f40ec59495
1 changed files with 18 additions and 18 deletions

View File

@ -24,22 +24,25 @@ func (h *logHandler) Handle(e *cluster.Event) error {
func manage(c *cli.Context) { func manage(c *cli.Context) {
refresh := func(c *cluster.Cluster, nodes []string) error { refresh := func(c *cluster.Cluster, nodes []string) {
for _, addr := range nodes { for _, addr := range nodes {
go func(addr string) {
if !strings.Contains(addr, "://") { if !strings.Contains(addr, "://") {
addr = "http://" + addr addr = "http://" + addr
} }
if c.Node(addr) == nil { if c.Node(addr) == nil {
n := cluster.NewNode(addr) n := cluster.NewNode(addr)
if err := n.Connect(nil); err != nil { if err := n.Connect(nil); err != nil {
return err log.Error(err)
return
} }
if err := c.AddNode(n); err != nil { if err := c.AddNode(n); err != nil {
return err log.Error(err)
return
} }
} }
}(addr)
} }
return nil
} }
cluster := cluster.NewCluster() cluster := cluster.NewCluster()
@ -52,9 +55,8 @@ func manage(c *cli.Context) {
log.Fatal(err) log.Fatal(err)
} }
if err := refresh(cluster, nodes); err != nil { refresh(cluster, nodes)
log.Fatal(err)
}
hb := time.Duration(c.Int("heartbeat")) hb := time.Duration(c.Int("heartbeat"))
go func() { go func() {
for { for {
@ -66,9 +68,7 @@ func manage(c *cli.Context) {
} }
}() }()
} else { } else {
if err := refresh(cluster, c.Args()); err != nil { refresh(cluster, c.Args())
log.Fatal(err)
}
} }
}() }()