首页 > 后端开发 > Golang > 正文

Golang装饰器模式写法 函数包装扩展功能

P粉602998670
发布: 2025-08-23 12:56:01
原创
588人浏览过
Go语言通过高阶函数实现装饰器模式,以函数包装函数的方式扩展功能而不修改原逻辑。1. 定义统一函数类型如HandlerFunc;2. 编写基础函数如Hello;3. 创建装饰器函数WithLogging添加日志;4. 实现WithTiming统计耗时;5. 支持链式组合如WithLogging(WithTiming(Hello)),执行顺序由外向内;6. 可针对HTTP场景封装HTTPLogger等通用装饰器。该模式依赖类型匹配与调用顺序,适用于日志、中间件等场景。

golang装饰器模式写法 函数包装扩展功能

在Go语言中,虽然没有像Python那样的

@decorator
登录后复制
语法糖,但通过函数式编程的思想,可以很自然地实现装饰器模式。装饰器本质是“函数包装函数”,在不修改原函数逻辑的前提下,为其扩展功能,比如日志记录、权限校验、耗时统计等。

1. 基本函数类型定义

要实现装饰器,先定义统一的函数签名类型。比如处理HTTP请求的处理器

type HandlerFunc func(string) string
登录后复制

这个函数接收一个字符串参数,返回一个字符串结果。你可以根据实际场景替换成

http.HandlerFunc
登录后复制
或其他类型。

2. 编写基础功能函数

先写一个简单的业务函数:

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

func Hello(name string) string {
    return "Hello, " + name
}
</font>
登录后复制

3. 实现装饰器:添加日志功能

写一个装饰器函数,接收原函数,返回一个功能增强的新函数:

func WithLogging(fn HandlerFunc) HandlerFunc {
    return func(name string) string {
        fmt.Printf("调用函数,参数: %s\n", name)
        result := fn(name)
        fmt.Printf("函数返回: %s\n", result)
        return result
    }
}
</font>
登录后复制

这个装饰器在调用前后打印日志,原函数逻辑不受影响。

火龙果写作
火龙果写作

用火龙果,轻松写作,通过校对、改写、扩展等功能实现高质量内容生产。

火龙果写作 106
查看详情 火龙果写作

4. 实现多个装饰器组合

可以再写一个统计执行时间的装饰器:

func WithTiming(fn HandlerFunc) HandlerFunc {
    return func(name string) string {
        start := time.Now()
        result := fn(name)
        fmt.Printf("执行耗时: %v\n", time.Since(start))
        return result
    }
}
</font>
登录后复制

多个装饰器可以链式调用:

handler := WithLogging(WithTiming(Hello))
result := handler("Alice")
</font>
登录后复制

执行顺序是从外到内:先日志 → 再计时 → 调用Hello。

5. 更通用的装饰器写法(支持任意函数)

如果想支持不同签名的函数,可以用

interface{}
登录后复制
和反射,但更推荐针对场景封装。例如针对
http.HandlerFunc
登录后复制
的装饰器:

func HTTPLogger(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        fmt.Printf("[%s] %s\n", r.Method, r.URL.Path)
        next(w, r)
    }
}
</font>
登录后复制

在HTTP服务中这样使用:

http.HandleFunc("/hello", HTTPLogger(func(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintln(w, "Hello World")
}))
</font>
登录后复制

Go的装饰器模式依赖高阶函数,通过传入函数并返回新函数来实现功能增强。这种方式清晰、灵活,适合中间件、日志、监控等场景。基本上就这些,不复杂但容易忽略的是类型匹配和调用顺序。

以上就是Golang装饰器模式写法 函数包装扩展功能的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号