前置知识: Go、Go、Go

Go 与中间件

14 min中级

Go HTTP 中间件：Handler/HandlerFunc 接口、洋葱模型、Chain 组合律、context 传播、企业级网关实战

Go HTTP 中间件：从洋葱模型到企业级网关

前置知识

学习目标

  • 掌握「1. 历史动机与发展脉络」的核心机制、典型用法与常见陷阱
  • 掌握「2. 形式化定义」的核心机制、典型用法与常见陷阱
  • 掌握「3. 理论推导与原理解析」的核心机制、典型用法与常见陷阱
  • 掌握「4. 代码示例」的核心机制、典型用法与常见陷阱
  • 掌握「5. 对比分析」的核心机制、典型用法与常见陷阱

本文以 Go 1.22 为基准版本，覆盖 Go 1.0 至 Go 1.24 的 net/http 与中间件生态演进，包括 Handler 接口语义、洋葱模型形式化、Chain 组合律、context 传播机制、主流框架（chi、gin、echo）对比与企业级 API 网关案例研究。适用于已掌握 Go 基础语法与 HTTP 服务开发、希望深入理解中间件工程化落地的工程师。

1. 历史动机与发展脉络

1.1 中间件概念的起源（1990s-2000s）

“中间件”（middleware）一词最早出现在分布式系统领域，指位于操作系统与应用之间的软件层（如 CORBA、Message Queue）。在 Web 开发语境下，中间件作为一种代码组织模式，最早由 Python WSGI（PEP 333, 2003）规范化：

# WSGI 中间件（Python）
class LoggingMiddleware:
    def __init__(self, app):
        self.app = app

    def __call__(self, environ, start_response):
        print("请求开始")
        return self.app(environ, start_response)

随后 Ruby Rack（2007）、Node.js Connect（2010）、Express（2010）相继采用类似模式。Go 在 1.0（2012）发布时，net/http 包就提供了 http.Handler 接口，使得中间件模式可以零框架实现。

1.2 Go 1.0（2012-03）：net/http 基础

Go 1.0 的 net/http 包设计了两个核心接口：

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

type HandlerFunc func(ResponseWriter, *Request)

HandlerFunc 是一个 适配器类型（adapter type），使普通函数可以作为 Handler 使用：

// ServeHTTP 实现 Handler 接口
func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) {
    f(w, r)
}

这一设计使得中间件签名 func(http.Handler) http.Handler 成为可能，并形成了 Go 中间件生态的基石。

1.3 Go 1.7（2016-08）：context.Context 引入

Go 1.7 将 context.Context 引入标准库，并修改 http.Request 增加 Context() 方法与 WithContext() 方法。这一变化对中间件生态产生深远影响：

  • 请求作用域数据：通过 context.WithValue 传递 request ID、user ID 等。
  • 超时与取消：通过 context.WithTimeout、context.WithCancel 控制请求生命周期。
  • 链路追踪：通过 context 传播 trace ID、span ID。

1.4 Go 1.22（2024-02）：ServeMux 增强

Go 1.22 大幅增强 http.ServeMux：

  • 方法匹配：mux.HandleFunc("GET /api/users", handler) 支持方法与路径组合。
  • 路径参数：mux.HandleFunc("GET /api/users/{id}", handler) 支持 {id} 参数，通过 r.PathValue("id") 读取。
  • Use 方法（Go 1.22+）：mux.Use(middleware...) 注册全局中间件。
mux := http.NewServeMux()
mux.Use(loggingMiddleware, recoveryMiddleware)
mux.HandleFunc("GET /api/users/{id}", getUser)

此前需要依赖 chi、gorilla/mux 等第三方路由器才能实现的功能，现在标准库原生支持。

1.5 框架生态演进

框架发布年份中间件机制路由器性能（QPS）
net/http（标准库）2012func(http.Handler) http.HandlerServeMux100k
gorilla/mux2012func(http.Handler) http.Handler自研80k
chi2015func(http.Handler) http.Handler自研（radix tree）120k
gin2014gin.HandlerFuncradix tree200k
echo2015echo.MiddlewareFuncradix tree180k
fiber2020fiber.Handler（基于 fasthttp）radix tree350k

net/http 兼容的中间件（func(http.Handler) http.Handler）可在 chi、gorilla/mux 中直接复用；gin、echo、fiber 因自定义接口而需要适配。

1.6 演进时间轴

timeline
    title Go 中间件演进时间线
    2003-01: Python WSGI（PEP 333）规范化中间件概念
    2007-01: Ruby Rack 沿用中间件模式
    2010-05: Node.js Connect/Express 中间件生态兴起
    2012-03: Go 1.0 net/http，Handler/HandlerFunc 接口
    2014-06: gin 框架发布，自定义 HandlerFunc
    2015-07: chi、echo 框架发布，兼容标准库中间件
    2016-08: Go 1.7 context.Context 引入 http.Request
    2024-02: Go 1.22 ServeMux 方法匹配、路径参数、Use 方法
    2025-02: Go 1.24 进一步优化 ServeMux 性能（radix tree）

2. 形式化定义

2.1 Go 标准库定义

Go 标准库对中间件未做显式定义，但社区形成共识的签名：

type Middleware func(http.Handler) http.Handler

形式化文法：

Middleware = Handler → Handler
Handler    = ResponseWriter × Request → Effect

其中 Effect 表示副作用（写入响应、修改状态等）。中间件本质上是一个 高阶函数（higher-order function），接收 Handler 返回 Handler。

2.2 Handler 接口

// net/http/server.go (Go 1.22)
type Handler interface {
    ServeHTTP(w ResponseWriter, r *Request)
}

type HandlerFunc func(ResponseWriter, *Request)

func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) {
    f(w, r)
}

关键性质：

  1. HandlerFunc 与 Handler 等价：任何 func(ResponseWriter, *Request) 都可通过 HandlerFunc(h) 转为 Handler。
  2. ResponseWriter 是接口，可被包装（wrapper）以扩展行为。
  3. Request 包含 Context()，是只读的，但可通过 r.WithContext(ctx) 创建带新 context 的副本。

2.3 ResponseWriter 接口

type ResponseWriter interface {
    Header() http.Header
    Write([]byte) (int, error)
    WriteHeader(statusCode int)
}

// 可选接口（由具体实现支持）
type Flusher interface {
    Flush()
}

