如何使用MongoDB聚合管道实现双层数据分区求和？

开发者阿泽

2026-6-13

按key和subkey双层分区对amount求和的最佳实现（除$setWindowFields外）

我有一份样本数据，需要按key和subkey两个字段进行双层分区，并对分组后的amount字段求和。

样本数据

[
  {
    "key": "1",
    "subkey": "1_1",
    "amount": 1
  },
  {
    "key": "1",
    "subkey": "1_1",
    "amount": 2
  },
  {
    "key": "1",
    "subkey": "1_2",
    "amount": 5
  },
  {
    "key": "2",
    "subkey": "2_1",
    "amount": 3
  },
  {
    "key": "2",
    "subkey": "2_2",
    "amount": 4
  }
]

当前使用的聚合管道（subkey层级分组效果不佳）

db.collection.aggregate({
  "$setWindowFields": {
    "partitionBy": "$key",
    "sortBy": {
      "subkey": 1
    },
    "output": {
      "total": {
        "$sum": "$amount",
        "window": {
          "documents": [
            "unbounded",
            "current"
          ]
        }
      }
    }
  }
})

预期输出

key | subkey | total
 1  | 1_1    | 3
 1  | 1_2    | 5
 2  | 2_1    | 3
 2  | 2_2    | 4

最佳实现方式：使用`$group`复合分组

直接通过$group阶段以key和subkey的组合作为分组键，是实现该需求最高效、最直接的方式，完全匹配预期输出：

db.collection.aggregate([
  {
    $group: {
      _id: {
        key: "$key",
        subkey: "$subkey"
      },
      total: { $sum: "$amount" }
    }
  },
  {
    $project: {
      _id: 0,
      key: "$_id.key",
      subkey: "$_id.subkey",
      total: 1
    }
  }
])

逻辑说明

$group阶段：将_id设置为包含key和subkey的复合对象，以此作为双层分区的依据，同时用$sum计算每组的amount总和。
$project阶段：将分组生成的_id对象拆解为独立的key和subkey字段，移除默认的_id字段，让输出结构与预期一致。