go语言实现装饰器

前言

装饰器作为一种特殊的设计模型，它可以让其他函数在不需要做任何代码修改的前提下增加额外功能，是面向切面编程思想的典型，在日志插入、性能测试、事务处理、缓存、权限校验等场景中应用广泛。 有了装饰器，我们就可以抽离出大量与函数功能本身无关的雷同代码到装饰器中并继续重用。概括的讲，装饰器的作用就是为已经存在的对象添加额外的功能。

装饰器实现

闭包

在Python程序中通常使用闭包和@关键字实现装饰器，那么在Go语言中也可以借鉴其思想。
在Go语言中匿名函数可以直接作为函数的返回值，因此Go语言中的闭包只需要实现一个嵌套函数即可。
下面是一个简单的闭包代码

func closure() func() int {
	i := 0
	return func() int {
		i += 1
		return i
	}
}

通过多次打印，可以发现外部函数closure在内部的变量i并没有随着函数周期结束而销毁，因此，每次调用内部函数时，变量i都会进行加一操作。

func character7() {
	nextNumber := closure()
	fmt.Println(nextNumber()) // 1
	fmt.Println(nextNumber()) // 2
	fmt.Println(nextNumber()) // 3
	fmt.Println(nextNumber()) // 4
}

测试函数

首先定义一个测试函数用于测试装饰器的效果，这是一个简单的累加操作，包括字符串和数字的累加，值得注意的是输入的参数可以为任意数量的值。

func addInt(params ...int) int {
	nums := 0
	for _, param := range params {
		nums += param
	}
	return nums
}

func addStr(params ...string) string {
	nums := ""
	for _, param := range params {
		nums += param
	}
	return nums
}

泛化类型装饰器

为了使装饰器能适应所有的类型函数，使用泛型定义，下面使一个计算函数运行时间的装饰器。

type MyAddType interface {
	// int | string
	any
}

func getTime[T MyAddType](f func(...T) T) func(...T) T {
	return func(params ...T) T {
		t1 := time.Now()
		res := f(params...)
		t2 := time.Now()
		fmt.Println("计算时间为：", t2.Sub(t1))
		return res
	}
}

测试和验证

分别验证字符型和数字型函数，addInt := getTime(addInt)等于Python中的@getTime。

func main() {
	addInt := getTime(addInt)
	addStr := getTime(addStr)
	contentInt := []int{}
	contentStr := []string{}
	for i := 1; i < 1000; i++ {
		contentInt = append(contentInt, i)
		contentStr = append(contentStr, "a")
	}
	fmt.Println(addInt(contentInt...))
	fmt.Println(addStr(contentStr...))
}

计算时间为： 0s
499500
计算时间为： 559µs
a…(此处省略999个a)