Go语言中间件职责链实现方法

Go语言中的HTTP中间件本质上是职责链模式的典型实践，其核心在于通过`func(http.Handler) http.Handler`函数签名构建可组合、可嵌套的处理链条，每个中间件必须正确调用`next.ServeHTTP(w, r)`以维持链路连通，并谨慎处理请求上下文（`*http.Request`需用`WithContext()`传递）、响应包装（自定义`ResponseWriter`须规避重复写头）及跨层数据共享（推荐使用类型安全的`context.WithValue`而非字符串key）。文章深入剖析了链式组装顺序、常见断链陷阱、调试验证技巧以及生产级避坑要点，为构建健壮、可维护的Go Web中间件体系提供了扎实的原理支撑与实战指南。

Go 里用 func(http.Handler) http.Handler 实现职责链最直接

HTTP 中间件本质就是职责链：每个环节处理请求、决定是否调用下一个。Go 标准库的 http.Handler 接口天然适配——只要返回一个新 http.Handler，就能串起链式调用。

常见错误是试图在中间件里直接 return 响应却不调用 next.ServeHTTP()，结果后续中间件永远不执行；或者忘了把 ResponseWriter 和 *http.Request 传给下一级，导致 panic 或数据丢失。

• 所有中间件必须接收 http.Handler 并返回 http.Handler，不能只写逻辑不包装
• 调用 next.ServeHTTP(w, r) 是链式继续的关键，漏掉就断链
• 如果要修改 *http.Request（比如加 context value），必须用 r.WithContext() 创建新请求对象再传下去

func logging(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        log.Printf("start: %s %s", r.Method, r.URL.Path)
        next.ServeHTTP(w, r) // 必须传原 w/r 或改造后的
        log.Printf("done")
    })
}

用 middleware.Chain 组合多个中间件时顺序不能反

职责链的执行顺序和退出时机高度依赖组装顺序。比如鉴权中间件必须放在日志之后、业务 handler 之前；否则日志可能记不到拒绝请求，或鉴权前就进了业务逻辑。

标准库没提供 Chain，但社区常用方式是手写一个可变参数函数，或用 alice 这类轻量库。自己实现时容易把参数展开顺序写反，导致链被倒着执行。

• Chain(m1, m2, m3).Then(h) 表示先 m1 → m2 → m3 → h，不是 h → m3 → m2 → m1
• 如果用 func(h http.Handler) http.Handler 手动嵌套，m1(m2(m3(h))) 才是对的，m3(m2(m1(h))) 就反了
• 调试时可在每个中间件开头加 log.Println("in m1")，看输出顺序验证链是否按预期流转

中间件里改写 ResponseWriter 要小心状态码和 header 冲突

有些中间件需要拦截响应体（如压缩、重写 HTML）、或统一设置 header（如 CORS）。这时必须用包装型 ResponseWriter，但 Go 的 http.ResponseWriter 接口极简，没有 StatusCode() 方法，容易误判响应是否已写出。

典型坑是：在 WriteHeader(401) 后又调用 Write([]byte{...})，但下游中间件不知道状态已发，还试图写 header，触发 http: multiple response.WriteHeader calls 错误。

• 自定义 ResponseWriter 必须实现 WriteHeader(int)、Write([]byte)、Header() http.Header 三个方法
• 用 written bool 字段记录是否已调用 WriteHeader，避免重复写头
• 不要在中间件里直接调用 w.Header().Set() 后再调用 next.ServeHTTP() —— 下游可能覆盖它

Context 传递值比全局变量或闭包更安全，但别滥用 context.WithValue

职责链中跨中间件传数据（如用户 ID、请求 ID）最稳妥的方式是往 context.Context 里塞，而不是靠闭包捕获变量或维护 map 查表。但 context.WithValue 不是万能钥匙，类型擦除和 key 冲突问题很实际。

常见错误是用字符串当 key：ctx = context.WithValue(ctx, "user_id", 123)，多个中间件一写就覆盖；或者取值时不检查 ok，直接类型断言 panic。

• key 必须是 unexported 类型（比如 type userIDKey struct{}），避免冲突
• 取值务必用两值断言：if uid, ok := ctx.Value(userIDKey{}).(int); ok { ... }
• 别往 context 塞大结构体或切片，影响 GC 和内存占用；只放轻量元数据

链越长，context 传递越关键，也越容易因一个中间件忘记传 context 而让下游拿不到值——检查每个 next.ServeHTTP() 前的 r = r.WithContext(newCtx) 是否存在。

好了，本文到此结束，带大家了解了《Go语言中间件职责链实现方法》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！