diff --git a/pkg/common/db/model.go b/pkg/common/db/model.go index 7991451a3..32fe76f5c 100644 --- a/pkg/common/db/model.go +++ b/pkg/common/db/model.go @@ -2,7 +2,6 @@ package db import ( "Open_IM/pkg/common/config" - "go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/x/bsonx" "strings" @@ -70,84 +69,28 @@ func init() { } fmt.Println("0", utils.GetSelfFuncName(), "mongo driver client init success: ", uri) // mongodb create index - opts := options.CreateIndexes().SetMaxTime(10 * time.Second) - dataBase := mongoClient.Database(config.Config.Mongo.DBDatabase) - //cSendLogModels := []mongo.IndexModel{ - // { - // //Keys: bson.M{"send_id": -1, "send_time": -1}, - // Keys: bson.D{ - // {Key: "send_id"}, - // {Key: "send_time", Value: -1}, - // }, - // }, - //} - //result, err := dataBase.Collection(cSendLog).Indexes().CreateMany(context.Background(), cSendLogModels, opts) - //if err != nil { - // fmt.Println("mongodb create cSendLogModels failed", result, err.Error()) - //} - if err := createUniqueIndex(mongoClient, cSendLog, false, "send_id", "-send_time"); err != nil { - fmt.Println("send_id", "-send_time", "create failed", err.Error()) + if err := createMongoIndex(mongoClient, cSendLog, false, "send_id", "-send_time"); err != nil { + fmt.Println("send_id", "-send_time", "index create failed", err.Error()) } - var result string - cChatModels := []mongo.IndexModel{ - { - Keys: bson.M{"uid": -1}, - Options: options.Index().SetUnique(true), - }, + if err := createMongoIndex(mongoClient, cChat, true, "uid"); err != nil { + fmt.Println("uid", " index create failed", err.Error()) } - _, err = dataBase.Collection(cChat).Indexes().CreateMany(context.Background(), cChatModels, opts) - if err != nil { - fmt.Println("mongodb create cChatModels failed", result, err.Error()) + if err := createMongoIndex(mongoClient, cWorkMoment, true, "-create_time", "work_moment_id"); err != nil { + fmt.Println("-create_time", "work_moment_id", "index create failed", err.Error()) + } + if err := createMongoIndex(mongoClient, cWorkMoment, true, "work_moment_id"); err != nil { + fmt.Println("work_moment_id", "index create failed", err.Error()) } - _, err = dataBase.Collection(cWorkMoment).Indexes().CreateOne(context.Background(), mongo.IndexModel{ - Keys: bson.D{ - {Key: "create_time", Value: -1}, - {Key: "work_moment_id"}, - }, - }, opts) - if err != nil { - fmt.Println("mongodb create cWorkMoment index failed", result, err.Error()) + if err := createMongoIndex(mongoClient, cWorkMoment, false, "user_id", "-create_time"); err != nil { + fmt.Println("user_id", "-create_time", "index create failed", err.Error()) } - _, err = dataBase.Collection(cWorkMoment).Indexes().CreateOne(context.Background(), mongo.IndexModel{ - Keys: bson.D{ - {Key: "work_moment_id"}, - }, - Options: options.Index().SetUnique(true), - }, opts) - if err != nil { - fmt.Println("mongodb create cWorkMoment index failed", err.Error()) + if err := createMongoIndex(mongoClient, cTag, false, "user_id", "-create_time"); err != nil { + fmt.Println("user_id", "-create_time", "index create failed", err.Error()) } - - _, err = dataBase.Collection(cWorkMoment).Indexes().CreateOne(context.Background(), mongo.IndexModel{ - Keys: bson.D{ - {Key: "user_id"}, - {Key: "create_time", Value: -1}, - }, - }, opts) - if err != nil { - fmt.Println("mongodb create cWorkMoment index failed", err.Error()) - } - - _, err = dataBase.Collection(cTag).Indexes().CreateOne(context.Background(), mongo.IndexModel{ - Keys: bson.D{ - {Key: "user_id"}, - {Key: "create_time", Value: -1}, - }, - }, opts) - if err != nil { - fmt.Println("mongodb create cSendLog index failed", err.Error()) - } - - _, err = dataBase.Collection(cTag).Indexes().CreateOne(context.Background(), mongo.IndexModel{ - Keys: bson.D{ - {Key: "tag_id"}, - }, - Options: options.Index().SetUnique(true), - }) - if err != nil { - fmt.Println("mongodb create cSendLog index failed", result, err.Error()) + if err := createMongoIndex(mongoClient, cTag, true, "tag_id"); err != nil { + fmt.Println("user_id", "-create_time", "index create failed", err.Error()) } DB.mongoClient = mongoClient @@ -171,7 +114,7 @@ func init() { } } -func createUniqueIndex(client *mongo.Client, collection string, isUnique bool, keys ...string) error { +func createMongoIndex(client *mongo.Client, collection string, isUnique bool, keys ...string) error { db := client.Database(config.Config.Mongo.DBDatabase).Collection(collection) opts := options.CreateIndexes().SetMaxTime(10 * time.Second)