在 Go 語言中,可以使用函數(shù)作為參數(shù)和返回值的特性來實(shí)現(xiàn)裝飾器模式。
下面是一個示例代碼,演示了如何使用裝飾器模式來增加函數(shù)的功能:
package main
import "fmt"
// 定義一個原始的函數(shù)類型
type FooFunc func()
// 原始的函數(shù)
func Foo() {
fmt.Println("This is the original function.")
}
// 裝飾器函數(shù)類型
type DecoratorFunc func(FooFunc) FooFunc
// 裝飾器函數(shù),用于在原始函數(shù)執(zhí)行前后增加額外的功能
func Decorator(f FooFunc) FooFunc {
return func() {
fmt.Println("Before calling the original function.")
f() // 調(diào)用原始函數(shù)
fmt.Println("After calling the original function.")
}
}
func main() {
// 創(chuàng)建一個裝飾器函數(shù)
decoratedFoo := Decorator(Foo)
// 調(diào)用裝飾后的函數(shù)
decoratedFoo()
}
運(yùn)行以上代碼,輸出結(jié)果為:
Before calling the original function.
This is the original function.
After calling the original function.
在這個示例代碼中,Foo
是一個原始的函數(shù),Decorator
是一個裝飾器函數(shù)。裝飾器函數(shù)接受一個函數(shù)類型作為參數(shù),返回一個具有相同函數(shù)類型的函數(shù)。
裝飾器函數(shù)在調(diào)用原始函數(shù)之前和之后都可以添加額外的功能代碼。在這個例子中,裝飾器函數(shù)在調(diào)用原始函數(shù)之前打印了一行 “Before calling the original function.” 的信息,在調(diào)用原始函數(shù)之后打印了一行 “After calling the original function.” 的信息。
通過使用裝飾器模式,我們可以在不修改原始函數(shù)代碼的情況下,動態(tài)地增加額外的功能。