feat: use robot to migrate code

Signed-off-by: kubbot & kubecub <3293172751ysy@gmail.com>

pkg/common/db/unrelation/mongo.go

@@ -0,0 +1,114 @@
+package unrelation
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"time"
+
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/table/unrelation"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/mw/specialerror"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"go.mongodb.org/mongo-driver/x/bsonx"
+)
+
+type Mongo struct {
+	db *mongo.Client
+}
+
+func NewMongo() (*Mongo, error) {
+	specialerror.AddReplace(mongo.ErrNoDocuments, errs.ErrRecordNotFound)
+	uri := "mongodb://sample.host:27017/?maxPoolSize=20&w=majority"
+	if config.Config.Mongo.Uri != "" {
+		// example: mongodb://$user:$password@mongo1.mongo:27017,mongo2.mongo:27017,mongo3.mongo:27017/$DBDatabase/?replicaSet=rs0&readPreference=secondary&authSource=admin&maxPoolSize=$DBMaxPoolSize
+		uri = config.Config.Mongo.Uri
+	} else {
+		//mongodb://mongodb1.example.com:27317,mongodb2.example.com:27017/?replicaSet=mySet&authSource=authDB
+		mongodbHosts := ""
+		for i, v := range config.Config.Mongo.Address {
+			if i == len(config.Config.Mongo.Address)-1 {
+				mongodbHosts += v
+			} else {
+				mongodbHosts += v + ","
+			}
+		}
+		if config.Config.Mongo.Password != "" && config.Config.Mongo.Username != "" {
+			uri = fmt.Sprintf("mongodb://%s:%s@%s/%s?maxPoolSize=%d&authSource=admin",
+				config.Config.Mongo.Username, config.Config.Mongo.Password, mongodbHosts,
+				config.Config.Mongo.Database, config.Config.Mongo.MaxPoolSize)
+		} else {
+			uri = fmt.Sprintf("mongodb://%s/%s/?maxPoolSize=%d&authSource=admin",
+				mongodbHosts, config.Config.Mongo.Database,
+				config.Config.Mongo.MaxPoolSize)
+		}
+	}
+	fmt.Println("mongo:", uri)
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second*60)
+	defer cancel()
+	mongoClient, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
+	if err != nil {
+		return nil, err
+	}
+	return &Mongo{db: mongoClient}, nil
+}
+
+func (m *Mongo) GetClient() *mongo.Client {
+	return m.db
+}
+
+func (m *Mongo) GetDatabase() *mongo.Database {
+	return m.db.Database(config.Config.Mongo.Database)
+}
+
+func (m *Mongo) CreateMsgIndex() error {
+	return m.createMongoIndex(unrelation.Msg, true, "doc_id")
+}
+
+func (m *Mongo) CreateSuperGroupIndex() error {
+	if err := m.createMongoIndex(unrelation.CSuperGroup, true, "group_id"); err != nil {
+		return err
+	}
+	if err := m.createMongoIndex(unrelation.CUserToSuperGroup, true, "user_id"); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (m *Mongo) CreateExtendMsgSetIndex() error {
+	return m.createMongoIndex(unrelation.CExtendMsgSet, true, "-create_time", "work_moment_id")
+}
+
+func (m *Mongo) createMongoIndex(collection string, isUnique bool, keys ...string) error {
+	db := m.db.Database(config.Config.Mongo.Database).Collection(collection)
+	opts := options.CreateIndexes().SetMaxTime(10 * time.Second)
+	indexView := db.Indexes()
+	keysDoc := bsonx.Doc{}
+	// create composite indexes
+	for _, key := range keys {
+		if strings.HasPrefix(key, "-") {
+			keysDoc = keysDoc.Append(strings.TrimLeft(key, "-"), bsonx.Int32(-1))
+		} else {
+			keysDoc = keysDoc.Append(key, bsonx.Int32(1))
+		}
+	}
+	// create index
+	index := mongo.IndexModel{
+		Keys: keysDoc,
+	}
+	if isUnique {
+		index.Options = options.Index().SetUnique(true)
+	}
+	result, err := indexView.CreateOne(
+		context.Background(),
+		index,
+		opts,
+	)
+	if err != nil {
+		return utils.Wrap(err, result)
+	}
+	return nil
+}