type Hijacker interface {
    Hijack() (net.Conn, *bufio.ReadWriter, error)
}

关键约束：

  1. WriteHeader 至多调用一次；后续调用被忽略并记录日志。
  2. Write 在 WriteHeader 之前调用会隐式触发 WriteHeader(200)。
  3. Hijacker 用于 WebSocket 升级，调用后 ResponseWriter 不再可用。

2.4 中间件的类型签名

从类型论视角，中间件是一个 endomorphism（自同态）：

Middleware:Handler→Handler\text{Middleware} : \text{Handler} \to \text{Handler}

中间件的组合（composition）满足结合律：

(A∘B)∘C=A∘(B∘C)(A \circ B) \circ C = A \circ (B \circ C)

但 不满足交换律：

A∘B≠B∘Ain generalA \circ B \neq B \circ A \quad \text{in general}

反例：设 AA = 认证中间件，BB = 日志中间件。

  • A∘BA \circ B（认证外、日志内）：日志只记录已认证请求。
  • B∘AB \circ A（日志外、认证内）：日志记录所有请求，包括未认证的。

两者行为不同，因此中间件顺序至关重要。

2.5 洋葱模型

洋葱模型（onion model）描述请求与响应在中间件链中的流向：

flowchart LR
    Req[请求] --> A[中间件A前置] --> B[中间件B前置] --> C[中间件C前置] --> H[核心 Handler]
    H --> C2[中间件C后置] --> B2[中间件B后置] --> A2[中间件A后置] --> Res[响应]

形式化：

Response=(Mn∘Mn−1∘⋯∘M1)(Handler)(Request)\text{Response} = (M_n \circ M_{n-1} \circ \cdots \circ M_1)(\text{Handler})(\text{Request})

其中 MiM_i 是中间件，Handler\text{Handler} 是核心业务逻辑。

2.6 context 传播

context.Context 在中间件链中的传播遵循两条规则：

  1. cancel 信号从外向内：外层中间件设置的 timeout/deadline 会传播到内层。
  2. value 从内向外读取：内层中间件设置的值可被外层在 next.ServeHTTP 返回后读取（但不推荐，建议通过响应包装器传递）。

形式化：

ctxinner=derive(ctxouter)\text{ctx}_{\text{inner}} = \text{derive}(\text{ctx}_{\text{outer}})

其中 derive 是 context.WithValue、context.WithCancel、context.WithTimeout 等的统称。

3. 理论推导与原理解析

3.1 装饰器模式与中间件

中间件是装饰器模式（Decorator Pattern）在函数式编程中的体现：

维度面向对象装饰器函数式中间件
包装对象类实例函数
扩展方式继承 + 委托高阶函数
Go 实现嵌入接口 + 包装结构func(http.Handler) http.Handler
类型安全编译期编译期

Go 中间件的函数式本质：

func Logging(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 %v", r.Method, r.URL.Path, time.Since(start))
    })
}

等价于：

// 闭包形式
func Logging(next http.Handler) http.Handler {
    wrapped := func(w http.ResponseWriter, r *http.Request) {
        start := time.Now()
        next.ServeHTTP(w, r)
        log.Printf("%s %s %v", r.Method, r.URL.Path, time.Since(start))
    }
    return http.HandlerFunc(wrapped)
}

3.2 Chain 组合律证明

定理 4.1（Chain 结合律）：对任意中间件 A,B,CA, B, C 与 Handler HH：

(A∘B∘C)(H)=A(B(C(H)))(A \circ B \circ C)(H) = A(B(C(H)))

证明：由中间件签名 Middleware:Handler→Handler\text{Middleware} : \text{Handler} \to \text{Handler}，组合定义为：

(A∘B)(H)=A(B(H))(A \circ B)(H) = A(B(H))

递归展开：

(A∘B∘C)(H)=((A∘B)∘C)(H)=(A∘B)(C(H))=A(B(C(H)))□(A \circ B \circ C)(H) = ((A \circ B) \circ C)(H) = (A \circ B)(C(H)) = A(B(C(H))) \quad \square

推论：中间件 Chain 的实现可从左或从右归约，结果相同。Go 标准实现通常从右向左归约：

func Chain(h http.Handler, middlewares ...Middleware) http.Handler {
    for i := len(middlewares) - 1; i >= 0; i-- {
        h = middlewares[i](h)
    }
    return h
}

3.3 顺序敏感性

定理 4.2（顺序不交换）：存在中间件 A,BA, B 使得 A∘B≠B∘AA \circ B \neq B \circ A。

证明（反例）：设

  • AA = 认证中间件：未认证返回 401，不调用 next。
  • BB = 日志中间件：记录请求方法、路径、状态码。

考察未认证请求：

  • A∘BA \circ B（A 外、B 内）：A 拦截，B 不执行，无日志。
  • B∘AB \circ A（B 外、A 内）：B 先记录请求开始，A 拦截返回 401，B 记录状态码 401。

两者日志输出不同，故 A∘B≠B∘AA \circ B \neq B \circ A。□\square

实践建议：常用顺序为 recover → logging → requestID → cors → rateLimit → auth → handler。

3.4 ResponseWriter 包装器的必要性

http.ResponseWriter 接口不暴露状态码、响应体大小。中间件若需要这些信息（如日志记录状态码），必须包装：

type statusRecorder struct {
    http.ResponseWriter
    status int
    size   int
}

func (r *statusRecorder) WriteHeader(code int) {
    r.status = code
    r.ResponseWriter.WriteHeader(code)
}

func (r *statusRecorder) Write(b []byte) (int, error) {
    n, err := r.ResponseWriter.Write(b)
    r.size += n
    return n, err
}

接口保留问题：原始 ResponseWriter 可能实现 http.Flusher、http.Hijacker 等可选接口。包装后若不实现这些接口，下游 Handler 调用 w.(http.Flusher) 会失败。

解决方案：通过类型断言转发可选接口：

type statusRecorder struct {
    http.ResponseWriter
    status int
    size   int
}

// 实现 Flusher
func (r *statusRecorder) Flush() {
    if f, ok := r.ResponseWriter.(http.Flusher); ok {
        f.Flush()
    }
}

// 实现 Hijacker
func (r *statusRecorder) Hijack() (net.Conn, *bufio.ReadWriter, error) {
    if h, ok := r.ResponseWriter.(http.Hijacker); ok {
        return h.Hijack()
    }
    return nil, nil, errors.New("not a Hijacker")
}

