如何使用Mongoose填充MongoDB自引用文档的嵌套children数组？

问题说明 #

我有一个MongoDB的Person集合，每个文档包含children数组字段，存储同集合下其他Person文档的ObjectId引用。需要查询指定Person文档时，自动填充所有嵌套层级的children字段（数据为无环结构，无循环引用）。

Schema定义 #

const personSchema = new mongoose.Schema({
  name: {
    type: String,
    required: true,
  },
  // 其他字段...
  children: [{
    type: mongoose.Schema.Types.ObjectId,
    ref: "Person",
  }],
});

const Person = mongoose.model("Person", personSchema); 

预期输出 #

查询_id为1的文档时，返回结构如下：

{
  "_id": 1,
  "name": "John Doe",
  "children": [
    {
      "_id": 2,
      "name": "Jane Doe",
      "children": [
        {
          "_id": 3,
          "name": "Billy Doe",
          "children": [
            {
              "_id": 4,
              "name": "Bobby Doe"
              // 其他字段...
            }
          ]
        }
      ]
    }
  ]
}

解决方案 #

方法一：递归调用Mongoose populate（适合小规模数据） #

Mongoose默认populate不支持无限层级，可通过自定义递归函数实现：

async function getPersonWithAllChildren(personId) {
  // 查询当前文档并填充第一层children
  const person = await Person.findById(personId).populate('children');
  
  // 递归填充每个子文档的children
  if (person.children.length > 0) {
    person.children = await Promise.all(
      person.children.map(child => getPersonWithAllChildren(child._id))
    );
  }
  
  return person;
}

// 使用示例
const fullPerson = await getPersonWithAllChildren('目标文档ID');

说明：逻辑简单易懂，但嵌套层级深、数据量大时会产生多次数据库查询，性能一般。

方法二：使用MongoDB聚合$graphLookup（推荐，性能更优） #

MongoDB 3.4+支持$graphLookup聚合操作，可一次性查询所有嵌套节点并构建树形结构：

async function getPersonWithAllChildren(personId) {
  // 先获取根文档
  const root = await Person.findById(personId);
  if (!root) return null;

  // 通过$graphLookup获取所有后代节点
  const allDescendants = await Person.aggregate([
    {
      $graphLookup: {
        from: 'people', // 注意：这里是集合实际名称，Mongoose默认是模型名小写复数
        startWith: '$children',
        connectFromField: 'children',
        connectToField: '_id',
        as: 'descendants',
      },
    },
    { $match: { _id: root._id } },
    { $unwind: '$descendants' },
    { $replaceRoot: { newRoot: '$descendants' } },
  ]);

  // 构建ID到文档的映射表
  const idMap = new Map();
  idMap.set(root._id.toString(), { ...root.toObject(), children: [] });
  allDescendants.forEach(desc => {
    idMap.set(desc._id.toString(), { ...desc, children: [] });
  });

  // 关联父子节点的嵌套关系
  idMap.forEach(node => {
    node.children = (node.children || []).map(childId => idMap.get(childId.toString())).filter(Boolean);
  });

  return idMap.get(root._id.toString());
}

// 使用示例
const fullPerson = await getPersonWithAllChildren('目标文档ID');

说明：仅需一次聚合查询，性能远高于递归方法，适合数据量大、层级深的场景；需要手动构建树形结构，逻辑稍复杂。

注意事项 #

1. 集合名称正确性：$graphLookup的from参数需填写MongoDB中实际的集合名称，Mongoose默认将模型名转为小写复数（如Person模型对应people集合），自定义集合名称的话需对应修改。
2. 无环结构前提：确保数据无循环引用，避免构建树形结构时出现死循环。

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