MongoDB聚合$lookup保留双层数组原结构的实现方法

问题原因 #

直接使用foo.bar作为localField和as时，MongoDB会将嵌套的双层数组扁平化处理，赋值时会覆盖原有的数组结构，导致外层的foo从数组变为对象，破坏了原始的双层数组层级。

解决方案 #

通过展开数组→关联数据→匹配替换→重组数组的步骤实现，完整聚合管道如下：

[
  // 1. 展开外层foo数组，将每个foo元素单独作为文档
  { $unwind: "$foo" },
  // 2. 关联目标集合，获取bar数组中_id对应的完整数据
  {
    $lookup: {
      from: "collection",
      localField: "foo.bar._id",
      foreignField: "_id",
      as: "barLookupResults"
    }
  },
  // 3. 将原bar数组中的元素替换为lookup到的完整数据
  {
    $addFields: {
      "foo.bar": {
        $map: {
          input: "$foo.bar",
          as: "barItem",
          in: {
            $arrayElemAt: [
              {
                $filter: {
                  input: "$barLookupResults",
                  cond: { $eq: ["$$this._id", "$$barItem._id"] }
                }
              },
              0
            ]
          }
        }
      }
    }
  },
  // 4. 移除临时的lookup结果字段
  { $unset: "barLookupResults" },
  // 5. 重新聚合foo数组，恢复原始双层结构
  {
    $group: {
      _id: "$_id",
      foo: { $push: "$foo" },
      // 原文档其他字段需在此用$first/$last保留，示例：
      // otherField: { $first: "$otherField" }
    }
  }
]

步骤说明 #

• $unwind: 拆分外层foo数组，让每个数组元素单独成为一个文档，便于处理内层bar数组。
• $lookup: 基于foo.bar._id关联目标集合，将匹配结果存入临时字段，避免直接覆盖原结构。
• $addFields + $map + $filter: 遍历原bar数组，通过_id匹配临时字段中的关联数据，替换原bar元素为完整文档。
• $unset: 清理临时字段，保持文档结构整洁。
• $group: 将拆分后的文档重新聚合为原始的foo数组结构，同时保留原文档的其他字段（需在group阶段显式声明）。

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