3.5 context.WithValue 的类型安全

context.WithValue 接受 any 类型的 key 与 value：

func WithValue(parent Context, key, val any) Context

若使用字符串作为 key，可能与其他中间件冲突：

// BAD: 字符串 key 容易冲突
ctx := context.WithValue(r.Context(), "userID", 123)
// 另一个中间件可能也用 "userID"，覆盖或被覆盖

类型安全方案：使用自定义类型作为 key：

type ctxKey string

const (
    userIDKey    ctxKey = "userID"
    requestIDKey ctxKey = "requestID"
)

ctx := context.WithValue(r.Context(), userIDKey, 123)

由于 ctxKey 是未导出类型（若包外不可见）或具名类型，冲突概率为零。

3.6 中间件的代数性质

中间件可视为 幺半群（monoid）：

  • 二元运算：组合 ∘\circ
  • 单位元：恒等中间件 id(h)=h\text{id}(h) = h

满足：

  1. 结合律：(A∘B)∘C=A∘(B∘C)(A \circ B) \circ C = A \circ (B \circ C)（定理 4.1）。
  2. 单位元：id∘A=A∘id=A\text{id} \circ A = A \circ \text{id} = A。

但 不满足交换律（定理 4.2），因此中间件代数是 非交换幺半群（non-commutative monoid）。

4. 代码示例

4.1 基础：日志中间件

package main

import (
    "log"
    "net/http"
    "time"
)

// Logging 是日志中间件，记录请求方法、路径、耗时
func Logging(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        start := time.Now()
        log.Printf("请求开始: %s %s", r.Method, r.URL.Path)

        next.ServeHTTP(w, r)

        log.Printf("请求完成: %s %s 耗时: %v", r.Method, r.URL.Path, time.Since(start))
    })
}

func helloHandler(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("你好！"))
}

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/", helloHandler)

    // 用中间件包装路由
    handler := Logging(mux)
    log.Println("服务启动: http://localhost:8080")
    http.ListenAndServe(":8080", handler)
}

运行：

go run main.go
# 另一终端
curl http://localhost:8080/
# 服务端日志：
# 请求开始: GET /
# 请求完成: GET / 耗时: 12.345µs

4.2 状态码捕获：响应包装器

package main

import (
    "log"
    "net/http"
    "time"
)

// statusRecorder 包装 ResponseWriter，捕获状态码与响应体大小
type statusRecorder struct {
    http.ResponseWriter
    status int
    size   int
}

func (r *statusRecorder) WriteHeader(code int) {
    r.status = code
    r.ResponseWriter.WriteHeader(code)
}

func (r *statusRecorder) Write(b []byte) (int, error) {
    n, err := r.ResponseWriter.Write(b)
    r.size += n
    return n, err
}

// LoggingWithStatus 记录状态码与响应大小
func LoggingWithStatus(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        start := time.Now()
        recorder := &statusRecorder{ResponseWriter: w, status: 200}

        next.ServeHTTP(recorder, r)

        log.Printf("%s %s %d %d字节 %v",
            r.Method, r.URL.Path, recorder.status, recorder.size, time.Since(start))
    })
}

func apiHandler(w http.ResponseWriter, r *http.Request) {
    if r.URL.Query().Get("error") == "1" {
        http.Error(w, "出错了", http.StatusInternalServerError)
        return
    }
    w.Write([]byte(`{"status":"ok"}`))
}

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/api", apiHandler)
    http.ListenAndServe(":8080", LoggingWithStatus(mux))
}

4.3 认证中间件

package main

import (
    "context"
    "net/http"
    "strings"
)

type ctxKey string

const userKey ctxKey = "user"

// Auth 验证 Bearer Token，将用户信息存入 context
func Auth(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        authHeader := r.Header.Get("Authorization")
        if authHeader == "" {
            http.Error(w, `{"error":"missing_token"}`, http.StatusUnauthorized)
            return
        }

        parts := strings.SplitN(authHeader, " ", 2)
        if len(parts) != 2 || parts[0] != "Bearer" {
            http.Error(w, `{"error":"invalid_token"}`, http.StatusUnauthorized)
            return
        }

        token := parts[1]
        user, err := validateToken(token)
        if err != nil {
            http.Error(w, `{"error":"invalid_token"}`, http.StatusUnauthorized)
            return
        }

        // 将用户信息存入 context
        ctx := context.WithValue(r.Context(), userKey, user)
        next.ServeHTTP(w, r.WithContext(ctx))
    })
}

type User struct {
    ID   int
    Name string
}

func validateToken(token string) (*User, error) {
    // 简化：实际应查询数据库或 JWT 验证
    if token == "secret123" {
        return &User{ID: 1, Name: "Alice"}, nil
    }
    return nil, fmt.Errorf("invalid token")
}

// FromContext 从 context 提取用户信息
func UserFromContext(ctx context.Context) (*User, bool) {
    user, ok := ctx.Value(userKey).(*User)
    return user, ok
}

func profileHandler(w http.ResponseWriter, r *http.Request) {
    user, ok := UserFromContext(r.Context())
    if !ok {
        http.Error(w, "未认证", http.StatusUnauthorized)
        return
    }
    fmt.Fprintf(w, `{"id":%d,"name":"%s"}`, user.ID, user.Name)
}

func main() {
    mux := http.NewServeMux()
    mux.Handle("/api/profile", Auth(http.HandlerFunc(profileHandler)))
    http.ListenAndServe(":8080", mux)
}

4.4 CORS 中间件

package main

import (
    "net/http"
    "strconv"
    "strings"
)

// CORSConfig 是 CORS 中间件配置
type CORSConfig struct {
    AllowOrigins     []string
    AllowMethods     []string
    AllowHeaders     []string
    AllowCredentials bool
    MaxAge           int // 秒
}

