API Gateway WebSocket API与私有ALB私有集成时，如何重写目标路径以关联特定后端服务

针对你遇到的问题——WebSocket API Gateway私有集成只能指定ALB ARN，无法直接通过URI路径关联后端服务——这里有几个实用的解决方案，结合你的场景详细说明：

方案1：API Gateway集成路径重写 + ALB路径规则 #

这是最贴近你需求的直观路径路由方案，核心思路是在API Gateway集成阶段，把WebSocket路由转换为带特定业务路径的HTTP请求，再通过ALB的路径匹配规则转发到对应服务的目标组。

具体操作： #

1. 修改API Gateway集成配置：在指定ALB监听器ARN的基础上，通过RequestParameters添加路径重写参数，将WebSocket路由映射到后端服务的专属路径。比如针对$connect路由，把请求路径重写为/My.Service/connect：

   connectIntegration:
     Type: AWS::ApiGatewayV2::Integration
     Properties:
       ApiId: !Ref websocketApiGateway
       Description: Websocket $connect integration
       IntegrationType: HTTP_PROXY
       IntegrationMethod: POST
       IntegrationUri: !ImportValue my-alb-http-listener-id
       # 添加路径重写参数，将请求指向后端服务的特定路径
       RequestParameters:
         "integration.request.path.proxy": "'My.Service/connect'"
         "integration.request.header.domainName": "context.domainName"
         "integration.request.header.stage": "context.stage"
         "integration.request.header.connectionId": "context.connectionId"
       PayloadFormatVersion: 1.0
       ConnectionType: VPC_LINK
       VpcLinkId: !Ref yourExistingVpcLink # 关联已有的VPC Link
   

2. 配置ALB路径规则：在你的ALB HTTP监听器中添加规则，匹配路径前缀/My.Service/connect，并转发到对应服务的目标组。示例CloudFormation片段：

   albConnectRule:
     Type: AWS::ElasticLoadBalancingV2::ListenerRule
     Properties:
       ListenerArn: !ImportValue my-alb-http-listener-id
       Priority: 10
       Conditions:
         - Field: path-pattern
           Values: ["*/My.Service/connect"]
       Actions:
         - Type: forward
           TargetGroupArn: !Ref myConnectServiceTargetGroup # 对应你的连接服务目标组
   

方案2：基于自定义请求头的ALB路由（优化你的临时Workaround） #

如果暂时无法调整路径规则，可以通过在API Gateway集成中注入自定义请求头，让ALB根据头信息精准路由到对应服务，步骤如下：

1. 在API Gateway集成中添加自定义请求头：

   connectIntegration:
     Type: AWS::ApiGatewayV2::Integration
     Properties:
       # 保留原有配置...
       RequestParameters:
         "integration.request.header.X-WebSocket-Action": "'connect'"
         "integration.request.header.X-Backend-Service": "'My.Service'"
         # 保留原有header参数...
   

2. 配置ALB监听器规则匹配请求头：

   albConnectRule:
     Type: AWS::ElasticLoadBalancingV2::ListenerRule
     Properties:
       ListenerArn: !ImportValue my-alb-http-listener-id
       Priority: 10
       Conditions:
         - Field: http-header
           HttpHeaderConfig:
             HttpHeaderName: X-Backend-Service
             Values: ["My.Service"]
         - Field: http-header
           HttpHeaderConfig:
             HttpHeaderName: X-WebSocket-Action
             Values: ["connect"]
       Actions:
         - Type: forward
           TargetGroupArn: !Ref myConnectServiceTargetGroup
   

方案3：使用AWS Cloud Map服务发现（适合动态微服务架构） #

根据你提到的IntegrationUri支持Cloud Map服务ARN的特性，可以通过Cloud Map自动发现后端服务，还能通过查询参数定位特定服务，适合服务实例动态增减的场景：

1. 将后端服务注册到Cloud Map：先在Cloud Map中创建命名空间和服务，把你的WebSocket相关服务实例注册到对应服务下。
2. 修改API Gateway集成的IntegrationUri：使用Cloud Map服务的ARN，并添加查询参数指定目标服务：

   connectIntegration:
     Type: AWS::ApiGatewayV2::Integration
     Properties:
       ApiId: !Ref websocketApiGateway
       IntegrationType: HTTP_PROXY
       IntegrationMethod: POST
       # 使用Cloud Map服务ARN，通过查询参数定位特定服务
       IntegrationUri: !Sub "arn:aws:servicediscovery:${AWS::Region}:${AWS::AccountId}:service/srv-xxxxxx?service=My.Service&action=connect"
       ConnectionType: VPC_LINK
       VpcLinkId: !Ref yourExistingVpcLink
       RequestParameters:
         # 保留必要参数...
       PayloadFormatVersion: 1.0
   

   API Gateway会通过DiscoverInstances API根据查询参数找到对应的服务实例，自动将请求转发到ALB后的目标服务。

这三个方案里，方案1最贴合你想要的“直观路由方式”，方案2适合快速临时适配，方案3则更适合动态微服务架构。你可以根据自己的部署场景选择最合适的方式。

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