介紹

在 Go 語言中，函數(shù)支持匿名函數(shù)，閉包就是一種特殊的匿名函數(shù)，它可以用于訪問函數(shù)體外部的變量。

需要注意的是，在 for ... range ... 中，使用 goroutine 執(zhí)行閉包時(shí)，經(jīng)常會(huì)掉“坑”。

因?yàn)槟涿瘮?shù)可以訪問函數(shù)體外部的變量，而 for ... range ... 返回的 val 的值是引用的同一個(gè)內(nèi)存地址的數(shù)據(jù)，所以匿名函數(shù)訪問的函數(shù)體外部的 val 值是循環(huán)中最后輸出的一個(gè)值。

“踩坑”示例代碼

func main() {
 done := make(chan bool)
 values := []string{"a", "b", "c"}
 for _, v := range values {
  go func() {
   fmt.Println(v)
   done <- true
  }()
 }
 // wait for all goroutines to complete before exiting
 for _ = range values {
  <-done
 }
}

輸出結(jié)果：

c

c

c

閱讀上面這段代碼，在 for ... range ... 中，使用 goroutine 執(zhí)行閉包，打印切片中的元素，實(shí)際輸出結(jié)果不是我們期望得到的輸出結(jié)果。

這是因?yàn)檠h(huán)的每次迭代都使用相同的變量 v 實(shí)例，因此每個(gè)閉包共享該單個(gè)變量。我們可以在示例代碼中簡(jiǎn)單修改，同時(shí)輸出變量 v 的內(nèi)存地址和值。

把 fmt.Println(v) 修改為 fmt.Printf("val=%s pointer=%p\n", v, &v)。

修改后的輸出結(jié)果：

val=c pointer=0xc000010200
val=c pointer=0xc000010200
val=c pointer=0xc000010200

我們可以在輸出結(jié)果中發(fā)現(xiàn)，打印變量 v 的內(nèi)存地址都是 0xc000010200。

當(dāng)閉包運(yùn)行時(shí)，它會(huì)在執(zhí)行 fmt.Println 時(shí)打印變量 v 的值，但 v 的值可能在 goroutine 啟動(dòng)后已被修改。感興趣的讀者朋友們可以使用 go vet 檢查。

怎么避免“踩坑”呢？

一種方法是將變量作為參數(shù)傳遞給閉包：

func main() {
 done := make(chan bool)
 values := []string{"a", "b", "c"}
 for _, v := range values {
  go func(param string) {
   // fmt.Println(v)
   fmt.Printf("val=%s pointer=%p\n", param, &param)
   done <- true
  }(v)
 }
 // wait for all goroutines to complete before exiting
 for _ = range values {
  <-done
 }
}

輸出結(jié)果：

val=c pointer=0xc000010200
val=a pointer=0xc00009a000
val=b pointer=0xc0000a4000

閱讀上面這段代碼，通過將變量 v 的值作為參數(shù)傳遞給閉包，然后，該值作為形參 param 的值，在函數(shù)體內(nèi)部被訪問。

另外一種方法是創(chuàng)建一個(gè)新變量

func main() {
 done := make(chan bool)
 values := []string{"a", "b", "c"}
 for _, v := range values {
  param := v
  go func() {
   // fmt.Println(v)
   fmt.Printf("val=%s pointer=%p\n", param, &param)
   done <- true
  }()
 }
 // wait for all goroutines to complete before exiting
 for _ = range values {
  <-done
 }
}

輸出結(jié)果：

val=c pointer=0xc000082200
val=a pointer=0xc0000821e0
val=b pointer=0xc0000821f0

通過輸出結(jié)果可以發(fā)現(xiàn)，該種方式也可以達(dá)到我們期望的結(jié)果。

總結(jié)

本文我們介紹了在 for ... range ... 中，Go 語言在每次迭代時(shí)，沒有定義一個(gè)新變量，導(dǎo)致使用 goroutine 運(yùn)行閉包時(shí)，經(jīng)常會(huì)掉“坑”。

我們給出避免“踩坑”的兩種方法，其中，第二種方法更簡(jiǎn)單，更多關(guān)于Go goroutine運(yùn)行閉包的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！