如何在MongoDB中跨集合关联文档（Node.js官方mongodb包）

你可以通过MongoDB的**聚合管道（Aggregation Pipeline）**结合$lookup操作实现跨集合关联查询，最终得到包含嵌套评论及用户信息的帖子数据。以下是基于Node.js官方mongodb包的具体实现：

代码示例 #

const { MongoClient } = require('mongodb');

async function getPostsWithComments() {
  const uri = 'mongodb://localhost:27017'; // 替换为你的MongoDB连接字符串
  const client = new MongoClient(uri);

  try {
    await client.connect();
    const db = client.db('your-db-name'); // 替换为你的数据库名称

    const postCollection = db.collection('post');
    const result = await postCollection.aggregate([
      // 第一步：关联comment集合，获取当前帖子的所有评论
      {
        $lookup: {
          from: 'comment',
          let: { postId: '$_id' }, // 传递当前帖子的_id作为变量
          pipeline: [
            // 匹配当前帖子对应的评论
            { $match: { $expr: { $eq: ['$postId', '$$postId'] } } },
            // 第二步：在评论管道内关联users集合，获取评论作者信息
            {
              $lookup: {
                from: 'users',
                localField: 'userId',
                foreignField: '_id',
                as: 'userInfo'
              }
            },
            // 展开用户信息数组（$lookup返回数组，即使只有一个匹配项）
            { $unwind: '$userInfo' },
            // 将原userId字段替换为用户信息对象
            { $addFields: { userId: '$userInfo' } },
            // 移除临时的userInfo字段
            { $project: { userInfo: 0 } }
          ],
          as: 'comments' // 将匹配到的评论存入comments字段
        }
      }
    ]).toArray();

    console.log(result);
    return result;
  } finally {
    await client.close();
  }
}

// 调用查询函数
getPostsWithComments().catch(console.error);

关键步骤说明 #

• 外层$lookup：从post集合出发，关联comment集合，通过post._id匹配comment.postId，将所有匹配的评论聚合为comments数组。
• 内层$lookup：在评论的子管道中，进一步关联users集合，通过comment.userId匹配users._id，获取评论作者的详细信息。
• $unwind：将$lookup返回的用户信息数组展开为单个对象（每个评论仅对应一个用户）。
• $addFields + $project：将原评论中的userId字段替换为完整的用户信息对象，并清理临时字段。

执行后即可得到符合需求的结构：每个帖子包含对应的评论数组，且评论中的userId字段已替换为用户的具体信息。

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