Golang如何使用装饰器模式扩展功能

Go语言通过函数式编程和接口组合实现装饰器模式,可在不修改原对象的情况下动态添加功能。1. 函数装饰器常用于HTTP处理链,如日志、认证、限流,通过高阶函数包装原函数并在前后插入逻辑;2. 结构体嵌入可模拟面向对象的装饰器,内嵌原对象并重写方法,适用于服务对象增强,如缓存装饰读取器;3. 多个装饰器可链式组合形成处理管道,提升代码复用与可维护性,关键在于清晰接口设计和单一职责原则。

Go语言没有类和继承,但可以通过函数式编程和接口组合实现装饰器模式。这种模式允许你在不修改原始对象的前提下,动态地给对象添加新功能。在Golang中,装饰器通常通过高阶函数或结构体嵌入来实现,特别适用于日志、权限校验、限流等横切关注点。

使用函数装饰器增强行为

函数装饰器是Golang中最常见的装饰方式,尤其适合HTTP处理函数或服务方法的包装。你可以定义一个接收函数并返回增强后函数的高阶函数。

  • 定义原始处理逻辑,比如一个HTTP handler
  • 编写装饰函数,接收原函数作为参数,在调用前后插入额外逻辑
  • 返回一个新的函数,具备原功能+新增能力

例如,为HTTP处理器添加请求日志:

func loggingMiddleware(next http.HandlerFunc) http.HandlerFunc {
  return func(w http.ResponseWriter, r *http.Request) {
    log.Printf("%s %s", r.Method, r.URL.Path)
    next(w, r)
  }
}

使用时链式包装:
http.HandleFunc("/", loggingMiddleware(myHandler))

通过结构体嵌入实现对象装饰

当需要装饰的是某个服务对象时,可以使用结构体嵌入模拟“继承+扩展”。这种方式更接近传统面向对象中的装饰器模式。

  • 定义统一接口,所有实现和装饰器都遵循该接口
  • 基础服务实现接口
  • 装饰器结构体内嵌基础服务,并重写需要增强的方法

比如有一个数据读取接口:

type Reader interface {
  Read() string
}

基础实现:

type FileReader struct{}
func (f *FileReader) Read() string { return "data from file" }

缓存装饰器:

type CachedReader struct {
  Reader
  cache string
}
func (c *CachedReader) Read() string {
  if c.cache == "" {
    c.cache = c.Reader.Read()
  }
  return c.cache
}

使用:
reader := &CachedReader{Reader: &FileReader{}}

多层装饰与链式组合

装饰器的优势在于可叠加。你可以将多个装饰器串联起来,形成处理管道。

继续以HTTP为例,组合日志、认证、超时控制:

handler := withLogging(withAuth(withTimeout(originalHandler)))

也可以写成更清晰的链式调用:

stack := middleware.Chain(loggingMiddleware, authMiddleware, timeoutMiddleware)
final := stack(originalHandler)

这种模式让代码职责分离,每个装饰器只关心自己的逻辑。

基本上就这些。Golang虽然语法简洁,但通过函数和接口的灵活组合,完全可以实现强大的装饰器模式,提升代码复用性和可维护性。关键是设计好边界清晰的接口,并保持装饰器的单一职责。