Mongoose查询多层嵌套数组中匹配指定ID的子文档方法

针对两层嵌套数组的过滤需求，直接使用find搭配投影无法实现二级数组的条件过滤，需要通过聚合管道分步处理，以下是可直接复用的查询语句，全表遍历场景下也能保证查询效率：

// 替换成你的实际集合名
db.products.aggregate([
  // 前置匹配筛掉不包含目标variant的文档，大幅减少后续计算量，数据量大时务必加这一步
  {
    $match: {
      "variants._id": "62be0271d373b2f2fc1826a3"
    }
  },
  // 第一层过滤：variants数组仅保留指定_id的子文档
  {
    $addFields: {
      variants: {
        $filter: {
          input: "$variants",
          cond: { $eq: ["$$this._id", "62be0271d373b2f2fc1826a3"] }
        }
      }
    }
  },
  // 第二层过滤：对保留的variant子文档，过滤其内部items数组，仅保留指定_id的元素
  {
    $addFields: {
      variants: {
        $map: {
          input: "$variants",
          as: "variant",
          in: {
            $mergeObjects: [
              "$$variant",
              {
                items: {
                  $filter: {
                    input: "$$variant.items",
                    cond: { $eq: ["$$this._id", "62be0271d373b2f2fc1826a5"] }
                  }
                }
              }
            ]
          }
        }
      }
    }
  },
  // 投影去掉所有无关字段，不需要根文档_id就设为0，需要保留就改成1
  {
    $project: {
      _id: 0,
      variants: 1
    }
  }
])

返回结果示例 #

针对你给出的测试文档，查询返回结果完全符合需求，无多余字段：

{
  "variants": [
    {
      "color_name": "Green",
      "items": [
        {
          "size": "M",
          "price": 100,
          "quantity": 2,
          "_id": "62be0271d373b2f2fc1826a5"
        }
      ],
      "_id": "62be0271d373b2f2fc1826a3"
    }
  ]
}

优化提示 #

• 可以提前给variants._id字段创建索引，配合第一步的$match可以直接命中索引过滤文档，避免全表扫描，百万级以上数据量性能提升非常明显
• 如果不需要把结果包裹在variants数组下，可以在管道最后追加两个阶段，直接返回目标variant对象：

  { $unwind: "$variants" },
  { $replaceRoot: { newRoot: "$variants" } }
  

• 不要尝试用find的$elemMatch投影实现该需求：$elemMatch仅能返回外层数组匹配的第一个元素，无法对匹配到的元素内部的二级数组做二次过滤，达不到过滤内层items的效果。

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