仅在满足特定条件时才向 mongoDB 文档数组添加新元素

Question

我有这个 mongoDB 文档结构：

Game = {
  _id: 'randomObjectId',
  players: [Array of players ObjectIds],
  maxPlayers: 2,
  status: 'created' (or 'fully-booked')
}

当一个新玩家加入游戏时，我想将他添加到玩家数组中，并且只有在满足此条件时才将游戏status更改为fully-booked ： players.length < maxPlayers 。

回顾一下：

• 如果players.length < maxPlayers ，检查数据库内部（使用查询）；
• 如果是，更新文档
• 如果没有发回错误响应（可选）

如果没有此检查，很容易使用以下方法进行查询：

 Game.findByIdAndUpdate('randomObjectId', { $addToSet: { players: 'playerObjId' }, gameStatus: 'fully-booked' }, { new: true })

我不知道如何将此条件添加到此（或其他类型的）查询中。 任何帮助表示赞赏。 谢谢

Answer 1

您可以尝试如下聚合管道更新。 但是，如果已经订满，该命令将返回null 。 另请注意，如果您尝试添加数组中已存在的玩家 ID，则不会添加。

db.collection.findOneAndUpdate(
    {
        _id: someObjectId,
        $expr: {
            $lt: [{ $size: "$players" }, "$maxPlayers"]
        }
    },
    [
        {
            $set: {
                players: { $setUnion: ["$players", [newPlayerObjectId]] }
            }
        },
        {
            $set: {
                status: {
                    $cond: {
                        if: { $lt: [{ $size: "$players" }, "$maxPlayers"] },
                        then: "created",
                        else: "fully-booked"
                    }
                }
            }
        }
    ],
    {
        returnNewDocument: true
    })