Golang如何使用装饰器模式增强方法功能

go语言可通过高阶函数实现装饰器模式,如用LoggingMiddleware为http处理函数添加日志;支持链式组合多个装饰器,执行顺序从外到内;还可利用泛型接口实现通用装饰器,如为函数添加重试机制。

Golang如何使用装饰器模式增强方法功能

Go语言中,虽然没有像python那样的语法糖直接支持装饰器,但可以通过函数式编程的思想实现类似装饰器模式的效果。这种方式能让你在不修改原函数逻辑的前提下,增强或修改其行为,比如添加日志、权限校验、耗时统计等功能。

使用函数包装实现基础装饰器

Go中的装饰器通常通过高阶函数实现:将一个函数作为参数传入另一个函数,并返回一个新的函数。这个新函数可以在调用原函数前后插入额外逻辑。

例如,为一个HTTP处理函数添加日志功能:

func loggingMiddleware(next http.HandlerFunc) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { fmt.Printf(“Request: %s %sn”, r.Method, r.URL.Path) next(w, r) fmt.Println(“Request completed”) } }

使用方式:

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

http.HandleFunc(“/”, loggingMiddleware(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintln(w, “Hello, World!”) }))

链式装饰器组合多个功能

你可以将多个装饰器串联起来,形成处理链。每个装饰器只关注单一职责,如认证、限流、日志等。

示例:添加身份验证和耗时统计:

Golang如何使用装饰器模式增强方法功能

AI图像编辑器

使用文本提示编辑、变换和增强照片

Golang如何使用装饰器模式增强方法功能46

查看详情 Golang如何使用装饰器模式增强方法功能

func authMiddleware(next http.HandlerFunc) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { Token := r.Header.Get(“Authorization”) if token == “” { http.Error(w, “Unauthorized”, http.StatusUnauthorized) return } next(w, r) } } func timingMiddleware(next http.HandlerFunc) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { start := time.Now() next(w, r) fmt.printf(“Request took: %vn”, time.Since(start)) } }

组合使用:

handler := loggingMiddleware(authMiddleware(timingMiddleware(myHandler))) http.HandleFunc(“/”, handler)

执行顺序是从外到内:logging → auth → timing → myHandler,返回时反向。

通用函数装饰器处理不同类型函数

除了HTTP处理器,你也可以为普通函数编写装饰器。通过接口或泛型(Go 1.18+)提高复用性。

例如,为任意无参无返回的函数添加重试机制:

func withRetry(fn func() error, maxRetries int) func() error { return func() error { var err error for i := 0; i ailed: %vn”, i+1, err) time.Sleep(time.Second) } return fmt.Errorf(“failed after %d retries: %w”, maxRetries, err) } }

使用示例:

work := func() error { // 模拟可能失败的操作 return errors.New(“network error”) } retryWork := withRetry(work, 3) retryWork()

基本上就这些。Go通过函数作为一等公民的特性,天然支持这种灵活的装饰模式。关键在于把“增强逻辑”从“核心逻辑”中解耦,提升代码可维护性和复用性。不复杂但容易忽略。

上一篇
下一篇
text=ZqhQzanResources