如何用MongoDB聚合为每个文档添加随机关联帖子数据？

需求说明 #

现有一个帖子集合，每个文档是简单的帖子数据。需要给每个帖子文档附加指定数量的其他随机帖子，避免每次单独查询随机帖子，直接通过聚合管道完成数据拼接，最终每个文档会包含randomPost1到randomPostN的字段，对应不同的随机帖子。

聚合解决方案 #

通过以下聚合管道步骤实现需求：

• 聚合所有帖子到数组，同时保留原始文档列表
• 展开原始文档，回到单个文档的处理维度
• 过滤掉当前文档，对剩余帖子随机排序后取指定数量
• 将随机帖子数组转换为randomPost1这类命名字段
• 合并原始帖子数据与随机帖子字段，生成最终输出文档

完整聚合代码 #

db.posts.aggregate([
  // 聚合所有帖子到数组，同时保留原始文档集合
  {
    $group: {
      _id: null,
      allPosts: { $push: "$$ROOT" },
      docs: { $push: "$$ROOT" }
    }
  },
  // 展开原始文档数组，逐个处理每个帖子
  { $unwind: "$docs" },
  // 过滤掉当前帖子，对剩余帖子随机排序后取前6个
  {
    $addFields: {
      filteredPosts: {
        $filter: {
          input: "$allPosts",
          cond: { $ne: ["$$this.slug", "$docs.slug"] } // 用slug排除自身，若用_id则替换为对应字段
        }
      },
      randomPosts: {
        $slice: [
          { $shuffle: "$filteredPosts" },
          6 // 调整此处数字可改变随机帖子的数量
        ]
      }
    }
  },
  // 将随机帖子数组转换为指定命名的字段
  {
    $project: {
      _id: 0,
      original: "$docs",
      randomPost1: { $arrayElemAt: ["$randomPosts", 0] },
      randomPost2: { $arrayElemAt: ["$randomPosts", 1] },
      randomPost3: { $arrayElemAt: ["$randomPosts", 2] },
      randomPost4: { $arrayElemAt: ["$randomPosts", 3] },
      randomPost5: { $arrayElemAt: ["$randomPosts", 4] },
      randomPost6: { $arrayElemAt: ["$randomPosts", 5] }
    }
  },
  // 合并原始文档和随机帖子字段，生成最终输出结构
  {
    $replaceRoot: {
      newRoot: {
        $mergeObjects: [
          "$original",
          {
            randomPost1: "$randomPost1",
            randomPost2: "$randomPost2",
            randomPost3: "$randomPost3",
            randomPost4: "$randomPost4",
            randomPost5: "$randomPost5",
            randomPost6: "$randomPost6"
          }
        ]
      }
    }
  }
])

注意事项 #

• 若集合数据量极大，$group阶段会将所有文档加载到内存，需评估性能，必要时结合分页或索引优化
• 如需调整随机帖子数量，需同步修改$slice的数值和$project、$replaceRoot中的对应字段
• 示例中用slug作为唯一标识排除自身，若你的集合使用_id作为主键，将$ne条件替换为{ $ne: ["$$this._id", "$docs._id"] }即可

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