使用@vendia/serverless-express在AWS Lambda实现流式响应失败求助

问题根源 #

你当前的实现存在两个核心问题：

1. @vendia/serverless-express 不支持流式响应：该库的设计逻辑是将Express的响应全部缓冲后一次性返回给Lambda执行环境，不会把res.write()的调用实时转发到Lambda的输出流。和serverless-http一样，它适配的是传统完整HTTP响应模型，而非流式输出场景。
2. 未配置Lambda与触发网关的流式响应模式：即使代码逻辑正确，Lambda函数本身需要开启流式响应配置，同时前端触发网关（如HTTP API、Function URL）也必须支持流式传输。

解决方案 #

1. 改用原生Lambda Handler实现流式响应 #

放弃Express适配层，直接编写支持流式的Lambda handler，示例代码如下：

const awslambda = require('@aws-sdk/util-lambda');

exports.handler = awslambda.streamifyResponse(async (event, responseStream, context) => {
  // 设置响应头
  const responseHeaders = {
    "Content-Type": "text/plain",
    "Access-Control-Allow-Origin": "*" // 根据业务需求调整CORS规则
  };

  // 初始化响应流
  responseStream = awslambda.HttpResponseStream.from(responseStream, responseHeaders);

  try {
    // 模拟分批次流式输出
    for (let i = 0; i < 10; i++) {
      const chunk = `Line ${i}\n`;
      responseStream.write(chunk);
      await new Promise(resolve => setTimeout(resolve, 1000));
    }
    responseStream.end();
  } catch (error) {
    console.error("流式输出出错:", error);
    responseStream.destroy(error);
  }
});

2. 配置Lambda函数为流式响应模式 #

• AWS控制台操作：进入Lambda函数的「配置 > 函数代码 > 运行时设置」，将「响应模式」修改为「流式」。
• Serverless Framework配置：在serverless.yml中添加如下配置：

functions:
  streamFunction:
    handler: handler.handler
    runtime: nodejs18.x
    responseMode: stream

3. 配置触发网关支持流式 #

• Function URL：在Lambda函数的「配置 > 函数URL」中，启用URL时将「调用模式」设置为RESPONSE_STREAM。
• HTTP API：创建HTTP API并集成Lambda时，将「负载格式版本」设为2.0，同时开启「响应流式」选项。

保留现有Express路由的折中方案 #

如果需要保留现有Express路由，仅对/stream接口做流式处理，可以在Lambda handler中区分路由逻辑：

const express = require('express');
const awslambda = require('@aws-sdk/util-lambda');
const serverlessExpress = require('@vendia/serverless-express');

const app = express();
// ... 你的现有Express中间件和非流式路由配置

exports.handler = async (event, context) => {
  // 判断是否为流式请求路由
  if (event.requestContext?.http?.path === '/stream') {
    return awslambda.streamifyResponse((event, responseStream) => {
      // ... 此处放入流式输出逻辑
      const responseHeaders = {
        "Content-Type": "text/plain",
        "Access-Control-Allow-Origin": "*"
      };
      responseStream = awslambda.HttpResponseStream.from(responseStream, responseHeaders);
      
      for (let i = 0; i < 10; i++) {
        setTimeout(() => {
          responseStream.write(`Line ${i}\n`);
          if (i === 9) responseStream.end();
        }, i * 1000);
      }
    })(event, context);
  } else {
    // 用serverless-express处理其他非流式路由
    const serverlessHandler = serverlessExpress({ app });
    return serverlessHandler(event, context);
  }
};

关键注意事项 #

• Lambda流式响应的最大持续时间受限于函数超时时间（最长15分钟），需根据业务需求合理调整。
• 流式响应接口无法依赖需要完整请求体的中间件（如body-parser），需确保流式接口的请求处理逻辑独立。
• 测试流式响应时，避免使用默认缓冲完整响应的工具（如部分版本的Postman），建议使用curl命令：curl -N https://your-function-url/stream

内容的提问来源于stack exchange，提问作者Sarthak M. Das