MongoDB聚合管道如何从嵌套子文档数组随机抽取指定数量元素

你可以通过聚合操作符组合实现单文档内嵌套数组的随机采样，以下提供两种适配不同MongoDB版本的实现方案：

方案1：适配MongoDB 4.4及以上版本（性能最优） #

直接通过数组内排序+截断实现，不需要拆分文档，性能开销极小：

// 新增2个聚合阶段即可完成需求
[
  {
    $set: {
      "productLine.product": {
        $slice: [
          {
            $sortArray: {
              input: {
                $map: {
                  input: "$productLine.product",
                  as: "productId",
                  in: {
                    id: "$$productId",
                    sortKey: { $rand: {} }
                  }
                }
              },
              sortBy: { sortKey: 1 }
            }
          },
          2
        ]
      }
    }
  },
  {
    $set: {
      "productLine.product": {
        $map: {
          input: "$productLine.product",
          as: "item",
          in: "$$item.id"
        }
      }
    }
  }
]

逻辑说明：

• 首先给数组内每个产品ID绑定一个0~1的随机值作为排序键
• 按随机值对数组做升序排序，实现数组随机打乱
• 用$slice截取排序后的前2个元素，得到随机采样结果
• 最后剥离临时生成的排序键，恢复product数组的ID字符串格式，输出结果完全匹配预期结构，后续可直接接$lookup做关联查询。

方案2：适配MongoDB 4.4以下版本 #

如果数据库版本不支持$rand和$sortArray，可以通过拆分数组+文档级采样+重组的方式实现：

[
  { $unwind: "$productLine.product" },
  { $sample: { size: 2 } },
  {
    $group: {
      _id: "$_id",
      originalDoc: { $first: "$$ROOT" },
      sampledProducts: { $push: "$productLine.product" }
    }
  },
  {
    $replaceRoot: {
      newRoot: {
        $mergeObjects: [
          "$originalDoc",
          {
            productLine: {
              $mergeObjects: [
                "$originalDoc.productLine",
                { product: "$sampledProducts" }
              ]
            }
          }
        ]
      }
    }
  }
]

逻辑说明：

• 先用$unwind把嵌套数组拆分为独立文档
• 用官方提供的$sample做文档级随机采样取2条
• 按原始文档ID分组，把采样得到的产品ID重新组装为数组
• 最后替换回原始文档结构，保留所有其他字段不变。

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