Node.js中如何基于多异步文件读取创建Promise？

现有代码如下：

fn1() { [filepath1, filepath2, ..., filepathN].forEach(process); function process(path) { fs.readFile(file, translate); } }

其中translate回调用于处理每个文件的内容。

需求：

1. 修改该函数使其返回Promise，从而支持fn1().then(fn2)的调用方式；
2. 同时提供基于async await的优秀解决方案。

假设条件 #

• fn2()需要等待fn1所用数组中的所有文件路径处理完成后再执行。
• 我的理解：由于fs.readFile()是异步操作，fn1()可能会在所有文件处理完成前返回，这个理解是否正确？

你的理解完全正确！ #

fs.readFile是异步I/O操作，调用后会立刻返回，后台才会去读取文件内容；而forEach只是同步遍历数组，触发每个process调用，但完全不会等待异步操作完成。所以原来的fn1会在所有文件读取甚至处理还没开始的时候就执行完毕返回了，根本没法让fn2等待所有任务完成。

下面给你两种解决方案：

方案一：改造为返回Promise的版本 #

要实现等待所有文件处理完成，核心思路是把每个文件的异步操作包装成Promise，然后用Promise.all()来统一等待所有Promise完成。

Node.js的fs模块已经内置了Promise版本的API（fs.promises.readFile），直接用就很方便，不用自己手动包装。改造后的代码如下：

function fn1() {
  // 把每个文件路径转换成一个处理它的Promise
  const fileProcessingPromises = [filepath1, filepath2, ..., filepathN].map(path => {
    // 先读取文件，拿到内容后传给translate处理
    return fs.promises.readFile(path)
      .then(fileContent => translate(fileContent));
  });

  // 返回一个Promise，它会在所有文件处理完成后才resolve
  return Promise.all(fileProcessingPromises);
}

// 现在就可以直接用.then()调用了
fn1().then(fn2);

关键点说明：

• 用map代替forEach：因为我们需要收集每个异步操作的Promise实例，forEach只是执行操作，不会返回结果，而map可以帮我们生成一个Promise数组；
• Promise.all()：这个方法会等待数组里所有Promise都成功完成后才会resolve，返回的结果是所有translate处理后的结果数组；如果有任何一个文件处理失败，它会立即reject，刚好符合我们需要所有文件都处理完再执行fn2的要求。

方案二：更简洁的async/await版本 #

async/await是Promise的语法糖，写起来更像同步代码，可读性更强。我们可以把fn1改成async函数，内部配合Promise.all使用：

async function fn1() {
  const fileProcessingPromises = [filepath1, filepath2, ..., filepathN].map(async path => {
    // 等待文件读取完成
    const fileContent = await fs.promises.readFile(path);
    // 处理内容并返回结果
    return translate(fileContent);
  });

  // 等待所有文件处理完成
  await Promise.all(fileProcessingPromises);
}

// 调用方式有两种：
// 方式1：继续用.then()
fn1().then(fn2);

// 方式2：在另一个async函数里用await，更直观
async function runTasks() {
  await fn1();
  fn2();
}
runTasks();

额外的错误处理建议：

如果担心文件读取或translate处理出错，可以加上错误捕获：

// Promise版本的错误处理
fn1().then(fn2).catch(error => {
  console.error('文件处理过程出错:', error);
});

// async/await版本的错误处理
async function runTasks() {
  try {
    await fn1();
    fn2();
  } catch (error) {
    console.error('文件处理过程出错:', error);
  }
}
runTasks();

另外，如果translate本身也是异步函数（比如返回Promise），上面的代码完全兼容，因为await会自动等待它完成，map生成的Promise也会包含这个异步处理的过程。

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