如何使用Golang实现中间件链_请求预处理和后处理

16次阅读

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

如何使用Golang实现中间件链_请求预处理和后处理

在 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 包装器捕获)
  • 鉴权中间件(预处理):检查 Tokensession,不通过直接写响应并 return
    避免调用 next.ServeHTTP,中断链路
  • 请求体解析中间件(预处理):读取并解析 jsON,存入 ctx 或 request 的 context.WithValue
  • 响应头注入中间件(后处理):在 next.ServeHTTP 后调用 w.Header().Set(...)

使用 context 传递请求级数据

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

立即学习go语言免费学习笔记(深入)”;

  • 预处理中间件中:用 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))
text=ZqhQzanResources