1. 引言
關於閉包的說明，曾在很多篇幅中都有過一些說明，包括Go基礎--函數2, go 函數進階，異常與錯誤 都有所提到， 但是會發現，好像原理（理論）都懂，但是就是不知道如何使用，或者在看到一些源碼時，依然一頭霧水。 剛好今天看到了一篇優質的博客，並結合以前幾篇關於類似閉包的介紹，再次對 閉包進行一次分析。
2. 什么是閉包？
函數內部包含另一個函數: 且返回的是一個函數類型
2.1. 一個簡單的例子
【示例1】
package main
import (
"fmt"
)
// 累加器
func AddWrapper() func(int) int {
var n int = 10
return func(x int) int {
n = n + x
return n
}
}
func main() {
f := AddWrapper()
fmt.Println(f(1)) // 11
fmt.Println(f(2)) // 13
fmt.Println(f(3)) // 16
}
2.2理解閉包
- 閉包本身是一個函數, 且返回的一個函數
func (int) int - 返回的是一個匿名函數,但是這個匿名函數 引用到了函數(匿名)函數外的變量(上面示例是 n),因此這個匿名函數就和n 形成了一個整體, 構成閉包.
- 相比較類可以這么理解: 閉包是類(class),函數是類里面的某個方法, n是字段(屬性),函數和他使用到的n構成閉包
當反復調用AddWrapper 函數時, n只被初始化一次,因此每調用一次就進行累計
例如上面的代碼中，AddWrapper函數返回了一個匿名函數，而該匿名函數中引用了AddWrapper函數中的局部變量n ，那這個函數就是一個閉包。
調用這個函數，發現每一次調用，n 的值都會保留在 閉包函數中以待使用。
閉包中引用的外部局部變量並不會隨着 AddWrapper 函數的返回而被從棧上銷毀。
3. 復雜的閉包場景
寫一個閉包是比較容易的事，但單單會寫簡單的閉包函數，還遠遠不夠，如果不搞清楚閉包真正的原理，那很容易在一些復雜的閉包場景中對函數的執行邏輯進行誤判。
別的不說，就拿下來這個例子來說吧？
你覺得它會打印什么呢？
是 6 還是 11 呢？
【示例2】
import "fmt"
func func1() (i int) {
i = 10
defer func() {
i += 1
}()
// 相當於兩條語句
// i = 5
// return i
return 5
}
func main() {
closure := func1()
fmt.Println(closure)
}
先不運行，先來了解下閉包的底層原理
4.閉包的底層原理？
以最上面的例子進行分析
package main
import (
"fmt"
)
// 累加器
func AddWrapper() func(int) int {
var n int = 10
return func(x int) int {
n = n + x
return n
}
}
func main() {
f := AddWrapper()
fmt.Println(f(1)) // 11
fmt.Println(f(2)) // 13
fmt.Println(f(3)) // 16
}
先對它進行逃逸分析，很容易發現 n 作為 AddWrapper 函數局部變量，並不是分配在棧上，而是分配在堆上的。
D:\gopath\src\Go_base\daily_test\閉包> go build -gcflags="-m -m -l" demo.go
# command-line-arguments
.\demo.go:11:7: AddWrapper.func1 capturing by ref: n (addr=true assign=true width=8)
.\demo.go:10:9: func literal escapes to heap:
.\demo.go:10:9: flow: ~r0 = &{storage for func literal}:
.\demo.go:10:9: from func literal (spill) at .\demo.go:10:9
.\demo.go:10:9: from return func literal (return) at .\demo.go:10:2
.\demo.go:9:6: n escapes to heap:
.\demo.go:9:6: flow: {storage for func literal} = &n:
.\demo.go:9:6: from func literal (captured by a closure) at .\demo.go:10:9
.\demo.go:9:6: from n (reference) at .\demo.go:11:7
.\demo.go:9:6: moved to heap: n
.\demo.go:10:9: func literal escapes to heap
.\demo.go:21:15: f(3) escapes to heap:
.\demo.go:21:15: flow: {storage for ... argument} = &{storage for f(3)}:
.\demo.go:21:15: from f(3) (spill) at .\demo.go:21:15
.\demo.go:21:15: from ... argument (slice-literal-element) at .\demo.go:21:13
.\demo.go:21:15: flow: {heap} = {storage for ... argument}:
.\demo.go:21:15: from ... argument (spill) at .\demo.go:21:13
.\demo.go:21:15: from fmt.Println(... argument...) (call parameter) at .\demo.go:21:13
.\demo.go:20:15: f(2) escapes to heap:
.\demo.go:20:15: flow: {storage for ... argument} = &{storage for f(2)}:
.\demo.go:20:15: from f(2) (spill) at .\demo.go:20:15
.\demo.go:20:15: from ... argument (slice-literal-element) at .\demo.go:20:13
.\demo.go:20:15: flow: {heap} = {storage for ... argument}:
.\demo.go:20:15: from ... argument (spill) at .\demo.go:20:13
.\demo.go:20:15: from fmt.Println(... argument...) (call parameter) at .\demo.go:20:13
.\demo.go:19:15: f(1) escapes to heap:
.\demo.go:19:15: flow: {storage for ... argument} = &{storage for f(1)}:
.\demo.go:19:15: from f(1) (spill) at .\demo.go:19:15
.\demo.go:19:15: from ... argument (slice-literal-element) at .\demo.go:19:13
.\demo.go:19:15: flow: {heap} = {storage for ... argument}:
.\demo.go:19:15: from ... argument (spill) at .\demo.go:19:13
.\demo.go:19:15: from fmt.Println(... argument...) (call parameter) at .\demo.go:19:13
.\demo.go:19:13: ... argument does not escape
.\demo.go:19:15: f(1) escapes to heap
.\demo.go:20:13: ... argument does not escape
.\demo.go:20:15: f(2) escapes to heap
.\demo.go:21:13: ... argument does not escape
.\demo.go:21:15: f(3) escapes to heap
這就解決了第一個疑惑：為什么 AddWrapper 函數返回后， n 不會隨之銷毀？
可另一個問題，又浮現出來了，就算它不會銷毀，那閉包函數若是存儲的若是 sum 拷貝后的值，那每次調用閉包函數，里面的 n 應該都是一樣的，調用3次應該是11，12，13，而不是可以累加記錄。
因此，可以大膽猜測，閉包函數的結構體里存儲的是 n 的指針。
為了驗證這一猜想，只能上匯編了。
通過執行下面的命令，可以輸出對應的匯編代碼。
D:\gopath\src\Go_base\daily_test\閉包>go build -gcflags="-S" demo.go
# command-line-arguments
"".AddWrapper STEXT size=159 args=0x8 locals=0x20 funcid=0x0
0x0000 00000 (D:\gopath\src\Go_base\daily_test\閉包\demo.go:8) TEXT "".AddWrapper(SB), ABIInternal, $32-8
0x0000 00000 (D:\gopath\src\Go_base\daily_test\閉包\demo.go:8) MOVQ TLS, CX
0x0009 00009 (D:\gopath\src\Go_base\daily_test\閉包\demo.go:8) PCDATA $0, $-2
0x0045 00069 (D:\gopath\src\Go_base\daily_test\閉包\demo.go:10) LEAQ type.noalg.struct { F uintptr; "".n *int }(SB), CX
輸出的很多，提取出上面最關鍵的一行代碼，它定義了閉包函數的結構體。
其中 F 是函數的指針，但這不是重點，重點是 n 存儲的確實是指針.
type.noalg.struct { F uintptr; "".n *int }
5. 迷題揭曉
有了對閉包原理的認識，那對於第3節給出的這道題，很明顯可以知道輸出的內容是6。
import "fmt"
func func1() (i int) {
i = 10
defer func() {
i += 1
}()
return 5
}
func main() {
closure := func1()
fmt.Println(closure)
}
-
首先，由於
i在函數定義的返回值上聲明，因此根據 go 的caller-save模式， i 變量會存儲在 main 函數的棧空間。- Callee-saved register（又稱非易失性寄存器AKA non-volatile registers, or call-preserved）用於保存應在每次調用中保留的長壽命值。
- 調用者保存”（ caller saving ）方法：如果采用調用者保存策略，那么在一個調用者調用別的過程時，必須保存調用者所要保存的寄存器，以備調用結束返回后，能夠再次訪問調用者。
-
然后，func1 的 return 重新把 5 賦值給了 i ，此時 i = 5
-
由於閉包函數存儲了這個變量 i 的指針。
-
因此最后，在 defer 中對 i 進行自增，是直接更新到 i 的指針上，此時 i = 5+1，所以最終打印出來的結果是 6
6. 可能的坑
上面那題聽懂了的話，再來看看下面這道題。
func1 的返回值我們不寫變量名 i 了，然后原先返回具體字面量，現在改成變量 i ，就是這兩小小小的改動，會導致運行結果大大不同。
【示例3】
import "fmt"
func func1() (int) {
i := 10
defer func() {
// 函數內部的作用域並不會作用到外部
i += 1
}()
return i
}
func main() {
closure := func1()
fmt.Println(closure)
}
如果在返回值里寫了變量名，那么該變量會存儲 main 的棧空間里，而如果你不寫，那 i 只能存儲在 func1 的棧空間里，與此同時，return 的值，不會作用於原變量 i 上，而是會存儲在該函數在另一塊棧內存里。
因此你在 defer 中對原 i 進行自增，並不會作用到 func1 的返回值上。
所以打印的結果，只能是 10。
7. 總結
示例1中的 n是存儲在堆內存中的，而后面幾個示例都是存儲在棧內存里。
這是為什么呢？
仔細對比，不難發現，示例1返回的是閉包函數，閉包函數在 AddWrapper 返回后還要在其他地方繼續使用，在這種情況下，為了保證閉包函數的正常運行，無論閉包函數在哪里，i 都不能回收，所以 Go 編譯器會智能地將其分配在堆上。
而后面的其他示例，都只是涉及了閉包的特性，並不是直接把閉包函數返回，因此完全可以將其分配在棧上，非常的合理。
閉包的優勢: 類似於面向對象編程中, 引用的變量相當於 類中的一個屬性, 返回的匿名函數相當於修改了屬性值, 但是在每次調用時,不需要重復傳入引用的變量.