Initial commit

qgrpc/register.go

@@ -0,0 +1,60 @@
+package qgrpc
+
+import (
+	"context"
+	"github.com/coreos/etcd/clientv3"
+	"globalfintech/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
+		}
+	}
+}