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