MongoDB聚合管道如何统计文档数同时保留输入文档？

$count阶段会消费所有输入文档，仅返回单条计数结果，无法保留原始输入文档。要实现计数结果与原文档同时传递到后续阶段，可根据部署的MongoDB版本选择以下方案：

方案1：$facet多分支聚合（兼容所有MongoDB版本） #

• 实现原理：$facet支持基于同一批输入文档并行执行多个独立子管道，用一个子管道单独统计文档总数，另一个子管道保留/处理所有原文档，最后将计数结果拼接到每一条原文档上。
• 管道示例：

db.collection.aggregate([
  // 原有匹配筛选阶段
  { $match: { sellerId: <传入的sellerId参数> } },
  // 并行执行计数和原文档处理
  { $facet: {
      totalCount: [ { $count: "count" } ],
      docs: [ 
        // 此处可插入你需要对原文档执行的中间处理逻辑：字段投影、过滤、关联查询等
      ]
    }
  },
  // 拆分文档数组
  { $unwind: "$docs" },
  // 将计数字段拼接到每个原文档中
  { $replaceRoot: {
      newRoot: {
        $mergeObjects: [
          "$docs",
          // 空结果时默认返回0，避免字段缺失
          { numDocs: { $ifNull: [ { $arrayElemAt: ["$totalCount.count", 0] }, 0 ] } }
        ]
      }
    }
  }
  // 后续业务所需的其他聚合阶段可直接追加，此时所有文档已携带numDocs字段
])

方案2：$setWindowFields窗口计算（MongoDB 5.0及以上版本推荐） #

• 实现原理：窗口函数可在不改变原文档数量的前提下，对指定范围内的文档执行聚合计算，直接将计算结果作为新字段追加到每个文档上，无需分支、拆分、重组文档，大数据量下性能远高于$facet方案。
• 管道示例：

db.collection.aggregate([
  // 原有匹配筛选阶段
  { $match: { sellerId: <传入的sellerId参数> } },
  // 直接为所有匹配文档追加总数字段
  { $setWindowFields: {
      output: {
        numDocs: { $count: {} }
      }
    }
  }
  // 后续业务所需的其他聚合阶段可直接追加，所有文档已携带numDocs字段
])

• 扩展用法：如果需要按维度分组计数（例如统计每个卖家名下的文档总数，而非全局计数），只需在$setWindowFields中增加分组配置即可，无需额外$group阶段：

{ $setWindowFields: {
    partitionBy: "$sellerId", // 按sellerId分组
    output: {
      numDocs: { $count: {} } // 每个组内独立计数
    }
  }
}

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