AWS Lambda抛出错误时如何终止所有并行运行的async函数

核心实现方案 #

• 直接调用对应运行时的进程终止方法，会立刻中断所有正在执行的异步任务，销毁当前Lambda实例，不会残留任何后台执行逻辑：
  • Node.js运行时：使用process.exit(-1)，该方法会直接终止Node.js进程，无需等待事件循环清空，所有未完成的async任务、IO操作都会被强制中断
  • Python运行时：使用os._exit(-1)，效果和Node.js的process.exit一致，不会等待协程、线程任务执行完成
• 调用时机需要提前到每个并行任务的错误捕获逻辑中，不要等待Promise.all的全局错误回调：Promise.all仅会在第一个任务失败时立刻抛出错误，但已经启动的其他异步任务仍会在后台继续执行，只有在单个任务失败的第一时间调用进程终止方法，才能避免残留任务运行

补充可选方案（支持终止前清理资源） #

如果需要在终止前执行临时资源清理逻辑，可以配合AbortController实现主动任务终止，示例代码如下：

const abortController = new AbortController();
const signal = abortController.signal;

// 封装并行任务，统一处理错误
const wrapTask = async (task) => {
  try {
    // 任务内部需要监听signal事件，收到abort信号时终止自身逻辑
    return await task(signal);
  } catch (err) {
    // 触发所有任务的终止信号
    abortController.abort(err);
    // 执行必要的清理逻辑后终止进程
    await doCleanup();
    process.exit(-1);
  }
}

// 批量提交并行任务
const parallelTasks = [task1, task2, task3].map(t => wrapTask(t));
await Promise.all(parallelTasks);

关键注意事项 #

• 不要使用普通throw Error的方式上报错误：抛出错误只会中断当前执行栈，已经启动的异步IO任务仍然会在事件循环中继续运行，这就是你之前出现流程混乱的根本原因
• 进程终止后当前Lambda实例会被直接销毁，CloudFormation后续触发的回滚Delete事件会请求到新的Lambda实例，不会和已终止的旧实例中的Create任务产生冲突

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