From 76b5affaaab8f9a63891c76f19156d2d03b64298 Mon Sep 17 00:00:00 2001
From: wangchuxiao <wangchuxiao97@outlook.com>
Date: Wed, 25 May 2022 17:46:02 +0800
Subject: [PATCH] script

---
 test/mongo/cmd/main.go    | 9 +++------
 test/mongo/mongo_utils.go | 8 +++++++-
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/test/mongo/cmd/main.go b/test/mongo/cmd/main.go
index 75ec8323f..8c8b2dbef 100644
--- a/test/mongo/cmd/main.go
+++ b/test/mongo/cmd/main.go
@@ -9,17 +9,14 @@ import (
 	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
-var (
-	client *mongo.Client
-)
-
 func init() {
 	clientOptions := options.Client().ApplyURI("mongodb://127.0.0.1:37017/openIM/?maxPoolSize=100")
-	client, err := mongo.Connect(context.TODO(), clientOptions)
+	var err error
+	mongo2.Client, err = mongo.Connect(context.TODO(), clientOptions)
 	if err != nil {
 		panic(err)
 	}
-	err = client.Ping(context.TODO(), nil)
+	err = mongo2.Client.Ping(context.TODO(), nil)
 	if err != nil {
 		panic(err)
 	}
diff --git a/test/mongo/mongo_utils.go b/test/mongo/mongo_utils.go
index e2c14ccdd..6f211cd69 100644
--- a/test/mongo/mongo_utils.go
+++ b/test/mongo/mongo_utils.go
@@ -4,15 +4,21 @@ import (
 	"Open_IM/pkg/common/config"
 	"Open_IM/pkg/common/db"
 	server_api_params "Open_IM/pkg/proto/sdk_ws"
+	"Open_IM/test/mongo/cmd"
 	"context"
 	"fmt"
 	"github.com/golang/protobuf/proto"
 	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
 	"gopkg.in/mgo.v2/bson"
 )
 
+var (
+	Client *mongo.Client
+)
+
 func GetUserAllChat(uid string) {
-	collection := client.Database(config.Config.Mongo.DBDatabase).Collection("msg")
+	collection := Client.Database(config.Config.Mongo.DBDatabase).Collection("msg")
 	var userChatList []db.UserChat
 	result, err := collection.Find(context.Background(), bson.M{"uid": primitive.Regex{Pattern: uid}})
 	if err != nil {