MongoDB中如何从父节点获取完整递归树形结构？

针对你的父子引用集合，要生成嵌套的树形结构，可以结合$graphLookup和递归处理（数据库端或客户端）来实现，以下是两种可行方案：

方案一：数据库端聚合管道（MongoDB 4.4+） #

利用$graphLookup获取所有后代节点，再通过$function递归构建嵌套结构：

db.yourCollection.aggregate([
  // 1. 递归查询所有后代节点，存入扁平数组
  {
    $graphLookup: {
      from: "yourCollection",
      startWith: "$_id",
      connectFromField: "_id",
      connectToField: "parentId",
      as: "descendants"
    }
  },
  // 2. 只保留根节点（parentId为null）
  {
    $match: {
      parentId: null
    }
  },
  // 3. 递归构建嵌套树形结构
  {
    $addFields: {
      children: {
        $function: {
          body: function(descendants) {
            const buildTree = (items, parentId) => {
              return items
                .filter(item => item.parentId?.toString() === parentId?.toString())
                .map(item => ({
                  ...item,
                  children: buildTree(descendants, item._id)
                }));
            };
            return buildTree(descendants, this._id.toString());
          },
          args: ["$descendants"],
          lang: "js"
        }
      }
    }
  },
  // 4. 清理不需要的字段
  {
    $project: {
      descendants: 0,
      parentId: 0
    }
  }
])

步骤说明： #

• $graphLookup：从集合中递归匹配所有子节点，将所有层级的后代存入descendants扁平数组。
• $match：筛选出顶层节点，后续只针对这些节点构建完整树形。
• $function：自定义递归函数buildTree，将扁平的后代数组转换为嵌套结构，每个节点下生成children字段存放其子节点。
• $project：移除临时字段descendants和parentId，输出整洁的树形结构。

方案二：客户端代码处理（兼容所有MongoDB版本） #

如果你的MongoDB版本低于4.4（不支持$function），或者数据量不大，直接在客户端查询所有文档后构建树形更简单：

// 示例：Node.js环境下的处理代码
const allDocs = await db.yourCollection.find().toArray();

// 递归构建树形结构的函数
const buildTree = (items, parentId = null) => {
  return items
    .filter(item => {
      // 处理parentId为null的根节点，以及匹配父ID的子节点
      if (parentId === null) {
        return item.parentId === null;
      }
      return item.parentId?.toString() === parentId.toString();
    })
    .map(item => ({
      ...item,
      children: buildTree(items, item._id)
    }));
};

// 生成树形结构
const resultTree = buildTree(allDocs);
console.log(JSON.stringify(resultTree, null, 2));

优势： #

• 无需依赖数据库版本，实现更灵活。
• 避免数据库端执行JavaScript的性能开销，适合中小规模数据。

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