script

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)
 	}

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 {