Golang 代理（装饰器）Decorator

Java里可以通过注解+反射的方式在目标方法神不知鬼不觉的情况下将它的参数，结果等数据进行修改，这种功能在GO里可以用装饰器来实现

所谓装饰器就是一个高阶函数，说白了就是参数是一个func，返回值也是一个func，如下

func Foo(stuff Stuff) stuff Stuff {}

// 构造一个与被代理函数一毛一样的方法
type Decorator func(a int, b int) int

// 原方法（将要被代理的方法）
func Foo(a int, b int) int {
    fmt.Println("Foo Running...")
    return a + b
}

// 构造一个高阶方法来入侵原方法
func MyDecorator(decorator Decorator) Decorator {
    return func(a int, b int) int {
        fmt.Println("Foo Run Before...")
        result := decorator(a, b)
        fmt.Println("Foo Run After...")
        return result
    }
}
// 如果不想构造装饰器也可以写成下面这种
// 这样的话，就不需要定义 type Decorator func(a int, b int) int
// func MyDecorator(decorator func(a int, b int) int) func(int, int) int {
//     return func(a int, b int) int {
//         fmt.Println("Foo Run Before...")
//         result := decorator(a, b)
//         fmt.Println("Foo Run After...")
//         return result
//     }
// }

func TestDecorator(t *testing.T) {
    result := MyDecorator(Foo)                       // 通过装饰器拿到被装饰（代理）的对象
    fmt.Printf("Foo Run Result: %d\n", result(1, 2)) // 通过执行装饰（代理）对象来实现对被装饰（代理）对象的"增强"
}

Golang 代理（装饰器）Decorator

相关文章