3、中间件设计模式：Go HTTP中间件原理、net/http Handler接口、中间件链式调用

在构建 OAuth2.0 授权码流程时，中间件模式是串联认证、日志、错误恢复等横切关注点的核心架构手段。Go 标准库 net/http 以 Handler 接口为基石，通过函数包装实现了简洁而强大的中间件链式调用。本章深入剖析其原理与设计模式，为后续集成 OAuth2.0 流程奠定基础。

3.1 net/http Handler 接口：中间件的基石

Go 的 HTTP 处理模型围绕 http.Handler 接口构建，该接口仅定义了一个方法：

type Handler interface {
    ServeHTTP(ResponseWriter, *Request)
}

任何实现了 ServeHTTP 方法的类型都可以作为 HTTP 处理器。中间件的本质是一个接收 Handler 并返回新 Handler 的函数，从而在请求处理链中插入额外逻辑。

概念	说明	在 OAuth2.0 中的角色
Handler	处理 HTTP 请求的核心接口	授权端点、令牌端点、受保护资源端点
HandlerFunc	将普通函数适配为 Handler 的适配器	快速定义回调处理函数
ServeMux	请求多路复用器，将 URL 路由到 Handler	分发 /authorize、/token、/callback 等路径
中间件函数	func(http.Handler) http.Handler	日志、速率限制、错误恢复、会话验证

3.2 中间件链式调用模式

中间件链式调用通过函数组合实现。典型的中间件函数签名如下：

// Middleware 类型定义
type Middleware func(http.Handler) http.Handler

// 链式组合函数
func Chain(handler http.Handler, middlewares ...Middleware) http.Handler {
    for i := len(middlewares) - 1; i >= 0; i-- {
        handler = middlewares[i](handler)
    }
    return handler
}

执行顺序遵循“洋葱模型”：请求从最外层中间件进入，逐层向内传递，响应则从内向外逐层返回。

3.3 中间件链式调用流程图

下图展示了包含四个中间件的典型请求处理流程，清晰呈现请求与响应的双向传递路径：

3.4 标准中间件实现示例

以下代码展示了三个在 OAuth2.0 流程中常用的中间件实现：

// 错误恢复中间件：捕获 panic，返回 500 错误
func RecoveryMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        defer func() {
            if err := recover(); err != nil {
                http.Error(w, "Internal Server Error", http.StatusInternalServerError)
                log.Printf("panic recovered: %v", err)
            }
        }()
        next.ServeHTTP(w, r)
    })
}

// 请求日志中间件：记录请求方法与路径
func LoggingMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        start := time.Now()
        next.ServeHTTP(w, r)
        log.Printf("[%s] %s %s", r.Method, r.URL.Path, time.Since(start))
    })
}

// 速率限制中间件：基于令牌桶的简单限流
func RateLimitMiddleware(next http.Handler) http.Handler {
    limiter := rate.NewLimiter(100, 200) // 每秒 100 个请求，突发 200
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if !limiter.Allow() {
            http.Error(w, "Too Many Requests", http.StatusTooManyRequests)
            return
        }
        next.ServeHTTP(w, r)
    })
}

3.5 中间件组合与路由集成

在实际 OAuth2.0 服务中，中间件通常与路由结合使用。以下示例展示了如何为不同端点应用不同的中间件链：

func main() {
    mux := http.NewServeMux()

    // 授权端点：需要会话验证 + 速率限制
    authorizeHandler := Chain(
        handleAuthorize,
        SessionMiddleware,
        RateLimitMiddleware,
    )

    // 令牌端点：需要客户端认证 + 速率限制
    tokenHandler := Chain(
        handleToken,
        ClientAuthMiddleware,
        RateLimitMiddleware,
    )

    // 回调端点：需要日志 + 错误恢复
    callbackHandler := Chain(
        handleCallback,
        LoggingMiddleware,
        RecoveryMiddleware,
    )

    mux.Handle("/authorize", authorizeHandler)
    mux.Handle("/token", tokenHandler)
    mux.Handle("/callback", callbackHandler)

    // 全局中间件：所有请求都经过错误恢复和日志
    globalHandler := Chain(mux, RecoveryMiddleware, LoggingMiddleware)
    http.ListenAndServe(":8080", globalHandler)
}

3.6 中间件设计模式对比

模式	实现方式	适用场景	OAuth2.0 示例
函数式中间件	func(http.Handler) http.Handler	通用横切关注点	日志、限流、超时控制
结构体中间件	实现 Handler 接口的结构体	需要依赖注入或状态管理	数据库会话、缓存客户端
闭包中间件	返回中间件函数的工厂函数	需要配置参数	带阈值的速率限制、自定义日志格式
装饰器模式	在 ServeHTTP 前后插入逻辑	请求/响应修改	添加 CORS 头、压缩响应

3.7 中间件链式调用的软件架构图

下图从架构视角展示了中间件在 OAuth2.0 授权码流程中的分层位置与协作关系：