Open-IM-Server/pkg/common/db/unrelation/super_group.go

// Copyright © 2023 OpenIM. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package unrelation

//
//import (
//	"context"
//
//	"go.mongodb.org/mongo-driver/bson"
//	"go.mongodb.org/mongo-driver/mongo"
//	"go.mongodb.org/mongo-driver/mongo/options"
//
//	"github.com/OpenIMSDK/tools/utils"
//
//	"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/unrelation"
//)
//
//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, "")
//}