You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
164 lines
4.8 KiB
164 lines
4.8 KiB
// 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, "")
|
|
//}
|