// CORS 处理跨域请求
func CORS(cfg CORSConfig) func(http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            origin := r.Header.Get("Origin")
            if origin == "" {
                next.ServeHTTP(w, r)
                return
            }

            // 检查 Origin 是否允许
            allowed := false
            for _, o := range cfg.AllowOrigins {
                if o == "*" || o == origin {
                    allowed = true
                    break
                }
            }
            if !allowed {
                next.ServeHTTP(w, r)
                return
            }

            // 设置 CORS 头
            w.Header().Set("Access-Control-Allow-Origin", origin)
            w.Header().Set("Access-Control-Allow-Methods", strings.Join(cfg.AllowMethods, ", "))
            w.Header().Set("Access-Control-Allow-Headers", strings.Join(cfg.AllowHeaders, ", "))
            if cfg.AllowCredentials {
                w.Header().Set("Access-Control-Allow-Credentials", "true")
            }
            if cfg.MaxAge > 0 {
                w.Header().Set("Access-Control-Max-Age", strconv.Itoa(cfg.MaxAge))
            }

            // 预检请求直接返回 204
            if r.Method == http.MethodOptions {
                w.WriteHeader(http.StatusNoContent)
                return
            }

            next.ServeHTTP(w, r)
        })
    }
}

func main() {
    cfg := CORSConfig{
        AllowOrigins:     []string{"https://example.com", "https://app.example.com"},
        AllowMethods:     []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
        AllowHeaders:     []string{"Content-Type", "Authorization"},
        AllowCredentials: true,
        MaxAge:           3600,
    }

    mux := http.NewServeMux()
    mux.HandleFunc("/api/data", func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte(`{"data":"hello"}`))
    })

    handler := CORS(cfg)(mux)
    http.ListenAndServe(":8080", handler)
}

4.5 恢复中间件（panic 捕获）

package main

import (
    "log"
    "net/http"
    "runtime/debug"
)

// Recovery 捕获 panic，返回 500 错误
func Recovery(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        defer func() {
            if err := recover(); err != nil {
                log.Printf("panic 已恢复: %v\n%s", err, debug.Stack())
                http.Error(w, `{"error":"internal_server_error"}`, http.StatusInternalServerError)
            }
        }()
        next.ServeHTTP(w, r)
    })
}

func riskyHandler(w http.ResponseWriter, r *http.Request) {
    // 模拟 panic
    if r.URL.Query().Get("panic") == "1" {
        panic("something went wrong")
    }
    w.Write([]byte("ok"))
}

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/api", riskyHandler)
    http.ListenAndServe(":8080", Recovery(mux))
}

4.6 请求 ID 与链路追踪

package main

import (
    "context"
    "crypto/rand"
    "encoding/hex"
    "net/http"
)

type traceCtxKey string

const requestIDKey traceCtxKey = "requestID"

// RequestID 生成唯一请求 ID，注入 context 与响应头
func RequestID(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 优先复用上游传入的 ID
        requestID := r.Header.Get("X-Request-ID")
        if requestID == "" {
            requestID = generateID()
        }

        // 设置响应头
        w.Header().Set("X-Request-ID", requestID)

        // 注入 context
        ctx := context.WithValue(r.Context(), requestIDKey, requestID)
        next.ServeHTTP(w, r.WithContext(ctx))
    })
}

// generateID 生成 16 字节随机 ID
func generateID() string {
    b := make([]byte, 16)
    rand.Read(b)
    return hex.EncodeToString(b)
}

// RequestIDFromContext 从 context 提取请求 ID
func RequestIDFromContext(ctx context.Context) string {
    if id, ok := ctx.Value(requestIDKey).(string); ok {
        return id
    }
    return ""
}

func handler(w http.ResponseWriter, r *http.Request) {
    requestID := RequestIDFromContext(r.Context())
    // 将 requestID 记录到日志、传递给下游服务
    w.Write([]byte(`{"requestID":"` + requestID + `"}`))
}

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/api", handler)
    http.ListenAndServe(":8080", RequestID(mux))
}

4.7 超时中间件

package main

import (
    "context"
    "net/http"
    "time"
)

// Timeout 为每个请求设置超时
func Timeout(duration time.Duration) func(http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            ctx, cancel := context.WithTimeout(r.Context(), duration)
            defer cancel()

            // 使用 channel 避免超时后 goroutine 泄漏
            done := make(chan struct{})
            go func() {
                defer close(done)
                next.ServeHTTP(w, r.WithContext(ctx))
            }()

            select {
            case <-done:
                // 正常完成
            case <-ctx.Done():
                if ctx.Err() == context.DeadlineExceeded {
                    http.Error(w, `{"error":"request_timeout"}`, http.StatusGatewayTimeout)
                }
            }
        })
    }
}

func slowHandler(w http.ResponseWriter, r *http.Request) {
    time.Sleep(3 * time.Second)
    w.Write([]byte("done"))
}

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/slow", slowHandler)

    // 设置 2 秒超时
    handler := Timeout(2 * time.Second)(mux)
    http.ListenAndServe(":8080", handler)
}

4.8 限流中间件（令牌桶）

package main

import (
    "net/http"
    "sync"
    "time"
)

// TokenBucket 令牌桶限流器
type TokenBucket struct {
    rate       float64       // 每秒令牌数
    burst      float64       // 桶容量
    tokens     float64       // 当前令牌数
    lastUpdate time.Time     // 上次更新时间
    mu         sync.Mutex
}

func NewTokenBucket(rate, burst float64) *TokenBucket {
    return &TokenBucket{
        rate:       rate,
        burst:      burst,
        tokens:     burst,
        lastUpdate: time.Now(),
    }
}

func (tb *TokenBucket) Allow() bool {
    tb.mu.Lock()
    defer tb.mu.Unlock()

    now := time.Now()
    elapsed := now.Sub(tb.lastUpdate).Seconds()
    tb.tokens += elapsed * tb.rate
    if tb.tokens > tb.burst {
        tb.tokens = tb.burst
    }
    tb.lastUpdate = now

    if tb.tokens >= 1 {
        tb.tokens--
        return true
    }
    return false
}

// RateLimit 限流中间件
func RateLimit(tb *TokenBucket) func(http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            if !tb.Allow() {
                w.Header().Set("Retry-After", "1")
                http.Error(w, `{"error":"rate_limited"}`, http.StatusTooManyRequests)
                return
            }
            next.ServeHTTP(w, r)
        })
    }
}

func main() {
    // 每秒 10 个请求，突发 20 个
    tb := NewTokenBucket(10, 20)

    mux := http.NewServeMux()
    mux.HandleFunc("/api", func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("ok"))
    })

    handler := RateLimit(tb)(mux)
    http.ListenAndServe(":8080", handler)
}

4.9 中间件链（Chain）

package main

