Mongoose populate方法能否返回树形结构数据？如何实现？

我有一个基于父引用的树形结构Menu模型，Schema定义如下：

export class Menu {
...
  @Prop({ type: Types.ObjectId, ref: 'Menu' })
  childrens: Menu[];
}
export const MenuSchema = SchemaFactory.createForClass(Menu);

数据库中存储的部分数据为：

[
  {
    _id: ObjectId('menu1'),
    name: 'menu1',
    ...
    childrens: [ObjectId('menu1-1')],
  },
  {
    _id: ObjectId('menu1-1'),
    name: 'menu1-1',
    ...
    childrens: [ObjectId('menu1-1-1')],
  },
  {
    _id: ObjectId('menu1-1-1'),
    name: 'menu1-1-1',
    ...
  }
]

使用this.menuModel.findOne({_id:ObjectId('menu1')}).populate('childrens')查询后，返回结果中menu1-1的childrens字段仍为ObjectId数组而非Menu模型列表。请问populate方法无法构建树形结构吗？如何实现返回树形结构的查询？我曾尝试用aggregate的$graphLookup方法，但似乎无效。

一、关于populate无法构建树形的原因 #

mongoose默认的populate仅支持一级关联填充，只会处理当前层级的关联字段，不会自动递归填充嵌套的childrens字段，所以只能得到根节点的一级子菜单，子菜单的关联字段仍为ObjectId。

二、两种实现树形结构查询的方法 #

1. 递归手动填充 #

通过自定义递归函数，手动逐层填充每个节点的子菜单：

import { Menu } from './menu.entity';

async function populateMenuTree(menu: Menu): Promise<Menu> {
  // 填充当前节点的直接子菜单
  await menu.populate('childrens');
  // 递归处理每个子节点的子菜单
  for (const child of menu.childrens) {
    await populateMenuTree(child);
  }
  return menu;
}

// 调用示例
const rootMenu = await this.menuModel.findOne({ _id: new Types.ObjectId('menu1') });
const fullTreeMenu = await populateMenuTree(rootMenu);

2. 正确配置$graphLookup聚合 #

之前$graphLookup无效大概率是参数配置错误，以下是可正常递归查询的写法：

import { Types } from 'mongoose';

const menuTree = await this.menuModel.aggregate([
  // 匹配根节点
  { $match: { _id: new Types.ObjectId('menu1') } },
  // 递归查询所有层级的子节点
  {
    $graphLookup: {
      from: 'menus', // 注意：这里是数据库实际集合名称，mongoose默认是模型名小写复数
      startWith: '$childrens', // 从当前节点的childrens字段开始查找
      connectFromField: 'childrens', // 子节点用于关联的字段
      connectToField: '_id', // 父节点匹配的字段
      as: 'childrens', // 存储结果的字段名
      depthField: 'depth' // 可选，添加层级字段，方便区分菜单层级
    }
  }
]);

注意事项：

• from参数必须与数据库中的集合名称完全一致，若你自定义了集合名称，需对应修改；
• $graphLookup返回的childrens是包含所有子节点的扁平数组，若需要严格的嵌套树形结构，可在聚合后通过代码递归整理，或使用$unwind+$group等聚合阶段进一步处理。

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