递归函数中Promise实现：文件夹分析函数超时问题解决

问题背景 #

需要实现一个analyze函数，按Node.js回调模式返回文件夹统计信息（总文件数、总子文件夹数、文件类型计数）。现有实现因异步递归处理不当，导致测试超时（测试超时时间2000ms，代码硬等4000ms），同时存在错误处理、计数逻辑不可靠的问题。

报错信息：

Timeout of 2000ms exceeded. For async tests and hooks, ensure "done()" is called; if returning a Promise, ensure it resolves.
(/tmp/builds/autocode_11661/test/analyze.js)

核心问题分析 #

1. 异步操作未正确等待：原代码用fs.readdir异步读取目录，但递归调用readDirectory时没有等待子目录处理完成，靠setTimeout猜测完成时间，完全不可靠，且测试超时时间短于设置的等待时间。
2. 错误处理错误：fs.readdir的错误直接throw，不会传给回调函数，导致错误无法被测试捕获。
3. 文件类型计数逻辑冗余：先按目录收集类型，再在超时后合并重复项，效率低且容易出错。

修正后的实现代码 #

使用fs.promises（Node.js v10+支持）将异步操作Promise化，通过async/await处理递归，确保所有异步操作完成后再调用回调：

const fs = require("fs").promises;
const path = require("path");

class FileTypeInformation {
  constructor(fileExtension, fileCount) {
    this.fileExtension = fileExtension;
    this.fileCount = fileCount;
  }
}

async function analyzeDirectory(currentPath, result) {
  try {
    const entries = await fs.readdir(currentPath, { withFileTypes: true });
    
    for (const entry of entries) {
      const fullPath = path.join(currentPath, entry.name);
      
      if (entry.isDirectory()) {
        result.totalSubFolders++;
        await analyzeDirectory(fullPath, result);
      } else {
        result.totalFiles++;
        // 处理文件扩展名：无扩展名的情况设为''，否则统一小写带点
        const ext = path.extname(entry.name).toLowerCase();
        // 更新类型计数
        const existingType = result.fileTypeMap.get(ext);
        if (existingType) {
          existingType.fileCount++;
        } else {
          result.fileTypeMap.set(ext, new FileTypeInformation(ext, 1));
        }
      }
    }
  } catch (err) {
    throw err;
  }
}

function analyze(folderPath, callback) {
  const result = {
    totalFiles: 0,
    totalSubFolders: 0,
    fileTypeMap: new Map() // 用Map统一维护类型计数，避免重复项
  };

  analyzeDirectory(folderPath, result)
    .then(() => {
      // 将Map转为要求的数组格式
      result.fileTypesInformation = Array.from(result.fileTypeMap.values());
      delete result.fileTypeMap;
      callback(null, result);
    })
    .catch((err) => {
      callback(err, undefined);
    });
}

module.exports = analyze;

关键改进点 #

1. 异步流程可控：用async/await等待所有子目录递归处理完成，彻底消除setTimeout的依赖。
2. 正确的错误处理：所有异步操作的错误通过Promise的catch捕获，最终传给回调函数。
3. 高效的类型计数：用Map实时维护文件类型计数，无需后续合并，避免重复项。
4. 路径处理更可靠：用path.join拼接路径，避免手动拼接的跨平台问题。
5. 性能优化：用readdir的withFileTypes选项，无需额外调用stat判断文件类型，减少IO操作。

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