MongoDB 聚合管道：递归展平嵌套 kids 字段的完整方案

本文介绍如何使用 MongoDB 5.0+ 的 $function 聚合阶段，通过内联 JavaScript 实现任意深度嵌套 kids 数组的递归展平，生成单层结构的 _id 列表，适用于构建只读视图或简化下游处理。

本文介绍如何使用 mongodb 5.0+ 的 `$function` 聚合阶段，通过内联 javascript 实现任意深度嵌套 `kids` 数组的递归展平，生成单层结构的 `_id` 列表，适用于构建只读视图或简化下游处理。

在 MongoDB 中处理深度未知的嵌套文档（如树形结构的 kids 字段）时，原生聚合操作符（如 $unwind、$reduce、$map）难以直接支持真正的递归遍历——因为它们不具备动态调用自身的能力。虽然可通过多级 $reduce + $concatArrays 手动展开固定层数（如 3 层），但无法应对任意深度，且易导致结构混乱（如数组嵌套数组）。

此时，MongoDB 5.0 引入的 $function 聚合阶段提供了关键能力：允许在聚合管道中安全执行内联 JavaScript 函数，从而实现真正的递归逻辑。该方案无需外部应用介入，可直接用于创建 materialized view 或聚合视图。

以下是一个生产就绪的聚合管道示例：

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

✅ 关键说明：

• body 中定义了递归函数 drill：接收当前层级 kids 数组和累积结果 out，将每个节点的 {_id} 推入结果，并对其子 kids（若存在）继续递归；
• 使用 elem.kids ?? [] 安全访问子字段，避免 undefined 报错；
• args: ["$kids", []] 将根级 kids 字段与空数组传入，作为递归起点；
• $set 直接覆写原始 kids 字段，输出即为扁平化结构。

⚠️ 注意事项：

• $function 需要 MongoDB 5.0+ 且服务器启用 JavaScript 引擎（默认开启，但某些云托管服务如 Atlas Serverless 可能禁用，请确认）；
• JavaScript 执行受 maxTimeMS 和内存限制约束，不建议用于超深（>100 层）或超宽（单节点数万子项）树结构；
• 该操作不可索引优化，属于计算密集型；如高频查询，建议在写入时预计算扁平化字段（如 allKidsIds），或使用变更流维护物化视图；
• 若需保留原始嵌套结构同时新增扁平字段，可将 $set 改为 $addFields 并指定新字段名（如 flattenedKids）。

最终输出严格符合预期：顶层文档保持不变，kids 数组变为仅含 {_id} 对象的一维列表，层级信息虽丢失但满足“展平”核心诉求。此方案简洁、可维护，并可无缝集成至 MongoDB View 定义中，是目前处理动态深度嵌套结构最实用的聚合级解法。