MongoDB视图中实现openBalance汇总并输出总计

嘿，这个需求其实很容易实现，MongoDB的聚合管道里有几个好用的方法能帮你搞定——既保留每条记录的原有数据，又把所有openBalance的总计值加到每条结果里。

方法1：使用$setWindowFields（推荐，MongoDB 4.2+） #

这是最简洁高效的方案，它可以在不改变文档数量的前提下，为每个文档添加全局聚合计算的结果。你只需要在现有聚合管道的最后添加一个$setWindowFields阶段：

{
  $setWindowFields: {
    partitionBy: null, // 表示对整个数据集进行计算，不分区
    sortBy: {}, // 不需要排序，留空即可
    output: {
      totalOpenBalance: { // 新增的总计字段名
        $sum: "$openBalance", // 计算openBalance的总和
        window: {
          documents: ["unbounded", "unbounded"] // 指定窗口包含所有文档
        }
      }
    }
  }
}

完整示例 #

假设你的现有聚合管道已经完成了匹配、投影等操作，最终的完整管道会是这样：

db.yourCollection.aggregate([
  // 你的现有聚合阶段（比如匹配、投影等）
  { $match: { /* 你的匹配条件 */ } },
  { $project: { customer_id: 1, openBalance: 1, /* 其他需要保留的字段 */ } },
  // 新增的总计计算阶段
  {
    $setWindowFields: {
      partitionBy: null,
      sortBy: {},
      output: {
        totalOpenBalance: {
          $sum: "$openBalance",
          window: { documents: ["unbounded", "unbounded"] }
        }
      }
    }
  }
])

执行后，每条输出文档都会保留原有字段，同时新增totalOpenBalance字段，值为所有记录openBalance的总和。

方法2：使用$facet（兼容低版本MongoDB） #

如果你的MongoDB版本低于4.2，无法使用$setWindowFields，可以用$facet来分别计算原有数据和总计值，再合并结果：

// 第一步：用facet同时获取原有数据和总计
{
  $facet: {
    originalData: [ /* 你的现有聚合阶段 */ ],
    total: [
      { $group: { _id: null, totalOpenBalance: { $sum: "$openBalance" } } }
    ]
  }
},
// 第二步：展开原有数据数组
{ $unwind: "$originalData" },
// 第三步：把总计值添加到每条原有数据中
{
  $set: {
    "originalData.totalOpenBalance": { $arrayElemAt: ["$total.totalOpenBalance", 0] }
  }
},
// 第四步：把合并后的文档作为根文档输出
{ $replaceRoot: { newRoot: "$originalData" } }

这个方法的思路是先通过$facet并行计算两个结果集：原有聚合数据和总计值，然后将总计值合并到每条原有数据中，最后恢复文档结构。

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