MongoDB中使用findOneAndUpdate条件更新/插入内嵌数组

集合结构 #

[
  {
    "postId": ObjectId("62dffd0acb17483cf015375f"),
    "userId": ObjectId("62dff9584f5b702d61c81c3c"),
    "state": [
      {
        "id": ObjectId("62dffc49cb17483cf0153220"),
        "notes": "these are my custom notes!",
        "lvl": 3
      },
      {
        "id": ObjectId("62dffc49cb17483cf0153221"),
        "notes": "hello again",
        "lvl": 0
      }
    ]
  }
]

需求场景 #

• 若新元素的id不在state数组中，则将其推入数组；
• 若新元素的id已存在且对应lvl字段为0，则用新信息更新该元素；
• 若新元素的id已存在但lvl字段不为0，则不执行任何操作，通过未匹配文档抛出错误。

当前尝试与疑问 #

原本计划用带upsert的findOneAndUpdate实现，已完成场景1的查询，但不确定如何在匹配到元素时根据lvl值决定更新逻辑。目前已通过多查询实现：先获取文档，遍历state数组检查id是否存在，再通过分支逻辑执行对应操作，想确认是否有更优的单查询实现方式，是否必须拆分多查询。

解决方案 #

方式一：单原子查询实现（推荐） #

可以通过updateOne结合数组过滤器、条件操作符实现原子性的条件更新/插入，同时覆盖所有场景：

const targetId = ObjectId("62dffc49cb17483cf0153222");
const newStateItem = {
  id: targetId,
  notes: "新的备注内容",
  lvl: 1
};

const result = db.collection.updateOne(
  {
    "postId": ObjectId("62dffd0acb17483cf015375f"),
    "userId": ObjectId("62dff9584f5b702d61c81c3c"),
    $or: [
      // 场景1：目标id不存在于state数组
      { "state.id": { $ne: targetId } },
      // 场景2：目标id存在且对应lvl为0
      { "state": { $elemMatch: { id: targetId, lvl: 0 } } }
    ]
  },
  {
    // 处理场景2：匹配到id存在且lvl为0时更新该元素
    $set: {
      "state.$[elem]": newStateItem
    },
    // 处理场景1：id不存在时推入新元素，否则不执行
    $push: {
      state: {
        $cond: [
          { $not: { $in: [targetId, "$state.id"] } },
          newStateItem,
          "$$REMOVE"
        ]
      }
    }
  },
  {
    arrayFilters: [ { "elem.id": targetId } ],
    upsert: true, // 若整个文档不存在则插入（根据需求可选）
    returnDocument: "after"
  }
);

// 处理场景3：若匹配数为0，说明id存在且lvl不为0，抛出错误
if (result.matchedCount === 0) {
  throw new Error("目标元素已存在且lvl不为0，禁止更新");
}

逻辑说明

• 查询条件通过$or覆盖场景1和场景2的匹配规则；
• 数组过滤器arrayFilters定位需要更新的数组元素，配合$set处理场景2的更新；
• $push结合$cond判断：仅当目标id不存在时执行推入，否则用$$REMOVE跳过该操作；
• 执行后通过matchedCount判断是否命中场景3，若为0则抛出错误。

这种方式是原子操作，避免了多查询可能带来的竞态条件（比如多个请求同时修改同一文档时的冲突）。

方式二：多查询实现（已采用） #

你当前使用的多查询方式逻辑可行，但存在竞态风险：

1. 查询文档并遍历state数组检查目标id的存在性及对应lvl值；
2. 根据检查结果执行$push或$set更新，或抛出错误。

如果业务场景中并发修改的概率极低，这种方式可以满足需求，但在高并发场景下，建议优先使用单原子查询方案。

内容的提问来源于stack exchange，提问作者lpares12