MongoDB中在$facet阶段内使用$count计算95百分位数的问题

你需要在MongoDB聚合中获取文档总数，并基于该数值为每个文档计算95百分位数索引，但原管道中$facet的子管道无法互相访问数据，导致Percentile95Index计算失败。

示例文档 #

[
  {
    "_id": ObjectId("178768747638364736373637"),
    "start_time": ISODate("2019-02-03T12:00:00.000Z"),
    "finish_time": ISODate("2019-02-03T12:01:00.000Z")
  },
  {
    "_id": ObjectId("266747364736363536353555"),
    "start_time": ISODate("2019-02-03T12:00:00.000Z"),
    "finish_time": ISODate("2019-02-03T12:03:00.000Z")
  },
  {
    "_id": ObjectId("367463536453623546353625"),
    "start_time": ISODate("2019-02-03T12:00:00.000Z"),
    "finish_time": ISODate("2019-02-03T12:08:00.000Z")
  }
]

期望输出 #

[
  {
    "Percentile95Index": 2.8499999999999996,
    "_id": ObjectId("178768747638364736373637"),
    "duration": 60,
    "totalCount": 3
  },
  {
    "Percentile95Index": 2.8499999999999996,
    "_id": ObjectId("266747364736363536353555"),
    "duration": 180,
    "totalCount": 3
  },
  {
    "Percentile95Index": 2.8499999999999996,
    "_id": ObjectId("367463536453623546353625"),
    "duration": 480,
    "totalCount": 3
  }
]

问题原因 #

$facet的每个子管道是并行独立执行的，彼此之间无法共享数据。你原管道中尝试在pipelineResults分支的$project阶段访问totalCount.value，这是不可能的，因为两个分支的执行上下文完全隔离。

解决方案 #

方案1：改进原$facet管道（最小修改）

将Percentile95Index的计算移到$replaceRoot阶段，此时已经可以访问到totalCount分支的结果：

db.collection.aggregate([
  {
    $facet: {
      totalCount: [{ $count: "value" }],
      pipelineResults: [
        {
          $project: {
            duration: {
              $divide: [
                { $subtract: ["$finish_time", "$start_time"] },
                1000
              ]
            }
          }
        }
      ]
    }
  },
  { $unwind: "$totalCount" },
  { $unwind: "$pipelineResults" },
  {
    $replaceRoot: {
      newRoot: {
        $mergeObjects: [
          "$pipelineResults",
          {
            totalCount: "$totalCount.value",
            Percentile95Index: { $multiply: [0.95, "$totalCount.value"] }
          }
        ]
      }
    }
  }
])

方案2：先获取总数再关联文档

先通过$count获取总文档数，再用$lookup将该数值关联到每个文档，最后计算所需字段：

db.collection.aggregate([
  // 获取总文档数
  { $count: "totalCount" },
  // 关联原集合所有文档并计算duration
  {
    $lookup: {
      from: "collection", // 替换为你的实际集合名称
      pipeline: [
        {
          $project: {
            duration: { $divide: [{ $subtract: ["$finish_time", "$start_time"] }, 1000] }
          }
        }
      ],
      as: "docs"
    }
  },
  // 展开文档列表
  { $unwind: "$docs" },
  // 合并字段并计算百分位数索引
  {
    $replaceRoot: {
      newRoot: {
        $mergeObjects: [
          "$docs",
          {
            totalCount: "$totalCount",
            Percentile95Index: { $multiply: [0.95, "$totalCount"] }
          }
        ]
      }
    }
  }
])

说明 #

• 方案1更贴近你原有的管道结构，修改量小，适合需要保留$facet多分支处理的场景。
• 方案2逻辑更直观，避免了$facet的上下文隔离问题，适合简单的总数+文档处理场景。

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