如何在MongoDB的$lookup后使用$match过滤关联数据？

问题原因 #

你之前的写法里，$lookup之后用$match过滤refData.unique: "first"，这个操作是筛选主文档——只要主文档的关联数组里存在至少一个符合条件的元素，就会保留该主文档，但不会对关联数组内部的元素做过滤，所以关联数组里还是会保留所有匹配的记录。

正确解决方案 #

方案1：在$lookup阶段直接过滤（推荐，性能更优） #

在$lookup的内部pipeline中提前过滤关联数据，这样关联时只会返回符合条件的记录，避免后续处理大量无用数据。

假设主文档通过_id关联RefData的masterId字段，示例代码如下：

db.MasterData.aggregate([
  // 先匹配目标主文档
  { $match: { /* 你的主文档匹配条件，比如 _id: ObjectId("xxx") */ } },
  {
    $lookup: {
      from: "RefData",
      let: { master_id: "$_id" }, // 定义主文档的关联字段变量
      pipeline: [
        {
          $match: {
            $expr: { $eq: ["$masterId", "$$master_id"] }, // 关联条件
            unique: "first" // 直接过滤unique为first的记录
          }
        }
      ],
      as: "refData"
    }
  }
])

方案2：$lookup后用$filter处理关联数组 #

如果需要先获取所有关联数据再做筛选，可以用$filter遍历关联数组，只保留符合条件的元素，替换原有的关联字段：

db.MasterData.aggregate([
  { $match: { /* 你的主文档匹配条件 */ } },
  {
    $lookup: {
      from: "RefData",
      localField: "_id",
      foreignField: "masterId",
      as: "refData"
    }
  },
  {
    $addFields: {
      refData: {
        $filter: {
          input: "$refData",
          cond: { $eq: ["$$this.unique", "first"] } // 筛选unique为first的元素
        }
      }
    }
  }
])

总结 #

• 方案1在关联阶段就完成过滤，数据传输量更小，性能更优，优先推荐。
• 方案2适合需要先保留全量关联数据，再进行二次筛选的场景。

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