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

在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!”)}))
链式装饰器组合多个功能
你可以将多个装饰器串联起来,形成处理链。每个装饰器只关注单一职责,如认证、限流、日志等。
示例:添加身份验证和耗时统计:
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通过函数作为一等公民的特性,天然支持这种灵活的装饰模式。关键在于把“增强逻辑”从“核心逻辑”中解耦,提升代码可维护性和复用性。不复杂但容易忽略。
以上就是Golang如何使用装饰器模式增强方法功能的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1417514.html
微信扫一扫
支付宝扫一扫