如何为AWS Lambda的Go语言实现通用型中间件

要实现兼容不同Lambda handler签名的通用中间件，利用Go 1.18+的泛型特性是最直接且类型安全的方案，替代之前尝试的interface{}方案（后者会导致类型不兼容问题）。

1. 定义泛型Handler类型 #

首先定义支持任意事件（E）和响应（R）类型的泛型Handler函数：

import (
    "context"
    "log"
    "github.com/aws/aws-lambda-go/events"
    "github.com/aws/aws-lambda-go/lambda"
)

// HandlerFunc 泛型处理器类型，支持任意事件E和响应R
type HandlerFunc[E any, R any] func(context.Context, E) (R, error)

2. 实现泛型中间件 #

基于泛型Handler实现通用日志中间件，这个中间件可以适配任何符合Lambda签名的处理器：

// LogMiddleware 通用日志中间件，兼容任意E和R类型的处理器
func LogMiddleware[E any, R any](next HandlerFunc[E, R]) HandlerFunc[E, R] {
    return func(ctx context.Context, event E) (R, error) {
        // 前置日志逻辑：打印事件类型、请求上下文信息等
        log.Printf("Incoming request - Event type: %T, Request ID: %s", 
            event, ctx.Value("lambda.requestId"))
        
        // 调用目标处理器
        resp, err := next(ctx, event)
        
        // 后置日志逻辑：打印响应和错误信息
        if err != nil {
            log.Printf("Request failed - Error: %v", err)
        } else {
            log.Printf("Request succeeded - Response: %+v", resp)
        }
        
        return resp, err
    }
}

3. 在不同处理器上复用中间件 #

WebSocket处理器示例 #

func WsHandler(ctx context.Context, event events.APIGatewayWebsocketProxyRequest) (events.APIGatewayProxyResponse, error) {
    // 业务逻辑：处理WebSocket连接、消息等
    return events.APIGatewayProxyResponse{StatusCode: 200}, nil
}

func main() {
    // 直接将中间件应用到WebSocket处理器
    lambda.Start(LogMiddleware(WsHandler))
}

API Gateway GraphQL处理器示例 #

func GraphqlQueryMutationHandler(ctx context.Context, event events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
    // 业务逻辑：处理GraphQL查询/变更
    return events.APIGatewayProxyResponse{
        StatusCode: 200,
        Body:       `{"data": "result"}`,
        Headers:    map[string]string{"Content-Type": "application/json"},
    }, nil
}

func main() {
    // 复用同一个日志中间件
    lambda.Start(LogMiddleware(GraphqlQueryMutationHandler))
}

关键说明 #

• 泛型会在编译时为每个具体的E（事件类型）和R（响应类型）生成对应的处理器函数，完全符合Lambda的签名要求，不会出现类型不兼容问题。
• 如果中间件需要访问事件的通用字段（比如请求ID、头部信息），可以定义一个通用接口，让事件类型实现该接口（对于AWS官方事件类型，可通过包装类型实现适配）。

内容的提问来源于stack exchange，提问作者Joey Yi Zhao