61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
package qgrpc
|
|
|
|
import (
|
|
"context"
|
|
"github.com/coreos/etcd/clientv3"
|
|
"gitlab.com/unitechdev/golib/tlog"
|
|
"path"
|
|
"time"
|
|
)
|
|
|
|
func etcdGrantWrapper(cli *clientv3.Client) (clientv3.LeaseID, error) {
|
|
resp, err := cli.Grant(context.Background(), 5)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return resp.ID, nil
|
|
}
|
|
|
|
func (self *Server) register() (err error) {
|
|
|
|
if len(self.userCfg.Prefix) < 1 || len(self.userCfg.Addr) < 1 {
|
|
tlog.Info("qgrpc server||register||prefix or addr less than 1 char")
|
|
return nil
|
|
}
|
|
|
|
if self.etcdId, err = etcdGrantWrapper(self.etcdCli); err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, server := range self.userCfg.Servers {
|
|
addr := self.userCfg.GetAddr()
|
|
etcdKey := path.Join(self.userCfg.Prefix, server, addr)
|
|
if _, err = self.etcdCli.Put(context.Background(), etcdKey, addr, clientv3.WithLease(self.etcdId)); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
ch, err := self.etcdCli.KeepAlive(context.Background(), self.etcdId)
|
|
if err != nil {
|
|
tlog.Error(err)
|
|
}
|
|
|
|
go self.keepalive(ch)
|
|
|
|
return err
|
|
}
|
|
|
|
func (self *Server) keepalive(ch <-chan *clientv3.LeaseKeepAliveResponse) {
|
|
for {
|
|
resp := <-ch
|
|
if resp == nil {
|
|
tlog.Error("qgrpc server||keepalive||nil response")
|
|
self.Revoke()
|
|
time.Sleep(time.Second)
|
|
self.register()
|
|
return
|
|
}
|
|
}
|
|
}
|