17370845950

Go中间件链通过函数式包装http.Handler实现，预处理逻辑在handler前执行（如鉴权），后处理在后执行（如日志、响应头），推荐用context传递数据并封装Chain工具链式调用。

在 Go 中实现中间件链，核心是利用函数式编程思想：每个中间件接收一个 http.Handler 并返回一个新的 http.Handler，形成可组合、可复用的处理链。请求预处理（如日志、鉴权、限流）放在 handler 执行前，后处理（如响应头注入、耗时统计）放在 handler 执行后。

基础中间件链结构

Go 的 http.Handler 接口只有一个方法：ServeHTTP(http.ResponseWriter, *http.Request)。中间件本质是“包装器”：

• 输入：原始 handler（通常是最终业务逻辑）
• 输出：新 handler，内部调用原 handler 前/后插入自定义逻辑
• 链式调用通过嵌套包装实现，例如 mw3(mw2(mw1(handler)))

编写典型中间件示例

以下为常见场景的中间件写法，注意顺序影响执行时机：

• 日志中间件（后处理为主）：记录请求路径、状态码、耗时
  在 next.ServeHTTP 后读取状态码（需用 ResponseWriter 包装器捕获）
• 鉴权中间件（预处理）：检查 token 或 session，不通过直接写响应并 return
  避免调用 next.ServeHTTP，中断链路
• 请求体解析中间件（预处理）：读取并解析 JSON，存入 ctx 或 request 的 context.WithValue
• 响应头注入中间件（后处理）：在 next.ServeHTTP 后调用 w.Header().Set(...)

使用 context 传递请求级数据

中间件间共享数据推荐用 request.Context()，而非全局变量或自定义 struct：

• 预处理中间件中：用 r = r.WithContext(context.WithValue(r.Context(), key, value)) 注入数据
• 后续中间件或 handler 中：用 r.Context().Value(key) 获取
• key 建议用私有类型（如 type userIDKey struct{}）避免冲突

封装链式调用工具函数

手动嵌套 mw1(mw2(mw3(handler))) 易出错且难维护，可封装 Chain 工具：

• 定义类型 type Middleware func(http.Handler) http.Handler
• 实现 func Chain(h http.Handler, mws ...Middleware) http.Handler
• 内部从右到左应用中间件（即 mws[0](mws[1](...mws[n](h)...))），符合直觉
• 使用示例：http.ListenAndServe(":8080", Chain(myHandler, logging, auth, headers))