MongoDB 3.4聚合框架：无需多次$lookup递归查询嵌套ID元素

因为你的MongoDB版本是3.4，我们可以通过两种方式实现需求：一种是客户端先扁平化递归对象提取所有elementId，再用聚合查询；另一种是纯聚合框架内处理（依赖临时集合）。下面分别介绍：

方案一：客户端扁平化+聚合查询（推荐） #

这个方法最简单高效，完全不需要编写多层$lookup，只需要先在客户端把递归对象里的所有elementId提取成一个数组，再用聚合的$match一次性查询对应元素。

步骤1：客户端提取所有elementId #

用递归函数遍历你的对象，收集所有层级的elementId：

function extractAllElementIds(recursiveObj) {
  let ids = [recursiveObj.elementId];
  // 遍历子对象递归收集ID
  if (recursiveObj.subObjects && recursiveObj.subObjects.length > 0) {
    recursiveObj.subObjects.forEach(subObj => {
      ids = ids.concat(extractAllElementIds(subObj));
    });
  }
  return ids;
}

// 替换成你的递归对象
const targetObj = {
  elementId: "jh442gd321",
  subObjects: [
    {
      elementId: "kjh34jh4y",
      subObjects: [
        { elementId: "df334n35j5", subObjects: [] }
      ]
    }
  ]
};

const elementIds = extractAllElementIds(targetObj);

步骤2：聚合查询elements集合 #

用$match配合$in操作符，一次性获取所有对应元素：

db.elements.aggregate([
  {
    $match: {
      id: { $in: elementIds }
    }
  }
]);

这个方案逻辑清晰、性能优异，完美贴合你“无需嵌套$lookup”的需求。

方案二：纯聚合框架内处理（依赖临时集合） #

如果必须在MongoDB内部完成所有逻辑，可以借助临时集合和3.4新增的$graphLookup特性来实现，不过这个方法更适合嵌套层数固定的场景：

步骤1：创建临时集合并插入递归对象 #

db.temp_recursive.insertOne(targetObj);

步骤2：用聚合管道递归提取所有elementId并关联elements集合 #

db.temp_recursive.aggregate([
  // 展开第一层子对象
  { $unwind: { path: "$subObjects", preserveNullAndEmptyArrays: true } },
  // 展开第二层子对象（如果有更多层级，需要继续添加对应$unwind）
  { $unwind: { path: "$subObjects.subObjects", preserveNullAndEmptyArrays: true } },
  // 收集所有层级的elementId到数组
  {
    $group: {
      _id: null,
      allIds: {
        $addToSet: {
          $concatArrays: [
            ["$elementId"],
            ["$subObjects.elementId"],
            ["$subObjects.subObjects.elementId"]
          ]
        }
      }
    }
  },
  // 扁平化数组并去重
  { $unwind: "$allIds" },
  { $unwind: "$allIds" },
  { $group: { _id: null, uniqueIds: { $addToSet: "$allIds" } } },
  // 关联elements集合获取对应文档
  {
    $lookup: {
      from: "elements",
      localField: "uniqueIds",
      foreignField: "id",
      as: "matchedElements"
    }
  },
  // 清理结果格式
  { $project: { _id: 0, matchedElements: 1 } }
]);

注意：如果你的递归对象嵌套层数不固定，这个方法需要动态添加$unwind阶段，灵活性不如方案一。

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