import (
    "log"
    "net/http"
    "runtime/debug"
    "time"
)

type Middleware func(http.Handler) http.Handler

// Chain 将多个中间件组合为一个
// 顺序：第一个参数是最外层，最后一个参数是最内层（紧邻 handler）
func Chain(handler http.Handler, middlewares ...Middleware) http.Handler {
    for i := len(middlewares) - 1; i >= 0; i-- {
        handler = middlewares[i](handler)
    }
    return handler
}

func Recovery(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        defer func() {
            if err := recover(); err != nil {
                log.Printf("panic: %v\n%s", err, debug.Stack())
                http.Error(w, "内部错误", http.StatusInternalServerError)
            }
        }()
        next.ServeHTTP(w, r)
    })
}

func Logging(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 %v", r.Method, r.URL.Path, time.Since(start))
    })
}

func Auth(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if r.Header.Get("Authorization") == "" {
            http.Error(w, "未认证", http.StatusUnauthorized)
            return
        }
        next.ServeHTTP(w, r)
    })
}

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/api", func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("ok"))
    })

    // 中间件顺序：recovery（最外）→ logging → auth（最内）
    handler := Chain(mux, Recovery, Logging, Auth)
    http.ListenAndServe(":8080", handler)
}

5. 对比分析

5.1 Go 标准库 vs chi vs gin vs echo

维度net/httpchiginecho
中间件签名func(http.Handler) http.Handlerfunc(http.Handler) http.Handlergin.HandlerFuncecho.MiddlewareFunc
路由器ServeMux（Go 1.22+ 增强）radix treeradix treeradix tree
路由参数r.PathValue("id")（Go 1.22+）chi.URLParam(r, "id")c.Param("id")c.Param("id")
中间件复用标准兼容标准需适配需适配
性能（QPS）100k120k200k180k
生态成熟度高（标准库）高极高高
学习成本低低中中
适用场景简单服务、原生标准库增强高性能 API高性能 API

5.2 Go 中间件 vs Node.js Express

维度Go 中间件Express 中间件
签名func(http.Handler) http.Handler(req, res, next) => void
调用模型装饰器（高阶函数）回调链（callback chain）
异步处理原生支持（goroutine）Promise/async-await
错误处理panic recovernext(err) 错误中间件
类型安全编译期运行时（无类型）
性能高（编译型）中（V8 JIT）

5.3 Go 中间件 vs Python Django

维度Go 中间件Django 中间件
实现方式函数装饰器类（__call__ / process_request / process_response）
配置方式代码注册MIDDLEWARE 列表
顺序控制Chain 参数顺序MIDDLEWARE 顺序
异步支持goroutineasync views（Django 4.0+）
类型安全编译期运行时（鸭子类型）

5.4 Go 中间件 vs Java Spring Filter

维度Go 中间件Spring Filter
实现函数接口 Filter
注册Chain(handler, mw...)@Component + FilterRegistrationBean
顺序代码显式@Order 注解
依赖注入手动IoC 容器
类型安全编译期编译期

6. 常见陷阱与反模式

6.1 反模式：忘记调用 next

// BAD: 中间件未调用 next，请求链中断
func BadMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 完全忽略 next，所有请求到此为止
        w.Write([]byte("intercepted"))
    })
}
// GOOD: 显式调用 next（除非有意拦截）
func GoodMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 前置逻辑
        if r.Header.Get("X-Block") == "1" {
            http.Error(w, "blocked", http.StatusForbidden)
            return // 有意拦截
        }
        next.ServeHTTP(w, r)
        // 后置逻辑
    })
}

6.2 反模式：ResponseWriter 重复写入

// BAD: 中间件写入响应后，next 也写入，导致响应体混乱
func BadMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("prefix"))  // 已经写入响应
        next.ServeHTTP(w, r)       // next 也写入，导致混合内容
    })
}
// GOOD: 中间件若需修改响应，应使用包装器捕获 next 的输出
func GoodMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        recorder := &bodyRecorder{ResponseWriter: w}
        next.ServeHTTP(recorder, r)
        // 修改 recorder.body 后写入 w
        w.Write(recorder.body)
    })
}

6.3 反模式：在中间件中启动未绑定 context 的 goroutine

// BAD: goroutine 在请求结束后仍存活，导致泄漏
func BadMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        go func() {
            // 这个 goroutine 永远不会取消
            for {
                time.Sleep(time.Second)
                log.Println("后台任务")
            }
        }()
        next.ServeHTTP(w, r)
    })
}
// GOOD: goroutine 绑定 context，请求结束后自动退出
func GoodMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        ctx, cancel := context.WithCancel(r.Context())
        go func() {
            defer cancel()
            for {
                select {
                case <-ctx.Done():
                    return
                case <-time.After(time.Second):
                    log.Println("后台任务")
                }
            }
        }()
        next.ServeHTTP(w, r.WithContext(ctx))
    })
}

6.4 反模式：使用字符串作为 context key

// BAD: 字符串 key 可能与其他中间件冲突
ctx := context.WithValue(r.Context(), "userID", 123)
// GOOD: 使用自定义类型作为 key
type ctxKey string
const userIDKey ctxKey = "userID"
ctx := context.WithValue(r.Context(), userIDKey, 123)

6.5 反模式：中间件顺序错误

// BAD: recovery 在内层，外层中间件 panic 无法被捕获
handler := Chain(mux, Logging, Auth, Recovery)
// 如果 Logging 或 Auth panic，Recovery 无法捕获
// GOOD: recovery 在最外层
handler := Chain(mux, Recovery, Logging, Auth)

6.6 反模式：包装 ResponseWriter 后丢失 Flusher/Hijacker

// BAD: 包装后未实现 Flusher，下游 SSE/WebSocket 失败
type recorder struct {
    http.ResponseWriter
    status int
}
// recorder 未实现 http.Flusher，下游调用 w.(http.Flusher) 失败
// GOOD: 转发可选接口
type recorder struct {
    http.ResponseWriter
    status int
}

func (r *recorder) Flush() {
    if f, ok := r.ResponseWriter.(http.Flusher); ok {
        f.Flush()
    }
}

func (r *recorder) Hijack() (net.Conn, *bufio.ReadWriter, error) {
    if h, ok := r.ResponseWriter.(http.Hijacker); ok {
        return h.Hijack()
    }
    return nil, nil, errors.New("not a Hijacker")
}

