MongoDB数组聚合问题：如何在$push中传入$value_sum

嘿，我来帮你搞定这个MongoDB聚合的问题！

你遇到的核心问题是聚合阶段的变量作用域限制：在同一个$group阶段里，$push是无法引用同一阶段中$sum计算出来的value_sum的——因为这些字段是并行计算的，彼此之间没有依赖关系。要解决这个问题，我们需要把「计算分组总和」和「处理条目占比/合并others」拆分成不同的聚合阶段，先算出总和再往下传递使用。

下面是完整的解决方案，包含按时间分组、计算占比，以及可选的「合并多余条目为others」逻辑：

假设你的数据结构是这样的 #

{
  "time_period": "2024-01", // 按这个字段分组
  "category": "A",          // 每个条目类别
  "value": 100              // 需要计算占比的数值
}

完整聚合管道代码 #

db.your_collection.aggregate([
  // 阶段1：按时间分组，先算出每个周期的总数值，同时收集所有条目明细
  {
    $group: {
      _id: "$time_period",
      total_value: { $sum: "$value" }, // 这就是你需要的value_sum
      raw_items: {
        $push: {
          category: "$category",
          value: "$value"
        }
      }
    }
  },
  // 阶段2：计算每个条目的占比，同时处理others分组（可选逻辑）
  {
    $set: {
      processed_items: {
        $let: {
          vars: {
            // 先给每个条目计算占比
            items_with_ratio: {
              $map: {
                input: "$raw_items",
                as: "item",
                in: {
                  category: "$$item.category",
                  value: "$$item.value",
                  ratio: { $divide: ["$$item.value", "$total_value"] } // 这里可以直接引用阶段1的total_value
                }
              }
            },
            // 可选：按数值降序排序，确保保留占比最高的前2个条目
            sorted_items: { $sortArray: { input: "$$items_with_ratio", sortBy: { value: -1 } } }
          },
          in: {
            $cond: {
              // 判断当前周期的条目数是否超过2
              if: { $gt: [{ $size: "$$sorted_items" }, 2] },
              then: [
                // 保留前2个条目
                { $arrayElemAt: ["$$sorted_items", 0] },
                { $arrayElemAt: ["$$sorted_items", 1] },
                // 合并剩余条目为others，计算总数值和总占比
                {
                  category: "others",
                  value: { $sum: { $map: { input: { $slice: ["$$sorted_items", 2] }, as: "rest", in: "$$rest.value" } } },
                  ratio: { $sum: { $map: { input: { $slice: ["$$sorted_items", 2] }, as: "rest", in: "$$rest.ratio" } } }
                }
              ],
              // 条目数≤2时，直接返回带占比的条目
              else: "$$sorted_items"
            }
          }
        }
      }
    }
  },
  // 阶段3：整理输出格式（可选）
  {
    $project: {
      _id: 0,
      time_period: "$_id",
      total_value: 1,
      items: "$processed_items"
    }
  }
])

关键逻辑说明 #

1. 拆分阶段解决变量引用问题：先在$group阶段算出每个时间周期的total_value，同时收集所有条目到raw_items数组里。这样后续阶段就能直接引用total_value来计算占比了。
2. 处理others分组：用$cond判断条目数量，超过2时用$slice截取剩余条目，再用$sum合并它们的数值和占比，生成"others"条目。
3. 可选排序：用$sortArray按数值降序排序，确保保留占比最高的两个条目，你也可以根据需求调整排序规则。

这样就能完美实现你要的按时间分组、计算占比，以及可选合并多余条目为others的需求啦！

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