如何在MongoDB聚合管道中获取匹配筛选条件的指定百分比随机结果？

原始数据结构 #

[
  {"key": 1, "val":"1"},
  {"key": 2, "val":"1"},
  {"key": 3, "val":"1"},
  {"key": 4, "val":"2"}
]

需求说明 #

筛选出val: "1"的文档后，随机抽取其中50%向上取整的数据，且必须在聚合管道内完成（输出直接作为后续管道阶段的输入，不能依赖外部代码处理）。

聚合管道实现方案 #

以下是完整的聚合管道代码，每个阶段的作用都有明确拆解：

db.collection.aggregate([
  // 1. 过滤出符合条件的文档
  { $match: { val: "1" } },
  // 2. 给每个文档添加随机排序字段，用于后续打乱顺序
  { $addFields: { randomSort: { $rand: {} } } },
  // 3. 按随机字段排序，实现文档随机打乱
  { $sort: { randomSort: 1 } },
  // 4. 聚合所有匹配文档，同时统计总条数
  { $group: {
      _id: null,
      docs: { $push: "$$ROOT" },
      total: { $sum: 1 }
    }
  },
  // 5. 计算需要抽取的数量：总条数/2后向上取整
  { $addFields: {
      takeCount: { $ceil: { $divide: ["$total", 2] } }
    }
  },
  // 6. 从打乱的文档数组中截取指定数量的随机数据
  { $project: {
      result: { $slice: ["$docs", "$takeCount"] },
      _id: 0
    }
  },
  // 7. 将数组拆回单个文档格式，适配后续管道阶段
  { $unwind: "$result" },
  { $replaceRoot: { newRoot: "$result" } },
  // 8. 清理临时添加的随机排序字段
  { $project: { randomSort: 0 } }
])

关键阶段解释 #

• 随机打乱：通过$rand生成0-1的随机数，再按该值排序，确保每次执行的抽取结果都是随机的
• 数量计算：用$ceil（向上取整）和$divide（除法）组合，精准计算需要抽取的文档数
• 格式适配：最后通过$unwind和$replaceRoot将聚合后的数组拆回单个文档结构，保证输出可以直接进入后续管道阶段处理

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