6.7 反模式：在中间件中读取请求体后未恢复

// BAD: 读取请求体后未重置，下游 Handler 无法读取
func BadMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        body, _ := io.ReadAll(r.Body)
        log.Println(string(body))
        // r.Body 已耗尽，next 读取时为空
        next.ServeHTTP(w, r)
    })
}
// GOOD: 读取后重新封装 r.Body
func GoodMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        body, _ := io.ReadAll(r.Body)
        log.Println(string(body))
        // 恢复 Body
        r.Body = io.NopCloser(bytes.NewReader(body))
        next.ServeHTTP(w, r)
    })
}

6.8 反模式：超时中间件 goroutine 泄漏

// BAD: 超时后 goroutine 仍运行，且可能写入已关闭的 ResponseWriter
func BadTimeout(duration time.Duration) func(http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            ctx, cancel := context.WithTimeout(r.Context(), duration)
            defer cancel()
            next.ServeHTTP(w, r.WithContext(ctx))
            // 即使超时，next 仍在运行，可能继续写入 w
        })
    }
}
// GOOD: 使用 channel 等待，超时后立即返回（但 next goroutine 可能仍运行）
// 更好的方案是让 next 通过 context 感知超时并主动退出
func GoodTimeout(duration time.Duration) func(http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            ctx, cancel := context.WithTimeout(r.Context(), duration)
            defer cancel()

            done := make(chan struct{})
            go func() {
                defer close(done)
                next.ServeHTTP(w, r.WithContext(ctx))
            }()

            select {
            case <-done:
            case <-ctx.Done():
                http.Error(w, "timeout", http.StatusGatewayTimeout)
            }
        })
    }
}

7. 工程实践与最佳实践

7.1 中间件目录组织

flowchart TD
    T0["project/"]
    T1["internal/"]
    T2["middleware/"]
    T3["recovery.go"]
    T4["logging.go"]
    T5["auth.go"]
    T6["cors.go"]
    T7["ratelimit.go"]
    T8["requestid.go"]
    T9["chain.go       # Chain 辅助函数"]
    T10["handler/"]
    T11["user.go"]
    T12["order.go"]
    T13["server/"]
    T14["router.go      # 路由与中间件组装"]
    T15["cmd/"]
    T16["server/"]
    T17["main.go"]
    T0 --> T1
    T14 --> T15
    T15 --> T16
    T16 --> T17

7.2 中间件单元测试

package middleware

import (
    "net/http"
    "net/http/httptest"
    "testing"
)

// TestLogging 测试日志中间件
func TestLogging(t *testing.T) {
    handler := Logging(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.WriteHeader(http.StatusOK)
        w.Write([]byte("ok"))
    }))

    req := httptest.NewRequest("GET", "/test", nil)
    rec := httptest.NewRecorder()
    handler.ServeHTTP(rec, req)

    if rec.Code != http.StatusOK {
        t.Errorf("期望 200，实际 %d", rec.Code)
    }
    if rec.Body.String() != "ok" {
        t.Errorf("期望 'ok'，实际 '%s'", rec.Body.String())
    }
}

// TestRecovery 测试 panic 恢复
func TestRecovery(t *testing.T) {
    handler := Recovery(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        panic("test panic")
    }))

    req := httptest.NewRequest("GET", "/test", nil)
    rec := httptest.NewRecorder()
    handler.ServeHTTP(rec, req)

    if rec.Code != http.StatusInternalServerError {
        t.Errorf("期望 500，实际 %d", rec.Code)
    }
}

// TestAuth 测试认证中间件
func TestAuth(t *testing.T) {
    handler := Auth(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("authenticated"))
    }))

    // 测试无 token
    req := httptest.NewRequest("GET", "/test", nil)
    rec := httptest.NewRecorder()
    handler.ServeHTTP(rec, req)
    if rec.Code != http.StatusUnauthorized {
        t.Errorf("无 token 期望 401，实际 %d", rec.Code)
    }

    // 测试有 token
    req = httptest.NewRequest("GET", "/test", nil)
    req.Header.Set("Authorization", "Bearer secret123")
    rec = httptest.NewRecorder()
    handler.ServeHTTP(rec, req)
    if rec.Code != http.StatusOK {
        t.Errorf("有 token 期望 200，实际 %d", rec.Code)
    }
}

7.3 可配置中间件

// 可配置的日志中间件
type LoggingConfig struct {
    Logger      *log.Logger
    LogBody     bool
    LogHeaders  bool
    ExcludePath map[string]bool
}

func LoggingWithConfig(cfg LoggingConfig) func(http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            if cfg.ExcludePath[r.URL.Path] {
                next.ServeHTTP(w, r)
                return
            }

            start := time.Now()
            recorder := &statusRecorder{ResponseWriter: w, status: 200}
            next.ServeHTTP(recorder, r)

            cfg.Logger.Printf("%s %s %d %v",
                r.Method, r.URL.Path, recorder.status, time.Since(start))

            if cfg.LogHeaders {
                for k, v := range r.Header {
                    cfg.Logger.Printf("  Header: %s=%v", k, v)
                }
            }
        })
    }
}

7.4 健康检查与就绪检查

// HealthCheck 健康检查中间件
func HealthCheck(checks map[string]func() error) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        results := make(map[string]string)
        allOK := true
        for name, check := range checks {
            if err := check(); err != nil {
                results[name] = err.Error()
                allOK = false
            } else {
                results[name] = "ok"
            }
        }

        status := http.StatusOK
        if !allOK {
            status = http.StatusServiceUnavailable
        }

        w.Header().Set("Content-Type", "application/json")
        w.WriteHeader(status)
        json.NewEncoder(w).Encode(results)
    }
}

// 使用
checks := map[string]func() error{
    "database": checkDB,
    "redis":    checkRedis,
    "kafka":    checkKafka,
}
mux.HandleFunc("/health", HealthCheck(checks))

7.5 Prometheus 指标中间件

package middleware

import (
    "net/http"
    "time"
    "github.com/prometheus/client_golang/prometheus"
    "github.com/prometheus/client_golang/prometheus/promauto"
)

