From b3229c54a25fd114758c3a988f57fa6c0c9c6830 Mon Sep 17 00:00:00 2001
From: withchao <993506633@qq.com>
Date: Wed, 28 Jun 2023 16:06:47 +0800
Subject: [PATCH] zk conn

---
 internal/rpc/auth/auth.go                   | 2 +-
 pkg/discoveryregistry/zookeeper/discover.go | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/internal/rpc/auth/auth.go b/internal/rpc/auth/auth.go
index c4e4877bf..d0a6176f4 100644
--- a/internal/rpc/auth/auth.go
+++ b/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()
diff --git a/pkg/discoveryregistry/zookeeper/discover.go b/pkg/discoveryregistry/zookeeper/discover.go
index 8a7c114fa..6c47c1674 100644
--- a/pkg/discoveryregistry/zookeeper/discover.go
+++ b/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()
+	}
 }