From dbeeb6d2b1ee224ee80c5b5890c9e666701bee08 Mon Sep 17 00:00:00 2001
From: withchao <993506633@qq.com>
Date: Wed, 22 Nov 2023 14:42:02 +0800
Subject: [PATCH] friend

---
 internal/rpc/friend/friend.go | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/internal/rpc/friend/friend.go b/internal/rpc/friend/friend.go
index c6caaf708..063015abc 100644
--- a/internal/rpc/friend/friend.go
+++ b/internal/rpc/friend/friend.go
@@ -16,6 +16,7 @@ package friend
 
 import (
 	"context"
+	tx2 "github.com/openimsdk/open-im-server/v3/pkg/common/db/tx"
 
 	"github.com/OpenIMSDK/protocol/sdkws"
 
@@ -32,7 +33,6 @@ import (
 	pbfriend "github.com/OpenIMSDK/protocol/friend"
 	registry "github.com/OpenIMSDK/tools/discoveryregistry"
 	"github.com/OpenIMSDK/tools/errs"
-	"github.com/OpenIMSDK/tools/tx"
 	"github.com/OpenIMSDK/tools/utils"
 
 	"github.com/openimsdk/open-im-server/v3/pkg/common/db/cache"
@@ -89,14 +89,17 @@ func Start(client registry.SvcDiscoveryRegistry, server *grpc.Server) error {
 		&msgRpcClient,
 		notification.WithRpcFunc(userRpcClient.GetUsersInfo),
 	)
-
+	tx, err := tx2.NewAuto(context.Background(), mongo.GetClient())
+	if err != nil {
+		return err
+	}
 	// Register Friend server with refactored MongoDB and Redis integrations
 	pbfriend.RegisterFriendServer(server, &friendServer{
 		friendDatabase: controller.NewFriendDatabase(
 			friendMongoDB,
 			friendRequestMongoDB,
 			cache.NewFriendCacheRedis(rdb, friendMongoDB, cache.GetDefaultOpt()),
-			tx.NewMongo(mongo.GetClient()),
+			tx,
 		),
 		blackDatabase: controller.NewBlackDatabase(
 			blackMongoDB,