闭包基本介绍
闭包就是 一个函数 和其相关的 引用环境 组合的一个整体
好处： 保存引用的变量，下次继续使用，不会销毁
下面通过闭包的方式，写一个数字累加器，体验一下闭包的妙处👇
闭包实现数字累加
package main import "fmt" // 累加器 // 闭包 - 函数柯里化 // 返回值类型： func(int) int func AddUpper() func(int) int { var n int = 100 return func(i int) int { n = n + i return n } } func main() { f := AddUpper() fmt.Println(f(1)) // 101 fmt.Println(f(2)) // 103 fmt.Println(f(3)) // 106 }
代码说明
- AddUpper是一个函数，返回的数据类型 func(int) int
- 闭包的说明
返回的是一个匿名函数，但是这个匿名函数引用到了函数外的变量 n ，因此这个匿名函数就和 n 形成一个整体，构成闭包
- 当反复调用
f函数时，因为 n 只初始化一次，保存了变量的值，因此每调用一次就相当于进行了累加。
- 我们要搞清楚闭包，关键就是要分析返回的函数使用到哪些变量
代码分析
这里我引入了一个字符串变量str，来帮助分析闭包是怎么保存变量的。
package main import ( "fmt" ) // 累加器 // 闭包 - 函数柯里化 // 返回值类型： func(int) int 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(36) // ascii 36 = '$' fmt.Printf("str==%s\n", str) return n } } func main() { f := AddUpper() // fmt.Println(AddUpper()(1)) // 101 fmt.Println("f(1)=", f(1)) // 101 fmt.Println("f(2)=", f(2)) // 103 fmt.Println("f(3)=", f(3)) // 106 } i= 1 str==hello$ f(1)= 101 i= 2 str==hello$$ f(2)= 103 i= 3 str==hello$$$ f(3)= 106
从输出可以看出来，闭包引用的变量n和str并没有在调用函数的时候重复声明，而是保留了下次函数调用后更新的值。
闭包案例
需求：
- 编写一个函数 makeSuffix(suffix string) ，可以接收一个文件后缀名，并返回一个闭包
- 调用闭包，可以传入一个文件名，如果该文件名没有指定后缀，则返回 文件名.jpg ，如果已经有.jpg，则返回原文件名。
strings.HasSuffix，该函数可以判断某个字符串是否有指定的后缀。
上代码
package main import ( "fmt" "strings" ) func makeSuffix(suffix string) func(string) string { return func(name string) string { if !strings.HasSuffix(name, suffix) { return name + suffix } return name } } // 传统写法 func makeSuffixV2(suffix string, name string) string { if !strings.HasSuffix(name, suffix) { return name + suffix } return name } func main() { // 闭包调用 f := makeSuffix(".jpg") fmt.Println(f("xiao")) // xiao.jpg fmt.Println(f("xiaoxiao.jpg")) // xiaoxiao.jpg fmt.Println(f("xiaoxiao.666")) // xiaoxiao.666.jpg // 传统写法调用 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"))
闭包则解决了这个问题，是代码看起来更加的简洁
闭包的好处之一： 参数复用
好处： 保存引用的变量，下次继续使用，不会销毁
