MongoDB如何基于文档字段计算结果更新文档字段？

最优实现方案（MongoDB 4.2及以上版本） #

你当前的实现需要先查询聚合结果再逐次更新，会产生大量客户端与数据库的往返请求，数据量较大时性能损耗非常明显。MongoDB 4.2版本开始支持在更新操作中直接传入聚合管道，仅需一步即可完成字段计算与更新，所有逻辑都在数据库侧执行，无需客户端中间处理：

db.collection.updateMany(
  // 匹配条件和原逻辑一致，只处理存在arrayField的文档
  { arrayField: { $exists: true } },
  // 传入聚合管道作为更新逻辑
  [
    {
      $set: {
        // 直接对arrayField下的所有number字段求和后转成整数
        cumulativeField: { $toInt: { $sum: "$arrayField.number" } }
      }
    }
  ]
)

这个方案的优势非常明显：

• 仅需一次数据库请求，完全省略客户端遍历、逐次请求的开销
• 计算和更新都在数据库侧原子执行，不会出现聚合查询后、更新前文档变动导致的数据不一致问题

低版本兼容方案（MongoDB <4.2） #

如果你的MongoDB版本不支持更新管道，可以用bulkWrite批量操作优化现有逻辑，把所有更新请求打包后一次性提交，性能远高于循环调用updateOne：

// 先聚合得到计算结果，生成批量更新操作列表
const updateOperations = db.collection
  .aggregate([
    { $match: { arrayField: { $exists: true } } },
    { $addFields: { cumulativeField: { $sum: "$arrayField.number" } } },
    // 只取需要的字段，减少数据传输
    { $project: { id: 1, cumulativeField: 1 } }
  ])
  .map(doc => ({
    updateOne: {
      filter: { id: doc.id },
      update: { $set: { cumulativeField: NumberInt(doc.cumulativeField) } }
    }
  }))

// 一次性执行所有更新
db.collection.bulkWrite(updateOperations)

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