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 {