在AWS Rust Lambda Runtime中替换默认Request为ApiGatewayCustomAuthorizerRequestTypeRequest

核心解决方案 #

直接基于lambda_runtime和aws_lambda_events库的原生类型实现，无需手动提取字段。Lambda运行时的核心是为handler指定正确的输入、输出泛型参数，对应API Gateway自定义授权器的请求与响应规范。

依赖版本确认 #

确保Cargo.toml中依赖版本匹配（以稳定版为例）：

[dependencies]
lambda_runtime = "0.12"
aws_lambda_events = "0.12"
serde = { version = "1.0", features = ["derive"] }
tokio = { version = "1.0", features = ["macros", "rt-multi-thread"] }

完整示例代码 #

use aws_lambda_events::apigateway::{
    ApiGatewayCustomAuthorizerRequestTypeRequest,
    ApiGatewayCustomAuthorizerResponse,
    ApiGatewayCustomAuthorizerPolicy,
    PolicyDocument,
    Statement
};
use lambda_runtime::{Error, LambdaEvent, service_fn};

#[tokio::main]
async fn main() -> Result<(), Error> {
    let handler_fn = service_fn(authorizer_handler);
    lambda_runtime::run(handler_fn).await?;
    Ok(())
}

async fn authorizer_handler(
    event: LambdaEvent<ApiGatewayCustomAuthorizerRequestTypeRequest>
) -> Result<ApiGatewayCustomAuthorizerResponse, Error> {
    // 直接从原生请求类型中获取授权token和资源ARN
    let auth_token = event.payload.authorization_token.unwrap_or_default();
    let method_arn = event.payload.method_arn.unwrap_or_default();

    // 替换为实际授权逻辑：验证token、查询用户权限等
    let is_authorized = validate_auth_token(&auth_token).await;

    // 生成符合规范的授权策略
    let policy = ApiGatewayCustomAuthorizerPolicy {
        principal_id: Some("authorized-user-id".to_string()),
        policy_document: PolicyDocument {
            version: Some("2012-10-17".to_string()),
            statement: vec![Statement {
                effect: if is_authorized { "Allow".to_string() } else { "Deny".to_string() },
                action: vec!["execute-api:Invoke".to_string()],
                resource: vec![method_arn],
                ..Default::default()
            }],
            ..Default::default()
        },
        ..Default::default()
    };

    Ok(ApiGatewayCustomAuthorizerResponse {
        policy_document: policy.policy_document,
        principal_id: policy.principal_id,
        ..Default::default()
    })
}

// 模拟授权验证逻辑
async fn validate_auth_token(token: &str) -> bool {
    // 实际场景中可实现JWT校验、调用认证服务等逻辑
    token == "valid-auth-token-123"
}

关键说明 #

• 输入类型：使用LambdaEvent<ApiGatewayCustomAuthorizerRequestTypeRequest>作为handler参数，运行时会自动完成事件的反序列化，无需手动解析原始请求。
• 输出类型：返回ApiGatewayCustomAuthorizerResponse，这是库原生的响应类型，完全匹配API Gateway自定义授权器的响应格式要求。
• 原生字段访问：ApiGatewayCustomAuthorizerRequestTypeRequest已封装所有授权器请求字段（如authorization_token、method_arn），直接访问即可，无需冗余的字段提取代码。

常见问题排查 #

• 若出现反序列化错误，检查lambda_runtime与aws_lambda_events的版本是否一致，版本不匹配会导致类型结构不兼容。
• 确保Lambda函数的触发类型为API Gateway自定义授权器，事件格式会自动匹配该类型。

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