[[442450]]

本文轉(zhuǎn)載自微信公眾號(hào)「crossoverJie」，作者crossoverJie。轉(zhuǎn)載本文請(qǐng)聯(lián)系crossoverJie公眾號(hào)。

前言

最近實(shí)現(xiàn)某個(gè)業(yè)務(wù)時(shí)，需要讀取數(shù)據(jù)然后再異步處理;在 Go 中實(shí)現(xiàn)起來(lái)自然就比較簡(jiǎn)單，偽代碼如下：

list := []*Demo{{"a"}, {"b"}} 
for _, v := range list { 
 go func() { 
  fmt.Println("name="+v.Name) 
 }() 
} 
 
type Demo struct { 
 Name string 
} 

看似非常簡(jiǎn)單幾行代碼卻和我們的預(yù)期不符，打印之后輸出的是：

name=b 
name=b 

并不是我們預(yù)期的：

name=a 
name=b 

坑一

由于寫(xiě) go 的資歷尚淺、道行更是淺薄，這 bug 我硬是找了個(gè)把小時(shí);剛開(kāi)始還以為是數(shù)據(jù)源的問(wèn)題，經(jīng)歷了好幾輪自我懷疑。總之過(guò)程先不表，先看看如何修復(fù)這個(gè)問(wèn)題。

首先第一種辦法是使用臨時(shí)變量：

list := []*Demo{{"a"}, {"b"}} 
for _, v := range list { 
 temp:=v 
 go func() { 
  fmt.Println("name="+temp.Name) 
 }() 
} 

這樣便可正確輸出，其實(shí)從這種寫(xiě)法中也能看出問(wèn)題的端倪。

在第一種沒(méi)有使用臨時(shí)變量時(shí)，主協(xié)程很快就運(yùn)行完畢，這時(shí)候打印的子協(xié)程可能還沒(méi)運(yùn)行;當(dāng)開(kāi)始運(yùn)行的時(shí)候，這里的 v 已經(jīng)被最后一個(gè)賦值了。

所以這里打印的一直都是最后一個(gè)變量。

而使用臨時(shí)變量會(huì)將當(dāng)前遍歷的值拷貝一份，自然就不會(huì)互相影響了。

當(dāng)然除了臨時(shí)變量也可使用閉包解決。

list := []*Demo{{"a"}, {"b"}} 
for _, v := range list { 
 go func(temp *Demo) { 
  fmt.Println("name="+temp.Name) 
 }(v) 
} 

將參數(shù)通過(guò)閉包傳遞時(shí)，每個(gè) goroutine 都會(huì)在自己的棧中存放一份參數(shù)的拷貝，這樣也能區(qū)分了。

坑二

與之類似的還有第二個(gè)坑：

list2 := []Demo{{"a"}, {"b"}} 
var alist []*Demo 
for _, test := range list2 { 
 alist = append(alist, &test) 
} 
fmt.Println(alist[0].Name, alist[1].Name) 

這段代碼與我們預(yù)期不不符：

b b 

但我們稍加修改就可以了：

 list2 := []Demo{{"a"}, {"b"}} 
 var alist []Demo 
 for _, test := range list2 { 
  fmt.Printf("addr=%p\n", &test) 
  alist = append(alist, test) 
 } 
 fmt.Println(alist[0].Name, alist[1].Name) 
addr=0xc000010240 
addr=0xc000010240 
a b 

順便打印了內(nèi)存地址，其實(shí)從結(jié)果中大概就能猜到原因;每次遍歷打印的內(nèi)存地址都是相同，所以如果我們存放的是指針，本質(zhì)上存儲(chǔ)的都是同一塊內(nèi)存地址的內(nèi)容，所以值相同。

而如果我們只存儲(chǔ)值，不存指針自然也不會(huì)有這個(gè)問(wèn)題。

但如果想使用指針如何處理呢?

list2 := []Demo{{"a"}, {"b"}} 
var alist []*Demo 
for _, test := range list2 { 
 temp := test 
 //fmt.Printf("addr=%p\n", &test) 
 alist = append(alist, &temp) 
} 
fmt.Println(alist[0].Name, alist[1].Name) 

也簡(jiǎn)單，同樣的使用臨時(shí)變量即可。

通過(guò)官方源碼可以得知，for range 只是語(yǔ)法糖，本質(zhì)上也是 for 循環(huán);因?yàn)槊看味际菍?duì)同一個(gè)對(duì)象遍歷賦值，所以便會(huì)出現(xiàn)這樣的“烏龍”。

defer 的坑

for 循環(huán) + defer 也是組合坑(雖然不推薦這么用)，還是先來(lái)看個(gè)例子：

// demo1 
func main() { 
 a := []int{1, 2, 3} 
 for _, v := range a { 
  defer fmt.Println(v) 
 } 
} 
 
// demo2 
func main() { 
 a := []int{1, 2, 3} 
 for _, v := range a { 
  defer func() { 
   fmt.Println(v) 
  }() 
 } 
} 

分別輸出：

//demo1 
3 
2 
1 
//demo2 
3 
3 
3 

demo1的結(jié)果很好理解，defer 可以理解為將執(zhí)行語(yǔ)句放入到棧中，所以呈現(xiàn)的結(jié)果是先進(jìn)后出。

而demo2中，由于是閉包，閉包對(duì)變量 v 持有的是引用，所以在最終延遲執(zhí)行時(shí) v 已經(jīng)被最后一個(gè)值賦值，所以打印出來(lái)都是相同的。

解決方法與上文類似，傳入?yún)?shù)即可解決：

for _, v := range a { 
 defer func(v int) { 
  fmt.Println(v) 
 }(v) 
} 

這類細(xì)節(jié)問(wèn)題日常開(kāi)發(fā)大概率是碰不上的，最有可能遇到的就是面試了，所以多了解了解也沒(méi)壞處。

總結(jié)

類似于第一種情況在 for 循環(huán)中 goroutine 調(diào)用，我覺(jué)得 IDE 完全是可以做到提醒的;比如 IDEA 中就把大部分認(rèn)為可能發(fā)的錯(cuò)誤包含進(jìn)去，期待后續(xù) goland 的更新。

但其實(shí)這幾種錯(cuò)誤官方博客已經(jīng)提醒過(guò)了。

圖片https://github.com/golang/go/wiki/CommonMistakes#using-reference-to-loop-iterator-variable只是大部分人估計(jì)都沒(méi)去看過(guò)，這事之后我也得花時(shí)間好好閱讀下。