rpc

pkg/common/db/localcache/conversation.go

@@ -7,14 +7,13 @@ import (
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/conversation"
-	"google.golang.org/grpc"
 )
 
 type ConversationLocalCache struct {
 	lock                              sync.Mutex
 	SuperGroupRecvMsgNotNotifyUserIDs map[string]Hash
 	ConversationIDs                   map[string]Hash
-	conn                              *grpc.ClientConn
+	client                            discoveryregistry.SvcDiscoveryRegistry
 }
 
 type Hash struct {
@@ -23,19 +22,19 @@ type Hash struct {
 }
 
 func NewConversationLocalCache(client discoveryregistry.SvcDiscoveryRegistry) *ConversationLocalCache {
-	conn, err := client.GetConn(context.Background(), config.Config.RpcRegisterName.OpenImConversationName)
-	if err != nil {
-		panic(err)
-	}
 	return &ConversationLocalCache{
 		SuperGroupRecvMsgNotNotifyUserIDs: make(map[string]Hash),
 		ConversationIDs:                   make(map[string]Hash),
-		conn:                              conn,
+		client:                            client,
 	}
 }
 
 func (g *ConversationLocalCache) GetRecvMsgNotNotifyUserIDs(ctx context.Context, groupID string) ([]string, error) {
-	client := conversation.NewConversationClient(g.conn)
+	conn, err := g.client.GetConn(ctx, config.Config.RpcRegisterName.OpenImConversationName)
+	if err != nil {
+		return nil, err
+	}
+	client := conversation.NewConversationClient(conn)
 	resp, err := client.GetRecvMsgNotNotifyUserIDs(ctx, &conversation.GetRecvMsgNotNotifyUserIDsReq{
 		GroupID: groupID,
 	})
@@ -46,7 +45,11 @@ func (g *ConversationLocalCache) GetRecvMsgNotNotifyUserIDs(ctx context.Context,
 }
 
 func (g *ConversationLocalCache) GetConversationIDs(ctx context.Context, userID string) ([]string, error) {
-	client := conversation.NewConversationClient(g.conn)
+	conn, err := g.client.GetConn(ctx, config.Config.RpcRegisterName.OpenImConversationName)
+	if err != nil {
+		return nil, err
+	}
+	client := conversation.NewConversationClient(conn)
 	resp, err := client.GetUserConversationIDsHash(ctx, &conversation.GetUserConversationIDsHashReq{
 		OwnerUserID: userID,
 	})