MongoDB聚合：更新数组中刚创建的销售记录

现有MongoDB数据结构定义如下：

Item = {
  name: String,
  price: Number
}

Sales = {
  items: [Item],
  totalSaleValue: Number
}

Clients = {
  clientName: String,
  sales: [Sales]
}

要求每次插入或更新Sales记录时，自动根据该记录items数组中所有商品的price总和，计算并设置totalSaleValue字段。

当前使用的插入语句如下，虽能成功向指定Clients文档的sales数组添加新记录，但无法在同一查询中访问刚插入的记录来计算totalSaleValue：

db.collection.findOneAndUpdate(
  { _id: <the _id for the client> },
  [
    { $set: { sales: { $concatArrays: [ '$sales', [ /* 待插入的Sales对象 */ ] ] } } }
  ]
)

需要在单个查询内完成：访问sales数组 -> 获取最后一条记录 -> 计算并设置其totalSaleValue为对应items的price总和。

方案一：插入时直接计算总和（最简洁高效） #

不需要先插入再修改，直接在构造新Sales对象时就计算好totalSaleValue，一步完成插入：

db.collection.findOneAndUpdate(
  { _id: <the _id for the client> },
  [
    { $set: {
      sales: {
        $concatArrays: [
          '$sales',
          [
            {
              items: /* 新销售的items数组 */,
              totalSaleValue: { $sum: /* 新销售的items数组 */.price }
            }
          ]
        ]
      }
    } }
  ],
  { returnNewDocument: true } // 可选，返回更新后的完整文档
)

说明：直接通过$sum计算传入的items数组中所有price的总和，将其作为totalSaleValue的值和items一起组成完整的Sales对象，再拼接到sales数组中，全程在一个更新管道内完成。

方案二：用$push实现语义更清晰的插入 #

MongoDB 4.2及以上版本支持在$push中使用聚合表达式，写法更贴合“向数组添加元素”的语义：

db.collection.findOneAndUpdate(
  { _id: <the _id for the client> },
  [
    { $set: {
      sales: {
        $push: {
          into: '$sales',
          item: {
            items: /* 新销售的items数组 */,
            totalSaleValue: { $sum: /* 新销售的items数组 */.price }
          }
        }
      }
    } }
  ],
  { returnNewDocument: true }
)

方案三：先插入再修正总和（适配特殊场景） #

如果必须先插入不带totalSaleValue的记录，再补全该字段，可以用临时字段配合$map、$last操作实现：

db.collection.findOneAndUpdate(
  { _id: <the _id for the client> },
  [
    // 1. 插入新销售记录，同时暂存该记录到临时字段
    { $set: {
      tempNewSale: /* 不带totalSaleValue的新Sales对象 */,
      sales: { $concatArrays: [ '$sales', [ /* 不带totalSaleValue的新Sales对象 */ ] ] }
    } },
    // 2. 计算新记录的price总和
    { $set: { tempTotal: { $sum: '$tempNewSale.items.price' } } },
    // 3. 更新sales数组的最后一条记录，设置totalSaleValue
    { $set: {
      sales: {
        $map: {
          input: '$sales',
          as: 'sale',
          in: {
            $cond: {
              if: { $eq: [ '$$sale', { $last: '$sales' } ] },
              then: { $mergeObjects: [ '$$sale', { totalSaleValue: '$tempTotal' } ] },
              else: '$$sale'
            }
          }
        }
      }
    } },
    // 4. 清理临时字段
    { $unset: [ 'tempNewSale', 'tempTotal' ] }
  ],
  { returnNewDocument: true }
)

说明：这种方式适合无法提前构造完整Sales对象的场景，但步骤更多，性能略低于前两种方案。

额外提示 #

• 如果需要更新已有Sales记录的totalSaleValue，可以用类似的$map逻辑，针对指定的sales元素重新计算$sum并更新。
• 确保传入的items数组每个元素都包含有效的price数值，否则$sum会返回null（视MongoDB版本可能略有差异）。

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