在 Go 语言中,可以使用函数作为参数和返回值的特性来实现装饰器模式。
下面是一个示例代码,演示了如何使用装饰器模式来增加函数的功能:
package main
import "fmt"
// 定义一个原始的函数类型
type FooFunc func()
// 原始的函数
func Foo() {
fmt.Println("This is the original function.")
}
// 装饰器函数类型
type DecoratorFunc func(FooFunc) FooFunc
// 装饰器函数,用于在原始函数执行前后增加额外的功能
func Decorator(f FooFunc) FooFunc {
return func() {
fmt.Println("Before calling the original function.")
f() // 调用原始函数
fmt.Println("After calling the original function.")
}
}
func main() {
// 创建一个装饰器函数
decoratedFoo := Decorator(Foo)
// 调用装饰后的函数
decoratedFoo()
}
运行以上代码,输出结果为:
Before calling the original function.
This is the original function.
After calling the original function.
在这个示例代码中,Foo
是一个原始的函数,Decorator
是一个装饰器函数。装饰器函数接受一个函数类型作为参数,返回一个具有相同函数类型的函数。
装饰器函数在调用原始函数之前和之后都可以添加额外的功能代码。在这个例子中,装饰器函数在调用原始函数之前打印了一行 “Before calling the original function.” 的信息,在调用原始函数之后打印了一行 “After calling the original function.” 的信息。
通过使用装饰器模式,我们可以在不修改原始函数代码的情况下,动态地增加额外的功能。