From b17037f8d55c2151eb522f282949c7245ac162e0 Mon Sep 17 00:00:00 2001
From: Gordon <1432970085@qq.com>
Date: Fri, 26 May 2023 10:20:10 +0800
Subject: [PATCH] fix: user logout recycle resources

---
 internal/msggateway/hub_server.go | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/internal/msggateway/hub_server.go b/internal/msggateway/hub_server.go
index 6f0660f36..e4124a5af 100644
--- a/internal/msggateway/hub_server.go
+++ b/internal/msggateway/hub_server.go
@@ -11,14 +11,13 @@ import (
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/msggateway"
-	"github.com/OpenIMSDK/Open-IM-Server/pkg/rpcclient"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/startrpc"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
 	"google.golang.org/grpc"
 )
 
 func (s *Server) InitServer(client discoveryregistry.SvcDiscoveryRegistry, server *grpc.Server) error {
-	s.LongConnServer.SetMessageHandler(rpcclient.NewMsgClient(client))
+	s.LongConnServer.SetDiscoveryRegistry(client)
 	msggateway.RegisterMsgGatewayServer(server, s)
 	return nil
 }