NodeJS实现Mongoose递归填充模型多字段的方法求助

Solution Code #

exports.getSingle = (req, res, next) => {
  Page.findOne({ _id: req.params.pageId })
    .exec(async (err, page) => {
      if (err || !page) {
        return next(err || 'No page with that _id');
      }

      // 通用递归填充函数：支持多字段与跨模型填充
      const recursivePopulate = async (Model, docId, fieldMap) => {
        const doc = await Model.findById(docId);
        if (!doc) return null;

        // 遍历所有需要填充的字段
        for (const [field, childModel] of Object.entries(fieldMap)) {
          if (!doc[field] || !doc[field].length) continue;

          // 并行递归填充当前字段下的所有子文档
          const populatedChildren = await Promise.all(
            doc[field].map(childId => recursivePopulate(childModel, childId, getFieldMap(childModel)))
          );
          // 过滤不存在的子文档，避免返回null值
          doc[field] = populatedChildren.filter(Boolean);
        }

        return doc;
      };

      // 根据模型动态获取字段-模型映射规则
      const getFieldMap = (Model) => {
        switch (Model.modelName) {
          case 'Page':
            return {
              children: Page,
              modules: Module
            };
          case 'Module':
            return {
              modules: Module
            };
          default:
            return {};
        }
      };

      // 启动递归填充流程
      const populatedPage = await recursivePopulate(Page, page._id, getFieldMap(Page));
      res.json({ error: null, data: populatedPage });
    });
};

思路解析 #

1. 重构通用递归逻辑
   原来的populate函数硬绑定了children字段和Page模型，完全不具备扩展性。我们把它改成recursivePopulate，让它接收三个核心参数：

• 当前操作的Mongoose模型（比如Page或Module）
• 要填充的文档ID
• 字段-模型映射表（告诉函数每个字段需要用哪个模型填充）

2. 动态字段映射配置
   新增getFieldMap函数，根据当前模型返回对应的填充规则：

• 对于Page模型，需要递归填充children（子Page）和modules（子Module）
• 对于Module模型，只需要递归填充modules（子Module）
  后续如果新增模型或字段，只需要在这里添加规则即可，不用修改递归核心逻辑。

3. 多字段批量处理
   在递归函数内部，我们遍历字段映射表的每一项，对每个字段的ID数组用Promise.all并行处理所有子文档的填充，既保证了效率，又能同时处理多个字段的递归需求。填充完成后替换原文档的对应字段，还过滤了不存在的子文档（避免返回null）。

4. 用Async/Await简化代码
   把原来的Promise链式调用换成async/await，代码结构更直观，调试也更方便，不用嵌套一堆then回调。

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