Go | 闭包使用

Go | 闭包使用

闭包基本介绍

闭包就是 一个函数 和其相关的 引用环境 组合的一个整体

• 好处： 保存引用的变量，下次继续使用，不会销毁

举例

例1-闭包实现数字累加

// 闭包实现数字累加
package closure

import (
	"fmt"
)

// AddUpper是一个函数，返回的数据类型 func(int) int
// AddUpper 返回的是一个匿名函数，但是这个匿名函数引用到了函数外的变量 n ，
// 因此这个匿名函数就和 n 形成一个整体，构成闭包;
// 当反复调用 f 函数时，因为 n 只初始化一次，保存了变量的值，因此每调用一次就相当于进行了累加。
func AddUpper() func(int) int {
	var n int = 100
	var str = "hello"
	return func(i int) int {
		n = n + i
		fmt.Println("i=", i)
		str += string(byte(36)) // ascii 36 = '$'
		fmt.Printf("str==%s\n", str)
		return n
	}
}

package closure

import (
	"fmt"
	"testing"
)

func TestAddUpper(t *testing.T) {

	f := AddUpper()
	fmt.Println(f(1)) // 101
	fmt.Println(f(2)) // 103
	fmt.Println(f(3)) // 106

}

例2

需求：

• 编写一个函数 makeSuffix(suffix string) ，可以接收一个文件后缀名，并返回一个闭包
• 调用闭包，可以传入一个文件名，如果该文件名没有指定后缀，则返回 文件名.jpg ，如果已经有.jpg，则返回原文件名。
• strings.HasSuffix，该函数可以判断某个字符串是否有指定的后缀。

func MakeSuffix(suffix string) func(string) string {
	return func(name string) string {
		if !strings.HasSuffix(name, suffix) {
			return name + suffix
		}
		return name
	}
}

// MakeSuffixV2 传统写法
func MakeSuffixV2(suffix string, name string) string {
	if !strings.HasSuffix(name, suffix) {
		return name + suffix
	}
	return name
}

func TestMakeSuffix(t *testing.T) {

	// 闭包调用
	f := MakeSuffix(".jpg")
	fmt.Println(f("xiao"))         // xiao.jpg
	fmt.Println(f("xiaoxiao.jpg")) // xiaoxiao.jpg
	fmt.Println(f("xiaoxiao.666")) // xiaoxiao.666.jpg

}

func TestMakeSuffixV2(t *testing.T) {

	// 传统写法调用
	fmt.Println("MakeSuffixV2=", MakeSuffixV2(".jpg", "allblue"))  // makeSuffixV2= allblue.jpg
	fmt.Println("MakeSuffixV2=", MakeSuffixV2(".jpg", "all.blue")) // makeSuffixV2= all.blue.jpg

}

代码说明:

• 返回的匿名函数和 MakeSuffix(suffix string) 的 suffix 变量组合成一个闭包

• 传统写法和闭包写法实现效果一样，但是，传统写法需要重复写变量， 比如上面的

makeSuffixV2(".jpg", "all.blue"))

闭包则解决了这个问题，是代码看起来更加的简洁

闭包的好处

1.参数复用，保存引用的变量，下次继续使用，不会销毁