AWS API Gateway WebSocket连接元数据传递至断开事件方案

AWS API Gateway WebSocket 原生没有提供可以跨connect、disconnect事件持久化自定义字段的内置上下文对象，你之前尝试修改requestContext不生效是正常的——每次生命周期事件触发Lambda时，API Gateway都会重新构造独立的event对象，你在connect阶段对event做的任何修改都仅在当前Lambda执行周期内有效，不会回传给API Gateway留存，自然不可能带到后续的断开事件里。

要实现你的需求，直接用下面两个经过生产验证的方案就行，不用绕弯路：

这个方案完全不影响你给每个团队配置独立活跃连接表的设计，只需要额外加一张极小的全局索引表做路由：

• 新建一张全局共享的DynamoDB表，主键设为connectionId，仅存三个字段即可：connectionId（主键）、teamDatabaseID、过期时间TTL字段，单条记录大小不到100字节，存储成本可以忽略
• connect阶段逻辑：
  1. 从握手请求的Headers、queryStringParameters里提取TeamDatabaseID
  2. 将连接信息写入对应团队的独立活跃连接表
  3. 同步往全局映射表写入{connectionId: xxx, teamDatabaseID: xxx, ttl: xxx}记录，两步写操作可以放到DynamoDB事务里保证一致性
• disconnect阶段逻辑：
  1. 从event的requestContext里拿到connectionId，先查全局映射表拿到对应的teamDatabaseID
  2. 定位到对应团队的独立活跃连接表，删除该connectionId对应的记录
  3. 顺手删除全局映射表里的这条索引记录即可

• 优化点：给全局映射表配置TTL规则，过期时间设为比业务允许的最大连接空闲时长多24小时即可，就算偶发disconnect事件投递丢失，残留的无效索引也会被自动清理，不会产生脏数据。

如果你不想额外维护一张全局表，可以调整API路由设计，靠API Gateway原生的路径参数透传能力实现：

• 修改$connect路由的资源路径配置，改为带动态参数的形式：/teams/{teamDatabaseID}/connect
• 强制要求客户端建立WebSocket连接时，将teamDatabaseID放在连接URL的路径中，例如wss://your-ws-domain/teams/TEAM00123
• 这种配置下，disconnect事件触发时，event.requestContext.pathParameters会自动携带teamDatabaseID字段，不需要额外查表就能定位到对应团队的连接表

• 注意这个方案的两个硬限制：
  • teamDatabaseID不能放在Header、query参数里传递，必须放在URL路径中，否则路径匹配失败会直接导致握手被拒
  • 如果你使用了自定义域名、多层基础路径映射的配置，上线前必须全流程测试路径参数是否能正常透传到disconnect事件，部分特殊配置下会出现参数丢失的问题。

避坑提醒：不要尝试用Lambda全局变量、执行上下文缓存来存connectionId和teamDatabaseID的映射关系。Lambda实例会被平台随时回收，且同一个实例会并发处理多个不同连接的请求，这种方案不仅缓存命中率极低，还会出现不同团队数据串扰的问题，完全不可用于生产。

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