golang装饰器模式怎么实现 - 问答

在 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.” 的信息。

通过使用装饰器模式，我们可以在不修改原始函数代码的情况下，动态地增加额外的功能。

0 赞

0 踩