如何在AWS API Gateway WebSocket连接时通过Mock集成返回Connection ID（无需Lambda）

安全阿岳

2026-6-18

问题解答

仅靠API Gateway的Mock集成无法实现客户端连接时自动收到Connection ID的需求，必须依赖Lambda这类后端服务，原因和解决步骤如下：

为什么Mock集成不行？

API Gateway的Mock集成只能返回静态响应，但WebSocket的$connect路由的响应是服务端内部用来确认连接状态的，不会主动推送给客户端。
要主动向WebSocket客户端发送消息，必须调用API Gateway的@connections管理API（具体是POST /@connections/{connectionId}接口），而Mock集成没有执行API调用的能力，只能返回固定内容。

你的尝试失败的原因

启用$default路由的双向通信：这个设置只是允许客户端向服务端发送消息，并不支持服务端主动触发消息推送。
$connect路由加Mock集成：Mock返回的内容不会被推送到客户端，仅用于API Gateway内部判断连接是否成功建立。

正确的实现方式（用Lambda）

给$connect路由配置Lambda集成

在Lambda函数中，从事件参数里提取Connection ID：

// Node.js示例
const connectionId = event.requestContext.connectionId;

使用AWS SDK调用API Gateway Management API，把Connection ID发送给客户端：

const { ApiGatewayManagementApiClient, PostToConnectionCommand } = require("@aws-sdk/client-apigatewaymanagementapi");

exports.handler = async (event) => {
  const connectionId = event.requestContext.connectionId;
  const endpoint = `https://${event.requestContext.domainName}/${event.requestContext.stage}`;
  const client = new ApiGatewayManagementApiClient({ endpoint });

  try {
    await client.send(new PostToConnectionCommand({
      ConnectionId: connectionId,
      Data: JSON.stringify({ connectionId: connectionId })
    }));
  } catch (err) {
    return { statusCode: 500, body: err.stack };
  }

  return { statusCode: 200 };
};