如何通过AWS API Gateway实现基于请求头的路由及响应码问题解决

一、用API Gateway原生路由规则实现（推荐） #

这种方式无需额外Lambda，性能更优，适合简单的请求头匹配场景：

1. 配置后端集成：
   • 若目标是EC2，建议将EC2加入ALB目标组，通过ALB转发请求（避免EC2 IP变动问题），再在API Gateway中创建HTTP集成，指向ALB的对应目标组地址。
   • 若直接对接特定服务，创建对应的HTTP/REST集成即可。
2. 添加路由条件：
   • 在API Gateway的路由配置中，为每个目标添加单独的路由，设置路由条件匹配请求头。比如：
     • 路由1：条件设为 $request.header.X-Target = "GroupA"，绑定到GroupA对应的集成。
     • 路由2：条件设为 $request.header.X-Target = "GroupB"，绑定到GroupB对应的集成。
3. 设置默认路由（可选）：添加无匹配条件的路由，处理未匹配请求头的情况，返回400或自定义响应。

二、Lambda自定义路由的响应码问题修复 #

如果必须用Lambda实现复杂路由逻辑，问题出在Lambda返回格式不符合API Gateway代理集成要求，导致响应码无法正确传递。需按以下规范返回：

正确的Lambda返回格式 #

Lambda作为代理集成时，必须返回包含statusCode、headers、body的JSON结构：

{
  "statusCode": 200,
  "headers": {"Content-Type": "application/json"},
  "body": "{\"msg\": \"success\"}",
  "isBase64Encoded": false
}

示例代码（Python） #

以下是转发请求并完整传递后端响应码的示例：

import requests

def lambda_handler(event, context):
    # 获取请求头中的目标标识
    target_header = event["headers"].get("X-Target-Group")
    
    # 根据请求头映射目标地址
    target_map = {
        "GroupA": "https://alb-group-a.example.com",
        "GroupB": "https://alb-group-b.example.com"
    }
    target_url = target_map.get(target_header)
    
    if not target_url:
        return {
            "statusCode": 400,
            "body": "Invalid X-Target-Group header"
        }
    
    # 转发请求到目标后端
    try:
        response = requests.request(
            method=event["httpMethod"],
            url=f"{target_url}{event['path']}",
            headers={k: v for k, v in event["headers"].items() if k.lower() != "host"},
            params=event.get("queryStringParameters"),
            data=event.get("body")
        )
    except Exception as e:
        return {
            "statusCode": 500,
            "body": str(e)
        }
    
    # 构造符合要求的响应，完整传递后端的状态码、头和内容
    return {
        "statusCode": response.status_code,
        "headers": dict(response.headers),
        "body": response.text,
        "isBase64Encoded": False
    }

关键注意事项 #

• 权限配置：如果后端在VPC内，需将Lambda配置到对应VPC，并赋予访问EC2/ALB的IAM权限。
• 头信息处理：转发时移除原请求的host头，避免与后端的host冲突。
• 日志排查：若仍有问题，查看Lambda的CloudWatch日志，确认是否正确捕获了后端的响应码。

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