Golang装饰器模式详解｜快速入门教程

Go虽无Python式的@decorator语法，却能通过高阶函数、统一HandlerFunc签名（如func(context.Context) error）和严谨的context透传，实现同样自然、可链式组合的装饰器效果；本文直击实战痛点——从避免日志装饰器阻塞IO、HTTP中间件中http.HandlerFunc转换陷阱，到用Chain扁平化嵌套调用、结构体与函数式装饰器的选型逻辑，再到所有装饰器必须透传上游context以保障超时/取消/链路追踪不失效——手把手教你写出健壮、可测、真正可用的Go装饰器。

Go 没有装饰器语法，但用高阶函数+统一签名就能写出和 Python @decorator 一样自然的装饰效果——关键不是“像不像”，而是“能不能链、会不会断、ctx 传没传”。

怎么写一个能链起来的日志装饰器

最常见错误是写成 func() error 签名，结果加了超时或重试就套不动：因为 WithTimeout 要求接收 func(context.Context) error，而你传进去的是无参函数，编译直接报错。

• 统一用 type HandlerFunc func(context.Context) error（或带参数/返回值的变体）作为契约
• 装饰器必须返回同类型函数，且内部调用时显式传入 ctx：next(ctx)，不能漏
• 日志装饰器里别同步写磁盘文件，否则整个请求卡住；用 log.Printf 或异步 logger

func WithLogging(next HandlerFunc) HandlerFunc {
    return func(ctx context.Context) error {
        log.Printf("→ %s start", ctx.Value("req_id"))
        err := next(ctx)
        log.Printf("← %s done, err: %v", ctx.Value("req_id"), err)
        return err
    }
}

为什么 HTTP 中间件是最稳的入门场景

因为 http.Handler 接口强制了签名一致：func(http.ResponseWriter, *http.Request)，所有中间件都按这个来，天然可链。新手在这里踩坑，90% 是忘了 http.HandlerFunc 这层转换。

• 不转 http.HandlerFunc，闭包就不是 http.Handler，http.Handle 直接拒绝注册
• 中间件里必须调用 next.ServeHTTP(w, r)，漏掉就等于请求静默丢弃
• 顺序决定执行流：写成 logging(auth(handler))，实际是 logging 先进、最后出，auth 在中间

func authMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if r.Header.Get("X-Token") == "" {
            http.Error(w, "no token", http.StatusUnauthorized)
            return // 不调 next.ServeHTTP → 请求终止
        }
        next.ServeHTTP(w, r) // 必须有
    })
}

多个装饰器嵌套时，怎么避免括号套娃

写成 WithRetry(3)(WithTimeout(5*time.Second)(WithLogging(handler))) 看着就晕，而且容易手抖少个括号，编译失败还不好定位。

• 用 Chain 辅助函数把装饰器列表“摊平”：Chain(WithLogging, WithTimeout(5*time.Second), WithRetry(3))(handler)
• Chain 内部必须从右往左包裹（即最后一个装饰器最靠近 handler），否则执行顺序反了
• 每个装饰器仍要遵守“接收 HandlerFunc，返回 HandlerFunc”的契约，不然 Chain 无法泛化

结构体嵌入 vs 函数式装饰器，选哪个

函数式更轻、更易测试、组合自由；结构体嵌入更适合需要维护状态的场景（比如熔断器要记失败次数、限流器要管令牌桶）。但别为了“像 OOP”硬套结构体。

• 纯日志、超时、认证？用函数式，零依赖、无状态、好 mock
• 重试带退避、熔断器需状态、指标收集要计数？结构体封装更清晰，比如 type Retrier struct { maxRetries int; backoff time.Duration }
• 混用也可以：结构体实现 Handler 接口，再被函数式中间件包一层，只要签名对得上

最容易被忽略的一点：所有装饰器里的 context.Context 都得透传，不能在某一层自己 new 一个 ctx 后就扔掉上游的 cancel 和 deadline —— 否则超时、取消、trace propagation 全失效。

今天关于《Golang装饰器模式详解｜快速入门教程》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！