GoLang学习笔记（十三)闭包的定义及用法

其实匿名函数就是闭包

写一个不使用闭包的实现的计数器

func add1() {
	val := 0
	for i := 0; i < 10; i++ {
		fmt.Printf("i=%d \t", i)
		val += add2(i)
		fmt.Println(val)
	}
	fmt.Println("总和为：", val)
}

func add2(x int) int {
	sum := 0
	sum += x
	return sum
}

当运行add1()函数的时候，会给出结果

i=0 	0
i=1 	1
i=2 	3
i=3 	6
i=4 	10
i=5 	15
i=6 	21
i=7 	28
i=8 	36
i=9 	45
总和为： 45

记录了计数并叠加的结果。

对比一下，再写一个使用闭包的计数器，并和上面的输出结果一致

func adder1() {
	result := adder2()
	for i := 0; i < 10; i++ {
		fmt.Printf("i=%d \t", i)
		fmt.Println(result(i))
	}
}

func adder2() func(int) int {
	sum := 0
	result := func(num int) int {
		sum += num
		return sum
	}
	return result
}

最后来写另外一个闭包计数器，当每次调用方法时，计数器加1。

func counter1() {
	res := counter2()
	fmt.Printf("%T \n", res)
	fmt.Println("res:", res)   
	fmt.Println("res:", res()) 
	fmt.Println("res:", res()) 
	fmt.Println("res:", res()) 
}

func counter2() func() int {
	i := 0
	res := func() int {
		i++
		return i
	}
	return res
}

当执行counter1()函数里的res()一次，计数器加1一次，所以输出的结果第一行就是res的数据类型，第二行就是res的内存地址，第三行为1，第四行为2，第五行为3……以此类推。

func() int 
res: 0x490070
res: 1
res: 2
res: 3

闭包方式引用defer语句，会导致一些问题：

func testCounter() {
	for i := 0; i < 3; i++ {
		defer func() { println(i) }()
	}
}

运行后，返回如下：

3
3
3

因为是闭包，在for迭代语句中，每个defer语句延迟执行的函数时引用的同一个i迭代变量，所以都是循环结束后的3。

解决方案：

第一，在循环体中，创建一个实体变量，每次在defer语句中打印那个实体变量。

第二，将迭代变量i，通过闭包参数的方式传递进去，defer语句会马上对参数进行求值。

func testCounter01() {
	for i := 0; i < 3; i++ {
		//定义一个循环体内局部变量i
		i := i
		defer func() { println(i) }()
	}
}

func testCounter02() {
	for i := 0; i < 3; i++ {
		//通过函数传入i，defer会马上对调用参数求值
		defer func(i int) {println(i)}(i)
	}
}

这两个方法的运行结果都是：

2
1
0

注：这两种方式都是可以工作的，但不建议在for循环中使用defer语句。

2019年9月12日，添加一个闭包新的使用例子：

func testFunction01(suffix string) func(string) string {
	return func(str string) string {
		if !strings.HasSuffix(str, suffix) {
			return str + suffix
		} else {
			return str
		}
	}
}

func main() {
	tf01 := testFunction01(".aaa")
	fmt.Println(tf01("aaaaa"))        //aaaaa.aaa
	fmt.Println(tf01("bbbbb.aaa"))    //bbbbb.aaa
	fmt.Println(tf01("ccccc.ccc"))    //ccccc.ccc.aaa
}

该例的作用是判断一串字符串是否以指定的字符串结尾，如果不是则加上指定的字符串，如果是则不变。

（该方法可以判断文件的类型，并修改文件名）

当tf01初始化时，利用闭包驻留的特性，后缀名.aaa就已进入参，再次调用时则不必再次输入。

可反复使用该变量。