Koa框架下如何立即返回响应后执行后续耗时代码？

问题描述 #

我正在搭建一个处理大量传入webhook的后端服务，这些webhook对响应时间要求极高。后端基于Koa框架，需要立即返回200响应，之后再执行计算、数据库访问等耗时任务。

尝试过调用next()或await next()，但响应都要等到所有中间件执行完毕后才会发送。已查看过Stack Overflow相关回答，但仍未解决问题。

基础代码如下：

const webhookHandler = async (ctx: Context, next: Next): Promise<void> => {
  console.log('Handling webhook event');

  // 1 校验请求
  const event = ctx.request.body;
  if (!event || !event.type) {
    ctx.status = 400;
    ctx.body = { received: false };
    return;
  }

  // 2 立即返回响应
  ctx.status = 200;
  ctx.body = { received: true };
  next();

  // 3 处理事件（耗时任务）
  switch (event.type) {
    case 'a':
      // 此处为长耗时任务
      break;

    default:
      console.log(`未处理的事件类型: ${event.type}`);
      break;
  }
};

解决方案 #

Koa的响应发送时机是在当前中间件及后续中间件全部执行完成后，直接在中间件内写后续代码会阻塞响应发送。要实现「先返回响应再执行任务」，需把耗时任务从当前请求的执行队列中剥离，放到异步执行链路里。

方法1：用setImmediate实现轻量异步任务 #

把耗时任务包裹在setImmediate中，让它在当前事件循环结束后再执行，Koa就能优先发送响应：

const webhookHandler = async (ctx: Context, next: Next): Promise<void> => {
  console.log('Handling webhook event');

  // 1 校验请求
  const event = ctx.request.body;
  if (!event || !event.type) {
    ctx.status = 400;
    ctx.body = { received: false };
    return;
  }

  // 2 立即返回响应
  ctx.status = 200;
  ctx.body = { received: true };

  // 3 把耗时任务放到事件循环下一轮执行
  setImmediate(() => {
    switch (event.type) {
      case 'a':
        // 长耗时任务：数据库操作、计算等
        console.log('开始处理类型a的事件');
        // 模拟耗时操作
        setTimeout(() => {
          console.log('类型a的事件处理完成');
        }, 3000);
        break;

      default:
        console.log(`未处理的事件类型: ${event.type}`);
        break;
    }
  });

  // 无后续中间件需求可省略next()
};

方法2：用独立任务队列适配高并发场景 #

如果webhook请求量极大，建议用专业任务队列（如BullMQ、Bee-Queue）处理耗时任务，避免阻塞Node.js事件循环：

1. 安装依赖（以BullMQ为例）：

npm install bullmq

2. 创建任务队列与处理器：

// queue.ts
import { Queue, Worker } from 'bullmq';

// 初始化队列
const webhookQueue = new Queue('webhook-processing');

// 配置任务处理器
new Worker('webhook-processing', async (job) => {
  const { event } = job.data;
  switch (event.type) {
    case 'a':
      console.log('处理类型a的事件');
      // 模拟耗时操作
      await new Promise(resolve => setTimeout(resolve, 3000));
      console.log('类型a事件处理完成');
      break;
    default:
      console.log(`未处理的事件类型: ${event.type}`);
      break;
  }
});

export { webhookQueue };

3. 在webhook中间件中投递任务：

import { webhookQueue } from './queue';

const webhookHandler = async (ctx: Context, next: Next): Promise<void> => {
  console.log('Handling webhook event');

  // 1 校验请求
  const event = ctx.request.body;
  if (!event || !event.type) {
    ctx.status = 400;
    ctx.body = { received: false };
    return;
  }

  // 2 立即返回响应
  ctx.status = 200;
  ctx.body = { received: true };

  // 3 将事件投递到任务队列异步处理
  await webhookQueue.add('process-event', { event });
};

关键注意点 #

• 绝对不要在当前中间件中直接写同步耗时代码，也不要用await阻塞式调用耗时任务，否则会卡死响应发送。
• setImmediate适合轻量异步场景，高并发下必须用独立任务队列，避免内存泄漏和事件循环阻塞。
• 无后续中间件执行需求时，无需调用next()，因为后续任务已脱离当前请求生命周期。

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