使用Node.js实现MongoDB两集合$lookup关联输出扁平化结果

问题核心 #

常规$lookup关联操作默认会将匹配到的副集合数据封装为数组字段嵌入主文档，无法直接得到顶层扁平字段结构，需要在聚合管道中增加数组拆解、字段投影的步骤完成结构调整。

完整实现代码 #

直接在locations集合上执行如下聚合管道即可得到目标结构：

db.locations.aggregate([
  // 1. 基于locationCategoryCode做等值关联
  {
    $lookup: {
      from: "locationCategory",
      localField: "locationCategoryCode",
      foreignField: "locationCategoryCode",
      as: "relatedCategory"
    }
  },
  // 2. 拆解关联产生的嵌套数组，将单条匹配结果展开为对象字段
  {
    $unwind: {
      path: "$relatedCategory",
      // 保留没有匹配到关联数据的主文档，避免数据丢失
      preserveNullAndEmptyArrays: true
    }
  },
  // 3. 提取需要的字段，整理为最终扁平结构
  {
    $project: {
      _id: 0,
      locationName: 1,
      locationCategoryName: "$relatedCategory.locationCategoryName"
    }
  }
])

效果说明 #

执行上述管道后，返回的文档为完全无嵌套的扁平结构，所有关联逻辑可正确匹配：

• locationCategoryCode为"1"的DC、New York会对应匹配locationCategoryName为"history"
• locationCategoryCode为"2"的Utah会对应匹配locationCategoryName为"nature"

替代优化方案 #

如果不想用$unwind拆解数组（避免极端场景下关联结果多条导致主文档被拆分），可以在投影阶段直接从关联数组中提取第一个匹配项的字段，效果完全一致：

// 替换上述管道中的$unwind和$project阶段
{
  $project: {
    _id: 0,
    locationName: 1,
    locationCategoryName: {
      $arrayElemAt: ["$relatedCategory.locationCategoryName", 0]
    }
  }
}

这种方式的优势是：哪怕副集合存在重复的locationCategoryCode，也不会把单个主文档拆成多条，只会取第一条匹配结果的字段值，逻辑更可控。 #

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