superGroup

3 years ago · edb5cb4ee5
parent 3b81108738
commit edb5cb4ee5
1 changed files with 5 additions and 1 deletions
--- a/pkg/common/db/mongoModel.go
+++ b/pkg/common/db/mongoModel.go
@ -981,7 +981,11 @@ func (d *DataBases) CreateSuperGroup(groupID string, initMemberIDList []string,
 			UserID: v,
 		})
 	}
-	_, err = c.UpdateOne(sCtx, users, bson.M{"$addToSet": bson.M{"group_id_list": groupID}})
+	upsert := true
+	opts := &options.UpdateOptions{
+		Upsert: &upsert,
+	}
+	_, err = c.UpdateMany(sCtx, bson.M{"user_id": bson.M{"$in": initMemberIDList}}, bson.M{"$addToSet": bson.M{"group_id_list": groupID}}, opts)
 	if err != nil {
 		session.AbortTransaction(ctx)
 		return utils.Wrap(err, "transaction failed")