如何将AWS SAM Node.js20.x Lambda登录API执行时间降至2秒内

当前状态：基于Node.js 20.x的Lambda登录API，冷启动耗时6秒，热启动平均3秒，目标将响应时间优化至2秒以内，已尝试调整MemorySize到1024未获提升。

以下是针对性优化方案：

一、数据库连接复用优化 #

每次请求内部调用await DBConn()会重复建立数据库连接，这是冷启动和热启动耗时的核心原因之一。Lambda容器复用期间，全局变量会被保留，因此可以将连接初始化移到函数外部：

// 全局变量保存数据库连接实例
let dbInstance;

// 初始化连接（仅在首次调用或连接断开时执行）
async function getDBInstance() {
  if (!dbInstance) {
    dbInstance = await DBConn();
  }
  return dbInstance;
}

export const loginHandler = catchTryAsyncErrors(async (event) => {
  const headers = generateCorsHeaders();
  // 复用已有的连接实例
  const DB = await getDBInstance();

  // 后续业务逻辑保持不变
  const requestBody = JSON.parse(event.body);
  // ...
});

同时检查DBConn实现，确保启用MongoDB连接池（默认已开启），并配置合理的池大小（如poolSize: 5），避免频繁创建新连接。

二、依赖包瘦身与替换 #

Node.js Lambda冷启动慢往往源于node_modules体积过大，加载耗时久：

• 剔除无用依赖：部署前执行npm prune --production仅保留生产依赖，或用工具（如depcheck）分析并删除未使用的包。
• 打包优化：使用ESBuild将代码和必要依赖打包为单个文件，减少文件加载数量。修改SAM配置的BuildMethod为esbuild：

  LoginFunction:
    Type: AWS::Serverless::Function
    Properties:
      # ...其他配置
    Metadata:
      BuildMethod: esbuild
      BuildProperties:
        Minify: true
        Bundle: true
        Target: "node20"
        EntryPoints:
          - app.js
  

• 替换重依赖：将bcryptjs替换为@node-rs/bcrypt（Rust编写，性能提升3-5倍且体积更小），JWT库jsonwebtoken替换为jose（更安全且性能更优）。

三、Lambda配置调优 #

• 优化内存与CPU：Lambda的CPU与内存线性关联，更高内存对应更多CPU资源。测试2048MB或3072MB内存的性能表现——CPU密集型操作（如bcrypt密码比对、JWT生成）会因更多CPU资源显著提速。
• 启用SnapStart：针对Node.js 18+的Lambda，SnapStart通过发布时创建函数快照，冷启动时快速恢复，可将冷启动时间降低70%以上。配置方式：

  LoginFunction:
    Type: AWS::Serverless::Function
    Properties:
      # ...其他配置
      SnapStart:
        ApplyOn: PublishedVersions
  

• 切换至Arm64架构：Graviton2（arm64）架构的Lambda性能比x86_64高约20%，成本更低。修改架构配置：

  Architectures:
    - arm64
  

四、业务逻辑异步化与优化 #

• 异步处理非核心逻辑：管理员登录日志插入属于非核心同步逻辑，可改为通过SQS异步发送，避免等待数据库插入完成再返回响应：

  if (user.role === "Admin") {
    const logData = {
      ip: event?.requestContext?.identity?.sourceIp,
      user: new ObjectId(user?._id),
      activity: "ADMIN_LOGIN",
      msg: `${user?.userName} login into Admin Site`,
      createdAt: new Date(),
      updatedAt: new Date(),
    };
    // 无需await，异步发送日志
    sqs.sendMessage({
      QueueUrl: process.env.LOG_QUEUE_URL,
      MessageBody: JSON.stringify(logData)
    }).catch(err => console.error('日志发送失败:', err));
  }
  

• 替换验证库：将loginValidation使用的验证库（如joi）替换为zod，zod的验证性能比joi高2-3倍，减少请求校验耗时。

五、API Gateway类型切换 #

使用HTTP API替代REST API，HTTP API更轻量，延迟更低，能有效减少端到端的响应时间。修改SAM事件配置：

Events:
  login:
    Type: HttpApi
    Properties:
      Path: /api/user/login
      Method: post

内容的提问来源于stack exchange，提问作者حمزہ لیاقت