Cognito登录系统PreToken Lambda调用报Socket超时错误如何解决

算力阿峰

2026-10-1

问题诱因

Lambda响应未正确返回：所有日志正常输出说明业务逻辑已执行完毕，但未按照Cognito要求正确返回事件对象，导致Cognito收不到响应触发超时。Node.js运行时中常见原因包括async handler未加return语句、异步Mongoose操作漏加await、callback写法漏调用回调函数。
Lambda执行超时阈值过低：默认Lambda执行超时为3秒，3-4次跨网Mongoose查询叠加网络延迟很容易超过阈值，Lambda被终止前刚好完成所有日志打印，但未完成响应返回。
Cognito触发器硬超时限制：Cognito对所有触发器有5秒的硬超时上限，即使Lambda配置了更高的超时时间，只要逻辑总耗时超过5秒，Cognito会直接抛出调用超时错误。
Mongoose连接未复用：每次Lambda调用都新建MongoDB连接，连接建立的额外耗时拉长了总执行时长，高并发场景下超时概率会大幅提升。
网络配置异常：Lambda如果部署在VPC内，未配置正确的NAT网关、安全组规则，会导致MongoDB访问请求被拦截，卡在网络层直到超时。

解决方案

修正Lambda响应逻辑：确保所有Mongoose异步操作都加了await，执行完成后正确返回Cognito要求的事件结构，示例代码如下：

// Node.js async handler 示例
exports.handler = async (event) => {
  // 所有数据库操作必须加await
  const userInfo = await UserModel.findOne({ cognitoId: event.request.userAttributes.sub });
  // 写入自定义令牌声明
  event.response.claimsOverrideDetails = {
    claimsToAddOrOverride: {
      customUserInfo: JSON.stringify(userInfo)
    }
  };
  // 必须返回完整event对象
  return event;
};

如果使用callback写法，需确保所有逻辑分支都调用callback(null, event)返回结果，错误场景也要调用callback(error)结束调用。

调整Lambda超时配置：将Lambda执行超时调整到4-5秒，不要超过5秒的Cognito触发器上限，避免无效配置。
复用Mongoose数据库连接：将连接初始化逻辑放在handler函数外部，利用Lambda执行环境复用特性减少连接建立耗时，示例代码如下：

// 连接逻辑放在handler外部，冷启动时仅初始化一次
let dbConnection = null;
const initMongo = async () => {
  if (dbConnection == null) {
    dbConnection = await mongoose.connect(process.env.MONGODB_URI, {
      serverSelectionTimeoutMS: 2000, // 缩短连接超时阈值，避免卡请求
      bufferCommands: false
    });
  }
  return dbConnection;
};

exports.handler = async (event) => {
  await initMongo();
  // 后续业务逻辑
};

优化网络与查询性能：将Lambda与MongoDB部署在同一AWS可用区，配置安全组放行MongoDB访问端口；给查询字段加数据库索引，将多次零散查询合并为单次聚合查询，降低数据库往返耗时。
裁剪非必要逻辑：如果优化后耗时仍然超过5秒，需将非必须写入令牌的逻辑从PreToken触发器中移除，转移到PostAuthentication触发器或者后端服务异步执行。

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