MongoDB聚合：如何基于数组元素关联集合并重组数据结构

嘿，我懂你已经掌握了基础的$lookup用法，但碰到数组元素要逐个关联其他集合的场景就懵了对吧？别慌，咱们一步步来搞定这个需求，两种方案任你选，先从最容易理解的分步方法说起~

方案一：分步拆分解法（直观易懂） #

假设你的第一个集合叫collectionA，第二个叫collectionB，直接用下面的聚合管道就能得到你想要的结果：

db.collectionA.aggregate([
  // 1. 把data数组拆成单个文档，让每个元素都能单独关联
  { $unwind: "$data" },
  // 2. 关联collectionB，用当前的data值匹配value字段
  {
    $lookup: {
      from: "collectionB",
      localField: "data",
      foreignField: "value",
      as: "matchedData"
    }
  },
  // 3. 把匹配到的数组转成单个对象（一对一场景下用）
  { $unwind: "$matchedData" },
  // 4. 按原文档_id分组，重新组合成目标数组结构
  {
    $group: {
      _id: "$_id",
      data: {
        $push: {
          id: "$data",
          name: "$matchedData.name"
        }
      }
    }
  }
])

各阶段作用详解： #

• $unwind: "$data"：把collectionA里每个文档的data数组“拆”成多个独立文档，比如原文档{_id: xxx, data: [1,2,3]}会变成3个文档，每个文档的data是单个数字。这一步是为了让数组里的每个元素都能单独去关联collectionB。
• $lookup：用拆出来的单个data值，去匹配collectionB里的value字段，匹配到的结果会存在matchedData数组里。
• $unwind: "$matchedData"：因为你的需求里每个数字应该只对应collectionB里的一个文档，所以把matchedData数组拆成单个对象，方便后续取值。
• $group：按原文档的_id重新分组，用$push把拆分的元素重新拼成数组，同时指定每个数组元素的结构为{id: 原数字, name: 匹配到的name}。

方案二：管道版$lookup（性能更优） #

如果你的数据集比较大，不想做两次$unwind，可以用$lookup的高级用法——带管道的关联，直接在关联阶段过滤数据，再用$map映射结果：

db.collectionA.aggregate([
  {
    $lookup: {
      from: "collectionB",
      let: { dataArray: "$data" },
      pipeline: [
        // 只匹配value在当前data数组里的文档
        { $match: { $expr: { $in: ["$value", "$$dataArray"] } } },
        // 只保留需要的字段，减少数据传输
        { $project: { value: 1, name: 1, _id: 0 } }
      ],
      as: "matchedDocs"
    }
  },
  {
    $addFields: {
      // 遍历原data数组，映射成目标结构
      data: {
        $map: {
          input: "$data",
          as: "item",
          in: {
            id: "$$item",
            // 通过value的位置找到对应的name
            name: {
              $arrayElemAt: [
                "$matchedDocs.name",
                { $indexOfArray: ["$matchedDocs.value", "$$item"] }
              ]
            }
          }
        }
      }
    }
  },
  // 移除临时的matchedDocs字段，保持结果整洁
  { $project: { matchedDocs: 0 } }
])

这个方案的优势： #

不需要拆分和重新组合数组，直接通过$map和$indexOfArray把原数组元素和匹配到的name对应起来，减少了聚合阶段的操作，对于大数据量的场景性能会更好。

两种方案都能得到你想要的结果：{ _id: "some ObjectId", data: [ { id: 1, name: "first"}, { id: 2, name: "first"}, { id: 3, name: "first"} ] }，选哪种取决于你的数据规模和理解习惯~

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