MongoDB聚合管道递归展平子文档方法

本文深入解析了在 MongoDB 5.0+ 中利用 `$function` 聚合阶段执行内联 JavaScript 实现任意深度嵌套 `kids` 数组的递归展平，将树状结构（如组织架构、评论链、BOM 表）一键转化为扁平化的 `_id` 对象数组，彻底摆脱静态展开的局限与维护噩梦；方案简洁可靠、可直接用于物化视图构建和高性能查询，兼顾灵活性与工程实用性——无论嵌套多深、结构多变，一条聚合管道即可优雅收场。

本文详解如何使用 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 表、权限树、分类目录）奠定了可扩展的数据处理基础。

今天关于《MongoDB聚合管道递归展平子文档方法》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！