MongoDB Lookup对象数组场景：三集合关联查询需求

嘿，我刚好做过类似的场景，给你一步步拆解怎么实现这个对象数组的关联查询！

首先，咱们先明确要达成的核心目标：

• 把collectionA里的company对象关联到company集合，获取完整的企业信息（替换或补充原有的企业字段）
• 处理conductors数组，把带userId的已注册用户关联到对应的用户集合（假设是users集合，你可以替换成实际的集合名称），未注册的用户则保留原有的名称字段

完整聚合查询代码 #

db.collectionA.aggregate([
  // 第一步：关联company集合，替换原company字段为完整企业信息
  {
    $lookup: {
      from: "company",
      localField: "company.id",
      foreignField: "_id",
      as: "company"
    }
  },
  {
    $set: {
      // 由于lookup返回的是数组，取第一个元素替换原company字段
      company: { $first: "$company" }
    }
  },
  // 第二步：关联已注册用户集合，获取conductors中已注册用户的详细信息
  {
    $lookup: {
      from: "users", // 替换为你的已注册用户集合名
      let: { conductorsList: "$conductors" },
      pipeline: [
        {
          $match: {
            $expr: {
              $in: ["$_id", "$$conductorsList.userId"]
            }
          }
        },
        {
          $project: {
            _id: 1,
            name: 1,
            email: 1 // 这里可以添加你需要的其他用户字段
          }
        }
      ],
      as: "registeredConductors"
    }
  },
  // 第三步：合并conductors数组与关联后的用户信息
  {
    $set: {
      conductors: {
        $map: {
          input: "$conductors",
          as: "item",
          in: {
            $cond: {
              if: { $ne: ["$$item.userId", undefined] },
              then: {
                $mergeObjects: [
                  "$$item",
                  {
                    $first: {
                      $filter: {
                        input: "$registeredConductors",
                        cond: { $eq: ["$$this._id", "$$item.userId"] }
                      }
                    }
                  }
                ]
              },
              else: "$$item"
            }
          }
        }
      }
    }
  },
  // 第四步：清理临时字段
  {
    $unset: "registeredConductors"
  }
])

关键步骤解释 #

1. 企业信息关联：
   先用基础的$lookup将collectionA.company.id与company._id关联，得到的结果是一个数组，所以用$set取数组第一个元素替换原有的company字段，这样就能拿到完整的企业数据。

2. 已注册用户关联：
   这里用到了$lookup的高级用法——自定义子pipeline。我们把conductors数组作为变量传入子pipeline，匹配users集合中_id存在于conductors.userId的用户，同时通过$project指定需要返回的用户字段，避免获取冗余数据。

3. 数组元素合并：
   用$map遍历原conductors数组，对每个元素做判断：

   • 如果元素包含userId（已注册用户），就用$mergeObjects把原元素和关联到的用户详细信息合并
   • 如果没有userId（未注册用户），直接保留原元素的结构

4. 临时字段清理：
   最后用$unset移除中间生成的registeredConductors临时字段，让返回结果更简洁。

特殊场景优化 #

如果存在conductors里的userId在users集合中无匹配记录的情况，可以在合并时添加$ifNull处理，避免合并到null值：

then: {
  $mergeObjects: [
    "$$item",
    {
      $ifNull: [
        { $first: { $filter: { input: "$registeredConductors", cond: { $eq: ["$$this._id", "$$item.userId"] } } } },
        {} // 无匹配时合并空对象，不破坏原字段结构
      ]
    }
  ]
}

这样处理后，你就能得到既关联了完整企业信息，又区分处理了已注册/未注册用户的查询结果啦！

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