feat: use robot to migrate code

Signed-off-by: kubbot & kubecub <3293172751ysy@gmail.com>
2026-05-11 04:25:59 +08:00 · 2023-06-30 09:45:02 +08:00
parent 2d41819008
commit 539e0fdfb6
529 changed files with 64588 additions and 54413 deletions
@@ -0,0 +1,111 @@
+package unrelation
+
+import (
+	"context"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/table/unrelation"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+func NewSuperGroupMongoDriver(database *mongo.Database) unrelation.SuperGroupModelInterface {
+	return &SuperGroupMongoDriver{superGroupCollection: database.Collection(unrelation.CSuperGroup), userToSuperGroupCollection: database.Collection(unrelation.CUserToSuperGroup)}
+}
+
+type SuperGroupMongoDriver struct {
+	superGroupCollection       *mongo.Collection
+	userToSuperGroupCollection *mongo.Collection
+}
+
+func (s *SuperGroupMongoDriver) CreateSuperGroup(ctx context.Context, groupID string, initMemberIDs []string) error {
+	_, err := s.superGroupCollection.InsertOne(ctx, &unrelation.SuperGroupModel{
+		GroupID:   groupID,
+		MemberIDs: initMemberIDs,
+	})
+	if err != nil {
+		return err
+	}
+	for _, userID := range initMemberIDs {
+		_, err = s.userToSuperGroupCollection.UpdateOne(ctx, bson.M{"user_id": userID}, bson.M{"$addToSet": bson.M{"group_id_list": groupID}}, &options.UpdateOptions{
+			Upsert: utils.ToPtr(true),
+		})
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func (s *SuperGroupMongoDriver) TakeSuperGroup(ctx context.Context, groupID string) (group *unrelation.SuperGroupModel, err error) {
+	if err := s.superGroupCollection.FindOne(ctx, bson.M{"group_id": groupID}).Decode(&group); err != nil {
+		return nil, utils.Wrap(err, "")
+	}
+	return group, nil
+}
+
+func (s *SuperGroupMongoDriver) FindSuperGroup(ctx context.Context, groupIDs []string) (groups []*unrelation.SuperGroupModel, err error) {
+	cursor, err := s.superGroupCollection.Find(ctx, bson.M{"group_id": bson.M{
+		"$in": groupIDs,
+	}})
+	if err != nil {
+		return nil, err
+	}
+	defer cursor.Close(ctx)
+	if err := cursor.All(ctx, &groups); err != nil {
+		return nil, utils.Wrap(err, "")
+	}
+	return groups, nil
+}
+
+func (s *SuperGroupMongoDriver) AddUserToSuperGroup(ctx context.Context, groupID string, userIDs []string) error {
+	_, err := s.superGroupCollection.UpdateOne(ctx, bson.M{"group_id": groupID}, bson.M{"$addToSet": bson.M{"member_id_list": bson.M{"$each": userIDs}}})
+	if err != nil {
+		return err
+	}
+	upsert := true
+	opts := &options.UpdateOptions{
+		Upsert: &upsert,
+	}
+	for _, userID := range userIDs {
+		_, err = s.userToSuperGroupCollection.UpdateOne(ctx, bson.M{"user_id": userID}, bson.M{"$addToSet": bson.M{"group_id_list": groupID}}, opts)
+		if err != nil {
+			return utils.Wrap(err, "transaction failed")
+		}
+	}
+	return nil
+}
+
+func (s *SuperGroupMongoDriver) RemoverUserFromSuperGroup(ctx context.Context, groupID string, userIDs []string) error {
+	_, err := s.superGroupCollection.UpdateOne(ctx, bson.M{"group_id": groupID}, bson.M{"$pull": bson.M{"member_id_list": bson.M{"$in": userIDs}}})
+	if err != nil {
+		return err
+	}
+	err = s.RemoveGroupFromUser(ctx, groupID, userIDs)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (s *SuperGroupMongoDriver) GetSuperGroupByUserID(ctx context.Context, userID string) (*unrelation.UserToSuperGroupModel, error) {
+	var user unrelation.UserToSuperGroupModel
+	err := s.userToSuperGroupCollection.FindOne(ctx, bson.M{"user_id": userID}).Decode(&user)
+	return &user, utils.Wrap(err, "")
+}
+
+func (s *SuperGroupMongoDriver) DeleteSuperGroup(ctx context.Context, groupID string) error {
+	group, err := s.TakeSuperGroup(ctx, groupID)
+	if err != nil {
+		return err
+	}
+	if _, err := s.superGroupCollection.DeleteOne(ctx, bson.M{"group_id": groupID}); err != nil {
+		return utils.Wrap(err, "")
+	}
+	return s.RemoveGroupFromUser(ctx, groupID, group.MemberIDs)
+}
+
+func (s *SuperGroupMongoDriver) RemoveGroupFromUser(ctx context.Context, groupID string, userIDs []string) error {
+	_, err := s.userToSuperGroupCollection.UpdateOne(ctx, bson.M{"user_id": bson.M{"$in": userIDs}}, bson.M{"$pull": bson.M{"group_id_list": groupID}})
+	return utils.Wrap(err, "")
+}