MongoDB单文档数组多值更新满足原子性与全有或全无要求的最优方法

问题1认知验证 #

你的认知是正确的。MongoDB所有单文档写操作本身就是原子性的，4.4及所有支持文档级锁的版本都遵循这个规则：单条更新操作执行期间会自动对目标文档加独占锁，不会有其他写操作能中途修改该文档，完全满足更新期间禁止其他操作修改文档的要求。

问题2最优实现方案 #

你的需求核心是只有所有预期要更新的数组元素全部匹配成功时才执行更新，否则整个操作完全不生效，可以通过「查询条件预校验+带过滤的数组更新运算符」组合的单条原子update实现，不需要依赖事务，也不存在并发修改风险，方案如下：

实现逻辑 #

1. 在更新的查询条件中新增校验规则：统计文档内符合要求的数组元素个数，只有数值等于你预期要更新的总数量时，才会命中查询进入更新逻辑
2. 使用带过滤条件的$[elem]数组更新运算符，批量修改所有匹配元素的qty值
   整个操作是单条原子请求，要么所有条件满足全量更新，要么完全不执行，不存在部分更新的情况。

代码示例 #

假设你需要给文档内someArray数组中id为"a"、"b"、"c"的三个元素的qty各加1，只有这三个元素都存在时才执行更新：

db.collection.updateOne(
  // 第一层查询条件：锁定目标文档，同时校验符合要求的数组元素数量是否符合预期
  {
    _id: 待更新的文档ID,
    $expr: {
      $eq: [
        {
          $size: {
            $filter: {
              input: "$someArray",
              cond: { $in: [ "$$this.id", ["a", "b", "c"] ] }
            }
          }
        },
        3 // 预期要匹配的数组元素总数量，可按需修改
      ]
    }
  },
  // 更新逻辑：所有匹配的元素qty加1
  {
    $inc: { "someArray.$[targetElem].qty": 1 }
  },
  // 数组元素过滤规则
  {
    arrayFilters: [ { "targetElem.id": { $in: ["a", "b", "c"] } } ]
  }
)

效果说明 #

• 如果数组中同时存在三个目标id的元素，查询条件命中，三个元素的qty会同时+1，操作原子生效
• 如果数组中缺少任意一个目标id，查询条件不命中，整个更新操作不会执行，不会修改任何数据，完全满足全部成功否则全部回滚的要求
• 整个操作是单条原子请求，天然规避并发修改风险，性能远高于事务方案

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