fix: 优化etcd重连逻辑

优化etcd客户端重新连接时无法复用lease ID问题
pull/242/head
Wang Guoqiang 3 years ago
parent 4f76a65078
commit e75611a74c

@ -4,7 +4,6 @@ import (
"Open_IM/pkg/common/log" "Open_IM/pkg/common/log"
"Open_IM/pkg/utils" "Open_IM/pkg/utils"
"context" "context"
"errors"
"fmt" "fmt"
"net" "net"
"strconv" "strconv"
@ -58,45 +57,32 @@ func RegisterEtcd(schema, etcdAddr, myHost string, myPort int, serviceName strin
serviceKey := GetPrefix(schema, serviceName) + serviceValue serviceKey := GetPrefix(schema, serviceName) + serviceValue
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
leaseErr := make(chan error, 1)
leaseKeepAlive := func() { go func() {
NEWLEASE: NEWLEASE:
//lease
resp, err := cli.Grant(ctx, int64(ttl)) resp, err := cli.Grant(ctx, int64(ttl))
if err != nil { if err != nil {
log.Error(operationID, "Grant failed ", err.Error(), ctx, ttl) log.Error(operationID, "Grant failed ", err.Error(), ctx, ttl)
leaseErr <- errors.New("grant failed") } else {
return if _, err := cli.Put(ctx, serviceKey, serviceValue, clientv3.WithLease(resp.ID)); err != nil {
} log.Error(operationID, "cli.Put failed ", err.Error(), ctx, args, resp.ID)
log.Info(operationID, "Grant ok, resp ID ", resp.ID) } else {
kresp, err := cli.KeepAlive(ctx, resp.ID)
//set key->value if err != nil {
if _, err := cli.Put(ctx, serviceKey, serviceValue, clientv3.WithLease(resp.ID)); err != nil { log.Error(operationID, "KeepAlive failed ", err.Error(), args, resp.ID)
log.Error(operationID, "cli.Put failed ", err.Error(), ctx, args, resp.ID) } else {
leaseErr <- fmt.Errorf("put failed, errmsg:%v, key:%s, value:%s", err, serviceKey, serviceValue) log.Info(operationID, "RegisterEtcd ok ", args)
return for resp := range kresp {
log.Debug(operationID, "KeepAlive kresp ok", resp, args)
}
}
}
} }
//keepalive
kresp, err := cli.KeepAlive(ctx, resp.ID)
if err != nil {
log.Error(operationID, "KeepAlive failed ", err.Error(), args, resp.ID)
leaseErr <- fmt.Errorf("keepalive failed, errmsg:%v, lease id:%d", err, resp.ID)
return
}
leaseErr <- nil
log.Info(operationID, "RegisterEtcd ok ", args)
for resp := range kresp {
log.Debug(operationID, "KeepAlive kresp ok", resp, args)
}
log.Error(operationID, "KeepAlive kresp failed ", args) log.Error(operationID, "KeepAlive kresp failed ", args)
time.Sleep(time.Duration(ttl/2) * time.Second) time.Sleep(time.Duration(ttl/2) * time.Second)
goto NEWLEASE goto NEWLEASE
} }()
go leaseKeepAlive()
if err := <-leaseErr; err != nil {
return err
}
rEtcd = &RegEtcd{ctx: ctx, rEtcd = &RegEtcd{ctx: ctx,
cli: cli, cli: cli,

Loading…
Cancel
Save