MongoDB聚合：如何建模集合并生成嵌套树形结构？

一、数据存储方式 #

采用父引用模型，每个文档包含核心字段：

• _id：节点唯一标识
• label：节点名称（如分类名）
• parent：父节点的_id（根节点的parent设为null或直接省略）

这种存储模式的优势：

• 新增、修改节点仅需操作单个文档，性能高效
• 可通过_id快速定位任意层级的文档
• 完全匹配你提供的扁平数据集结构

示例单文档结构：

{ "_id": "a", "label": "Cat", "parent": null }
{ "_id": "b", "label": "Big cat", "parent": "a" }
{ "_id": "c", "label": "Lion", "parent": "b" }

二、聚合管道生成嵌套树形结构 #

通过$graphLookup递归获取所有后代节点，再用$function自定义逻辑将扁平数组转换为多层嵌套结构，完整聚合管道如下：

db.collection.aggregate([
  // 筛选根节点（如需查询指定节点的子树，修改为{ _id: "目标节点ID" }即可）
  { $match: { parent: null } },
  // 递归查询当前节点的所有后代
  {
    $graphLookup: {
      from: "collection",
      startWith: "$_id",
      connectFromField: "_id",
      connectToField: "parent",
      as: "descendants"
    }
  },
  // 将扁平的后代数组转换为嵌套的children结构
  {
    $addFields: {
      children: {
        $function: {
          body: function(descendants) {
            // 构建ID到文档的映射表，快速查找父节点
            const nodeMap = {};
            descendants.forEach(node => {
              nodeMap[node._id] = { ...node, children: [] };
            });
            // 遍历所有节点，将子节点挂载到对应父节点的children数组中
            descendants.forEach(node => {
              if (node.parent && nodeMap[node.parent]) {
                nodeMap[node.parent].children.push(nodeMap[node._id]);
              }
            });
            // 返回当前根节点的子节点列表
            return nodeMap[this._id]?.children || [];
          },
          args: ["$descendants"],
          lang: "js"
        }
      }
    }
  },
  // 移除临时字段descendants，保留最终树形结构
  { $project: { descendants: 0 } }
])

关键步骤说明： #

1. $match：定位目标根节点，支持查询任意节点的完整子树
2. $graphLookup：递归拉取所有后代节点，返回扁平数组
3. $function：通过自定义JavaScript逻辑完成扁平数组到嵌套结构的转换，核心是用映射表快速关联父子节点
4. $project：清理冗余字段，输出符合需求的嵌套树形结构

注意：$function需要MongoDB 4.4及以上版本支持；若使用更低版本，可通过嵌套$reduce和条件判断实现，但代码复杂度会显著提升。

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