zk conn

internal/rpc/auth/auth.go

@@ -109,7 +109,7 @@ func (s *authServer) forceKickOff(ctx context.Context, userID string, platformID
 		client := msggateway.NewMsgGatewayClient(v)
 		kickReq := &msggateway.KickUserOfflineReq{KickUserIDList: []string{userID}, PlatformID: platformID}
 		_, err := client.KickUserOffline(ctx, kickReq)
-		v.Close()
+		s.RegisterCenter.CloseConn(v)
 		return utils.Wrap(err, "")
 	}
 	return errs.ErrInternalServer.Wrap()

pkg/discoveryregistry/zookeeper/discover.go

@@ -3,6 +3,7 @@ package zookeeper
 import (
 	"context"
 	"fmt"
+	"io"
 	"strings"
 	"sync"
 
@@ -109,5 +110,7 @@ func (s *ZkClient) GetConn(ctx context.Context, serviceName string, opts ...grpc
 }
 
 func (s *ZkClient) CloseConn(conn grpc.ClientConnInterface) {
-	//conn.Close()
+	if closer, ok := conn.(io.Closer); ok {
+		closer.Close()
+	}
 }