Node.js中如何等待异步Stream执行完成后再返回函数

嗨，我明白你的问题了——你给函数加了async，但用await调用时根本等不到Stream执行完就继续走了，对吧？核心问题在于：你的async函数没有返回一个能让await等待的Promise，async关键字只是让函数内部可以用await，但如果函数本身没返回Promise，它会默认返回一个已resolve的空Promise，自然不会等Stream的事件触发。

下面给你两种场景的解决方案：

场景1：client.delete是同步操作 #

如果client.delete是同步执行的，我们只需要把整个Stream逻辑包裹在一个Promise里，在end事件触发时resolve，error事件触发时reject即可：

const removeMapping = async function (query) {
  return new Promise((resolve, reject) => {
    const stream = query.foreach(); // 注：这里是不是拼写错了？通常是forEach驼峰写法，按你的代码保留
    stream.on('data', (record) => {
      client.delete(record);
    });
    stream.on('error', (error) => {
      console.error('Stream执行出错:', error);
      reject(error); // 将错误抛出，让调用方可以捕获处理
    });
    stream.on('end', () => {
      console.log("completed");
      resolve(); // 这里resolve后，await才会结束等待
    });
  });
};

现在你用await removeMapping(yourQuery)时，就会等到Stream的end事件触发后，才继续执行后续代码。

场景2：client.delete是异步操作（返回Promise） #

如果client.delete是异步的（比如返回Promise），要注意：Stream的end事件只是表示没有更多数据要读取了，但之前data事件里触发的delete操作可能还没执行完。这时候需要收集所有delete的Promise，等全部完成后再resolve：

const removeMapping = async function (query) {
  return new Promise((resolve, reject) => {
    const deletePromises = [];
    const stream = query.foreach();

    stream.on('data', (record) => {
      // 假设client.delete返回Promise，将其收集起来
      const deletePromise = client.delete(record);
      // 如果不想因为单个delete失败导致整个流程终止，可以在这里单独捕获错误
      // deletePromise.catch(err => console.error('单条记录删除失败:', record, err))
      deletePromises.push(deletePromise);
    });

    stream.on('error', (error) => {
      console.error('Stream执行出错:', error);
      reject(error);
    });

    stream.on('end', async () => {
      try {
        // 等待所有异步删除操作完成
        await Promise.all(deletePromises);
        console.log("completed");
        resolve();
      } catch (err) {
        console.error('删除操作批量失败:', err);
        reject(err);
      }
    });
  });
};

这样就能确保所有异步删除操作都完成后，函数才会resolve，await也能正确等待整个流程结束。

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