如何通过MongoDB聚合管道从对象数组计算正确百分比

需求说明 #

• 计算所有文档中members数组的全局总积分（示例中为5500）
• 计算单个文档的members数组总积分（如Josh文档为2000、Carl文档为3500）
• 按公式(单文档总积分/全局总积分)*100计算占比，最终返回_id、leader、memberpoints、percentage字段

问题描述 #

现有聚合管道计算出的百分比不符合预期，例如Josh文档的占比应为36.3636，却得到18.1818，问题出在$set阶段的$divide逻辑及管道顺序。

现有错误聚合管道 #

db.users.aggregate([
{$unwind: "$members"}, 
{$setWindowFields: { output: { total: { $sum: "$members.points" }}}},
{$set: { percentage: { $multiply: [{ $divide: ["$members.points", "$total"] }, 100] }}},
{$group: { _id: "$_id", leader: {$first: "$leader"}, memberpoints: {$sum: "$members.points"}, percentage: {$first: "$percentage"}}}
])

测试文档示例 #

[
  {
    "_id": { "$oid": "00001" },
    "leader": "Josh",
    "members": [
      {"name": "Person A", "points": 500},
      {"name": "Person B", "points": 500},
      {"name": "Person C", "points": 1000}
    ]
  },
  {
    "_id": { "$oid": "00002" },
    "leader": "Carl",
    "members": [
      {"name": "Person D", "points": 1000},
      {"name": "Person E", "points": 1000},
      {"name": "Person F", "points": 1500}
    ]
  }
]

错误原因分析 #

1. 过早$unwind拆分数组：提前拆分members数组后，每条记录对应单个成员，后续计算百分比时用的是单个成员的积分而非整个文档的成员总积分除以全局总积分
2. $group阶段取随机值：$first取的是拆分后第一条成员的占比，完全不是整个文档的占比，结果毫无意义

正确聚合管道实现 #

db.users.aggregate([
  // 计算当前文档的成员总积分
  {
    $addFields: {
      memberpoints: { $sum: "$members.points" }
    }
  },
  // 计算全集合的全局总积分（默认全窗口）
  {
    $setWindowFields: {
      output: {
        globalTotal: { $sum: "$memberpoints" }
      }
    }
  },
  // 计算占比并保留需要的字段
  {
    $project: {
      _id: 1,
      leader: 1,
      memberpoints: 1,
      percentage: { $multiply: [{ $divide: ["$memberpoints", "$globalTotal"] }, 100] }
    }
  }
])

执行结果验证 #

• Josh文档：memberpoints=2000，百分比2000/5500*100≈36.3636
• Carl文档：memberpoints=3500，百分比3500/5500*100≈63.6364
  完全符合需求预期。

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