如何在AWS API Gateway HTTP集成中用Lambda处理请求响应头？

方案1：利用Lambda授权器（自定义授权器）填充请求头 #

Lambda授权器虽主打鉴权，但可借其执行逻辑完成请求头的动态填充：

• 编写Lambda函数：在函数内查询数据库获取目标头信息，返回包含context字段的响应（即使不需要鉴权，也要返回principalId和允许访问的policyDocument），示例响应：

{
  "principalId": "dummy-user",
  "policyDocument": {
    "Version": "2012-10-17",
    "Statement": [
      {
        "Action": "execute-api:Invoke",
        "Effect": "Allow",
        "Resource": "arn:aws:execute-api:region:account-id:api-id/stage/method/resource-path"
      }
    ]
  },
  "context": {
    "custom-header": "value-from-db"
  }
}

• 在API Gateway中配置该Lambda授权器，关联到目标API方法。
• 进入HTTP集成的集成请求设置，添加请求头映射：将$context.authorizer.custom-header映射到后端需要的请求头（比如X-Custom-Header）。
• 注意：可开启授权器缓存，减少Lambda执行次数和数据库查询压力，降低成本。

方案2：使用API Gateway请求转换Lambda #

API Gateway支持在集成请求阶段通过Lambda修改请求内容，专门用于处理这类自定义转换需求：

• 创建Lambda函数，接收API Gateway的请求事件（结构参考API Gateway代理集成事件），在函数内查询数据库，将获取的头信息添加到headers字段中，返回修改后的事件。
• 在API Gateway的目标方法集成设置中，找到请求转换选项，选择“Lambda”作为转换类型，指定你的Lambda函数。
• 配置完成后，API Gateway会先调用该Lambda处理请求，再将修改后的请求转发到后端HTTP服务。

方案3：CloudFront + Lambda@Edge（适合全局架构） #

如果你的API已通过CloudFront分发，可借助Lambda@Edge在边缘节点处理请求头：

• 创建Lambda@Edge函数（需部署到us-east-1区域），绑定到CloudFront的Origin Request事件。
• 在函数内查询数据库，将目标头信息添加到请求头中，返回修改后的请求。
• 这种方式的优势是请求在边缘节点处理，减少回源延迟，适合多区域用户访问的场景。

各方案对比 #

• 方案1：无需额外组件，复用授权器机制，但需处理鉴权相关的返回逻辑，适合已有授权需求或轻量场景。
• 方案2：最直接的请求转换方案，逻辑清晰，无需处理鉴权逻辑，推荐优先考虑。
• 方案3：适合CloudFront架构，边缘处理更高效，但增加了架构复杂度。

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