func update(watcher client.Watcher, space *Namespace) {
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel()
res, err := watcher.Next(ctx)
if err != nil {
if err != context.DeadlineExceeded {
log.Errorf("watcher next failed, err[%v]", err)
}
return
}
node := res.Node
if node == nil {
log.Errorf("node is nil")
return
}
if node.Dir {
log.Infof("node is dir")
return
}
service, url, err := parsePath(node.Key, space.name)
if err != nil {
log.Errorf("parse path failed, err[%v]", err)
return
}
log.Infof("%s %s %s", res.Action, service, url)
switch res.Action {
case "set", "update":
space.updateEndpoint(service, url)
case "delete":
space.deleteEndpoint(service, url)
case "expire":
space.expireEndpoint(service, url)
}
}
func watchService(ctx *done.Context, api client.KeysAPI, space *Namespace) {
dir := watchDir(space.name)
opts := &client.WatcherOptions{Recursive: true}
watcher := api.Watcher(dir, opts)
log.Infof("watch %s start", dir)
for {
update(watcher, space)
select {
case <-ctx.Done():
log.Infof("watch %s done", dir)
ctx.OK()
return
default:
continue
}
}
}
func (p *pinger) offline() error {
_, err := p.api.Delete(context.Background(), p.path, &client.DeleteOptions{})
if err != nil {
log.Errorf("%s heartbeat offline failed, err[%v]", p.path, err)
return err
}
log.Infof("%s heartbeat offline success", p.path)
return nil
}
func TestLog(t *testing.T) {
//log.SetFlag(log.StdFlags | log.MsecFlag | log.ShortFileFlag | log.FuncNameFlag)
//log.SetFileOutput("./test.log", 1024, 2)
log.Fatalf("test %d %s", 1, "hello")
log.Errorf("test %d", 1)
log.Warnf("test %d", 1)
log.Infof("test %d", 1)
log.Debugf("test %d", 1)
}
func (p *pinger) online() error {
opts := &client.SetOptions{TTL: p.ttl}
_, err := p.api.Set(context.Background(), p.path, "online", opts)
if err != nil {
log.Errorf("%s heartbeat ping failed, err[%v]", p.path, err)
return err
}
log.Infof("%s heartbeat ping success", p.path)
return nil
}