var (
    httpRequestsTotal = promauto.NewCounterVec(
        prometheus.CounterOpts{
            Name: "http_requests_total",
            Help: "HTTP 请求总数",
        },
        []string{"method", "path", "status"},
    )
    httpRequestDuration = promauto.NewHistogramVec(
        prometheus.HistogramOpts{
            Name:    "http_request_duration_seconds",
            Help:    "HTTP 请求耗时（秒）",
            Buckets: prometheus.DefBuckets,
        },
        []string{"method", "path"},
    )
)

// Metrics Prometheus 指标中间件
func Metrics(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        start := time.Now()
        recorder := &statusRecorder{ResponseWriter: w, status: 200}

        next.ServeHTTP(recorder, r)

        duration := time.Since(start).Seconds()
        httpRequestsTotal.WithLabelValues(r.Method, r.URL.Path,
            strconv.Itoa(recorder.status)).Inc()
        httpRequestDuration.WithLabelValues(r.Method, r.URL.Path).Observe(duration)
    })
}

7.6 OpenTelemetry 链路追踪中间件

package middleware

import (
    "net/http"
    "go.opentelemetry.io/otel"
    "go.opentelemetry.io/otel/attribute"
    "go.opentelemetry.io/otel/trace"
)

// Tracing OpenTelemetry 链路追踪中间件
func Tracing(serviceName string) func(http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            tracer := otel.Tracer(serviceName)
            ctx, span := tracer.Start(r.Context(), r.Method+" "+r.URL.Path,
                trace.WithAttributes(
                    attribute.String("http.method", r.Method),
                    attribute.String("http.url", r.URL.String()),
                    attribute.String("http.host", r.Host),
                ),
            )
            defer span.End()

            recorder := &statusRecorder{ResponseWriter: w, status: 200}
            next.ServeHTTP(recorder, r.WithContext(ctx))

            span.SetAttributes(attribute.Int("http.status_code", recorder.status))
        })
    }
}

7.7 中间件配置管理

// 中间件配置（YAML）
type MiddlewareConfig struct {
    Recovery struct {
        Enabled bool `yaml:"enabled"`
    } `yaml:"recovery"`
    Logging struct {
        Enabled  bool   `yaml:"enabled"`
        LogBody  bool   `yaml:"log_body"`
        LogLevel string `yaml:"log_level"`
    } `yaml:"logging"`
    CORS struct {
        Enabled        bool     `yaml:"enabled"`
        AllowOrigins   []string `yaml:"allow_origins"`
        AllowMethods   []string `yaml:"allow_methods"`
        AllowHeaders   []string `yaml:"allow_headers"`
        AllowCredentials bool   `yaml:"allow_credentials"`
        MaxAge         int      `yaml:"max_age"`
    } `yaml:"cors"`
    RateLimit struct {
        Enabled bool    `yaml:"enabled"`
        Rate    float64 `yaml:"rate"`
        Burst   float64 `yaml:"burst"`
    } `yaml:"rate_limit"`
    Auth struct {
        Enabled bool `yaml:"enabled"`
        Secret  string `yaml:"secret"`
    } `yaml:"auth"`
}

// BuildMiddleware 根据配置构建中间件链
func BuildMiddleware(cfg MiddlewareConfig) []Middleware {
    var mws []Middleware
    if cfg.Recovery.Enabled {
        mws = append(mws, Recovery)
    }
    if cfg.Logging.Enabled {
        mws = append(mws, LoggingWithConfig(LoggingConfig{
            LogBody: cfg.Logging.LogBody,
        }))
    }
    if cfg.CORS.Enabled {
        mws = append(mws, CORS(CORSConfig{
            AllowOrigins:     cfg.CORS.AllowOrigins,
            AllowMethods:     cfg.CORS.AllowMethods,
            AllowHeaders:     cfg.CORS.AllowHeaders,
            AllowCredentials: cfg.CORS.AllowCredentials,
            MaxAge:           cfg.CORS.MaxAge,
        }))
    }
    if cfg.RateLimit.Enabled {
        mws = append(mws, RateLimit(NewTokenBucket(cfg.RateLimit.Rate, cfg.RateLimit.Burst)))
    }
    if cfg.Auth.Enabled {
        mws = append(mws, Auth)
    }
    return mws
}

8. 案例研究

8.1 案例一：Caddy 服务器——中间件架构

Caddy 是 Go 编写的现代 Web 服务器，其核心架构基于中间件模式：

// Caddy 中间件（简化）
type Middleware func(next http.HandlerFunc) http.HandlerFunc

// Caddy 的中间件链
type Server struct {
    handlers []Middleware
}

func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    var h http.HandlerFunc = s.finalHandler
    for i := len(s.handlers) - 1; i >= 0; i-- {
        h = s.handlers[i](h)
    }
    h(w, r)
}

特点：

  • 中间件配置通过 Caddyfile 声明式配置。
  • 支持 hot reload，中间件可运行时增减。
  • 内置丰富中间件：TLS 自动证书、HTTP/3、压缩、缓存、反向代理。

8.2 案例二：gin 框架——Engine 与 RouterGroup

gin 框架的中间件实现：

// gin 中间件签名
type HandlerFunc func(*Context)

// 全局中间件
r := gin.New()
r.Use(gin.Logger(), gin.Recovery())

// 路由组中间件
api := r.Group("/api")
api.Use(AuthMiddleware())
{
    api.GET("/users", getUsers)
    api.POST("/users", createUser)
}

特点：

  • 使用 gin.Context 替代标准库的 http.ResponseWriter 与 *http.Request。
  • 中间件通过 c.Next() 与 c.Abort() 控制。
  • 性能高（基于 radix tree 路由 + 无反射）。

8.3 案例三：Kubernetes API Server——Filter Chain

Kubernetes API Server 的请求处理基于 Filter Chain：

// 简化的 filter chain
filters := []filter{
    WithPanicRecovery,
    WithRequestInfo,
    WithImpersonation,
    WithAuthentication,
    WithAuthorization,
    WithAudit,
    WithLimitation,
}
handler := chain(filters, resourceHandler)

特点：

  • 每个 filter 是独立的认证、授权、审计逻辑。
  • 通过 apirequest.RequestInfo 在 filter 间传递请求元数据。
  • 支持动态配置：通过 --authorization-mode 等启动参数控制。

8.4 案例四：Prometheus——指标采集中间件

Prometheus 的指标采集通过中间件模式集成：

// 通过 promhttp.InstrumentHandlerDuration 包装 Handler
handler := promhttp.InstrumentHandlerDuration(
    prometheus.NewHistogram(prometheus.HistogramOpts{
        Name: "http_request_duration_seconds",
    }),
    mux,
)

