阅读量:2
在 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.” 的信息。
通过使用装饰器模式,我们可以在不修改原始函数代码的情况下,动态地增加额外的功能。