|
|
@ -986,11 +986,13 @@ func (d *DataBases) CreateSuperGroup(groupID string, initMemberIDList []string,
|
|
|
|
Upsert: &upsert,
|
|
|
|
Upsert: &upsert,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
c = d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cUserToSuperGroup)
|
|
|
|
c = d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cUserToSuperGroup)
|
|
|
|
_, err = c.UpdateMany(sCtx, bson.M{"user_id": bson.M{"$in": initMemberIDList}}, bson.M{"$addToSet": bson.M{"group_id_list": groupID}}, opts)
|
|
|
|
for _, userID := range initMemberIDList {
|
|
|
|
|
|
|
|
_, err = c.UpdateOne(sCtx, bson.M{"user_id": userID}, bson.M{"$addToSet": bson.M{"group_id_list": groupID}, "user_id": userID}, opts)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
session.AbortTransaction(ctx)
|
|
|
|
session.AbortTransaction(ctx)
|
|
|
|
return utils.Wrap(err, "transaction failed")
|
|
|
|
return utils.Wrap(err, "transaction failed")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
session.CommitTransaction(ctx)
|
|
|
|
session.CommitTransaction(ctx)
|
|
|
|
return err
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
@ -1031,11 +1033,13 @@ func (d *DataBases) AddUserToSuperGroup(groupID string, userIDList []string) err
|
|
|
|
opts := &options.UpdateOptions{
|
|
|
|
opts := &options.UpdateOptions{
|
|
|
|
Upsert: &upsert,
|
|
|
|
Upsert: &upsert,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
_, err = c.UpdateMany(sCtx, bson.M{"user_id": bson.M{"$in": userIDList}}, bson.M{"$addToSet": bson.M{"group_id_list": groupID}}, opts)
|
|
|
|
for _, userID := range userIDList {
|
|
|
|
|
|
|
|
_, err = c.UpdateOne(sCtx, bson.M{"user_id": userID}, bson.M{"$addToSet": bson.M{"group_id_list": groupID}, "user_id": userID}, opts)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
session.AbortTransaction(ctx)
|
|
|
|
session.AbortTransaction(ctx)
|
|
|
|
return utils.Wrap(err, "transaction failed")
|
|
|
|
return utils.Wrap(err, "transaction failed")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
session.CommitTransaction(ctx)
|
|
|
|
session.CommitTransaction(ctx)
|
|
|
|
return err
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
@ -1102,7 +1106,7 @@ func (d *DataBases) RemoveGroupFromUser(ctx, sCtx context.Context, groupID strin
|
|
|
|
})
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cUserToSuperGroup)
|
|
|
|
c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cUserToSuperGroup)
|
|
|
|
_, err := c.UpdateOne(sCtx, bson.M{"user_id": groupID}, bson.M{"$pull": bson.M{"group_id_list": groupID}})
|
|
|
|
_, err := c.UpdateOne(sCtx, bson.M{"user_id": userID}, bson.M{"$pull": groupID})
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
return utils.Wrap(err, "UpdateOne transaction failed")
|
|
|
|
return utils.Wrap(err, "UpdateOne transaction failed")
|
|
|
|
}
|
|
|
|
}
|
|
|
|