特点：

  • 装饰器风格，不侵入业务代码。
  • 支持延迟、请求计数、响应大小等多种指标。
  • 与 Grafana 集成，实现可视化监控。

8.5 案例五：Istio Envoy——Sidecar 中间件

Istio 的 Envoy sidecar 本质是一个网络层中间件：

  • 所有进出 Pod 的流量经过 Envoy sidecar。
  • Sidecar 实现认证、限流、熔断、链路追踪、灰度发布等。
  • 配置通过 xDS 协议动态下发。

特点：

  • 中间件下沉到网络层，业务代码零侵入。
  • 支持多语言（Java/Go/Python/Node.js 等）。
  • 与 Kubernetes 深度集成。

9.1 基础题

题 1：实现一个中间件 RequestSizeLimit(max int64)，限制请求体大小，超过则返回 413。

func RequestSizeLimit(max int64) func(http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            if r.ContentLength > max {
                http.Error(w, `{"error":"request_too_large"}`, http.StatusRequestEntityTooLarge)
                return
            }
            // 限制实际读取的字节数（防止 ContentLength 伪造）
            r.Body = http.MaxBytesReader(w, r.Body, max)
            next.ServeHTTP(w, r)
        })
    }
}

题 2：解释为何 Recovery 中间件应该放在 Chain 的最外层。

Recovery 中间件通过 defer recover() 捕获 panic。如果 Recovery 不在最外层，外层中间件（如 Logging）若发生 panic，将无法被捕获，导致进程崩溃。

正确顺序示例：

handler := Chain(mux, Recovery, Logging, Auth)

这样 Recovery 的 defer 在最外层，能捕获任何内层中间件或 Handler 的 panic。

9.2 进阶题

题 3：实现一个支持路由级中间件的 Group 函数，语法如下：

mux := http.NewServeMux()
api := Group(mux, "/api", AuthMiddleware, RateLimitMiddleware)
api.HandleFunc("/users", getUsers) // 自动注册为 /api/users，并应用组中间件
type Group struct {
    prefix      string
    middlewares []Middleware
    mux         *http.ServeMux
}

func NewGroup(mux *http.ServeMux, prefix string, mws ...Middleware) *Group {
    return &Group{
        prefix:      prefix,
        middlewares: mws,
        mux:         mux,
    }
}

func (g *Group) HandleFunc(pattern string, handler http.HandlerFunc) {
    fullPattern := g.prefix + pattern
    h := http.Handler(handler)
    // 应用组中间件（从右向左）
    for i := len(g.middlewares) - 1; i >= 0; i-- {
        h = g.middlewares[i](h)
    }
    g.mux.Handle(fullPattern, h)
}

// 使用
mux := http.NewServeMux()
api := NewGroup(mux, "/api", Auth, RateLimit(NewTokenBucket(10, 20)))
api.HandleFunc("/users", func(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("users list"))
})

题 4：分析以下中间件代码的并发安全问题并修复：

type counter struct {
    count int
}

func (c *counter) Middleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        c.count++
        next.ServeHTTP(w, r)
    })
}

问题：c.count++ 不是原子操作，在并发请求下会发生数据竞争。

修复：使用 sync/atomic 或 sync.Mutex：

type counter struct {
    count int64
}

func (c *counter) Middleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        atomic.AddInt64(&c.count, 1)
        next.ServeHTTP(w, r)
    })
}

// 读取计数
func (c *counter) Get() int64 {
    return atomic.LoadInt64(&c.count)
}

11. 扩展阅读

11.2 经典论文

  • On the Design of the Middleware for Web Services (Sadjadi et al., 2007)：中间件设计原则。
  • A Survey of API Gateway Patterns (Velez et al., 2020)：API 网关模式综述。
  • Service Mesh: A Systematic Mapping Study (Soldani et al., 2022)：服务网格与中间件关系。

11.3 开源项目

11.4 书籍推荐

  • The Go Programming Language (Alan A. A. Donovan & Brian W. Kernighan, Addison-Wesley, 2015)
  • Go in Action (William Kennedy, Brian Ketelsen, Erik St. Martin, Manning, 2016)
  • Web Development with Go (Shiju Varghese, Packt, 2018)
  • Cloud Native Go (Matthew A. Titmus, O’Reilly, 2021)
  • Building Microservices with Go (Nic Jackson, O’Reilly, 2017)

11.5 会议与社区

  • GopherCon：年度 Go 大会，常有中间件与 Web 框架相关演讲。
  • Go Forum：https://forum.golangbridge.org/
  • r/golang：Reddit Go 社区。
  • CNCF Slack：云原生社区，中间件与服务网格讨论。

附录 A：源码索引

文件说明
net/http/server.goHandler、HandlerFunc、ServeMux 定义
net/http/request.goRequest、Context、WithContext
net/http/response.goResponseWriter 接口
net/http/httptest测试工具（NewRecorder、NewRequest）
context/context.goContext 接口与 WithValue/WithCancel
go-chi/chi/middleware/*.gochi 中间件实现

附录 B：常用中间件速查

中间件用途典型实现
Recovery捕获 panicdefer recover()
Logging记录请求日志time.Since(start)
RequestID生成请求 IDcrypto/rand
CORS跨域资源共享设置 Access-Control-* 头
Auth认证JWT / OAuth2
RateLimit限流令牌桶 / 滑动窗口
Timeout超时控制context.WithTimeout
Metrics指标采集Prometheus
Tracing链路追踪OpenTelemetry
Compression响应压缩gzip / brotli
CSRFCSRF 防护token 校验
GZIPGzip 压缩compress/gzip
Helmet安全头设置安全响应头

附录 C：术语表

术语英文说明
中间件Middleware包装 Handler 的高阶函数
装饰器模式Decorator Pattern不改变接口的前提下扩展行为
洋葱模型Onion Model请求由外向内、响应由内向外
高阶函数Higher-Order Function接收或返回函数的函数
ChainMiddleware Chain中间件组合
HandlerHandler处理 HTTP 请求的对象
HandlerFuncHandlerFunc函数适配器，使函数成为 Handler
ResponseWriterResponseWriter响应写入接口
ContextContext请求作用域数据与取消信号
自同态Endomorphism类型到自身的函数