MongoDB聚合管道中递归展平嵌套子文档的完整方案

本文详解如何使用 MongoDB 5.0+ 的 $function 聚合阶段，通过内联 JavaScript 实现任意深度嵌套 kids 数组的递归展平，生成扁平化的一维 _id 列表，适用于视图构建与高性能查询。

本文详解如何使用 mongodb 5.0+ 的 `$function` 聚合阶段，通过内联 javascript 实现任意深度嵌套 `kids` 数组的递归展平，生成扁平化的一维 `_id` 列表，适用于视图构建与高性能查询。

在 MongoDB 中处理树形结构（如父子嵌套、组织架构、评论回复链）时，常需将深层嵌套的子文档（如 kids.kids.kids...）一次性展开为单层数组。原生聚合操作符（如 $unwind、$reduce、$map）虽强大，但不支持真正的递归遍历——因为嵌套深度未知且动态变化，静态展开（如连续 $unwind + $group）既不可靠也不可维护。

此时，MongoDB 5.0 引入的 $function 聚合阶段成为关键解法：它允许在聚合管道中安全执行内联 JavaScript 函数，实现真正递归逻辑，且结果可直接用于物化视图（create view）或后续聚合阶段。

✅ 推荐方案：使用 $function 递归展平

以下聚合管道将原始嵌套文档中的所有后代 _id 提取为扁平化的 kids 数组（仅保留 _id 字段，符合目标结构）：

db.collection.aggregate([  {    $set: {      kids: {        $function: {          body: "function flattenKids(kids, acc) { " +                "  if (!Array.isArray(kids) || kids.length === 0) return acc; " +                "  for (let child of kids) { " +                "    acc.push({ _id: child._id }); " +                "    flattenKids(child.kids || [], acc); " +                "  } " +                "  return acc; " +                "}",          args: ["$kids", []],          lang: "js"        }      }    }  }])

? 关键说明：

• body：定义纯函数 flattenKids(kids, acc)，接收当前层级子数组和累加器（初始为空数组），递归压入每个节点的 {_id: ...}。
• args: ["$kids", []]：将根级 kids 字段和空数组传入，作为递归起点。
• lang: "js"：指定执行环境为 JavaScript（需 MongoDB 启用 JS 引擎，生产环境默认启用）。
• $set 替换原 kids 字段，输出即为目标格式——无嵌套、无冗余字段。

⚠️ 注意事项：

• 安全性：$function 在启用了 javascriptEnabled: true 的部署中可用（MongoDB Atlas 默认开启；自托管需确认 --javascriptEnabled 配置）。
• 性能：适用于中等数据量（单文档嵌套深度 ≤ 100，后代总数 ≤ 数万）。超大规模建议预计算并持久化 ancestors 或 path 字段。
• 兼容性：仅支持 MongoDB ≥ 5.0。4.x 及更早版本需改用应用层递归或 $graphLookup（但后者要求明确字段名且不支持任意深度“展平”，仅适合查找路径）。

? 替代思路（非递归，有限深度）

若因安全策略禁用 JS，且已知最大嵌套深度（如 ≤ 3 层），可组合 $reduce + $concatArrays 分层展开：

{  $set: {    kids: {      $reduce: {        input: {          $concatArrays: [            "$kids",            { $ifNull: [{ $reduce: {                input: "$kids.kids",                initialValue: [],                in: { $concatArrays: ["$$value", "$$this"] }              }}, []] },            { $ifNull: [{ $reduce: {                input: "$kids.kids.kids",                initialValue: [],                in: { $concatArrays: ["$$value", "$$this"] }              }}, []] }          ]        },        initialValue: [],        in: {          $cond: {            if: { $gt: [{ $size: "$$this" }, 0] },            then: { $concatArrays: ["$$value", [{ _id: "$$this._id" }]] },            else: "$$value"          }        }      }    }  }}

⚠️ 此方式代码冗长、难以维护，且无法覆盖未知深度，强烈推荐优先采用 $function 方案。

✅ 总结

• 使用 $function 是 MongoDB 中实现任意深度递归展平最简洁、可靠、可复用的方法；
• 输出结构严格匹配需求：根文档 _id 不变，kids 变为扁平化对象数组，每个对象仅含 _id；
• 该聚合可直接封装为只读视图（db.createView("flattened_tree", "collection", [...])），供下游服务统一消费；
• 开发阶段建议在 Mongo Playground 验证逻辑，上线前务必进行压力测试。

通过这一方案，你不仅能解决当前嵌套展平问题，更为未来处理复杂层次关系（如 BOM 表、权限树、分类目录）奠定了可扩展的数据处理基础。