下面由golang教程欄目給大家分享Go的循環(huán)遍歷使用小坑 ,希望對需要的朋友有所幫助!
在Golang的流程控制中,循環(huán)語句有for和range兩種。
for語句
1.for 賦值表達式; 關(guān)系表達式或邏輯表達式; 賦值表達式 { }
for i := 0; i < 10; i++ {}
2.for 關(guān)系表達式或邏輯表達式 { }
n := 10for n > 0 { n--}
3.for { }
for { fmt.Println("hello world") } // 等價于 // for true { // fmt.Println("hello world") // }
range語句
Golang range類似迭代器操作,可以對 slice、map、數(shù)組、字符串等進行迭代循環(huán)。在字符串、數(shù)組和切片中它返回 (索引, 值) ,在集合中返回 (鍵, 值),但若當只有一個返回值時,第一個參數(shù)是索引或鍵。
str := "abc" for i, char := range str { fmt.Printf("%d => %sn", i, string(char)) } for i := range str { //只有一個返回值 fmt.Printf("%dn", i) } nums := []int{1, 2, 3} for i, num := range nums { fmt.Printf("%d => %dn", i, num) } kvs := map[string]string{"a": "apple", "b": "banana"} for k, v := range kvs { fmt.Printf("%s => %sn", k, v) } for k := range kvs { //只有一個返回值 fmt.Printf("%sn", k) } // 輸出結(jié)果 // 0 => a // 1 => b // 2 => c // 0 // 1 // 2 // 0 => 1 // 1 => 2 // 2 => 3 // a => apple // b => banana // a // b
for循環(huán)尤其是range語句,在平時開發(fā)過程中頻繁使用,但很多開發(fā)者(本人算一個)經(jīng)常會在以下場景中踩坑。
場景一,使用循環(huán)迭代器的變量
先來看一個明顯的錯誤:
func main() { var out []*int for i := 0; i < 3; i++ { // i := i out = append(out, &i) } fmt.Println("值:", *out[0], *out[1], *out[2]) fmt.Println("地址:", out[0], out[1], out[2]) } // 輸出結(jié)果 // 值: 3 3 3 // 地址: 0xc000012090 0xc000012090 0xc000012090
分析
out
是一個整型指針數(shù)組變量,在for循環(huán)中,聲明了一個i
變量,每次循環(huán)將i
的地址追加到out
切片中,但是每次追加的其實都是i
變量,因此我們追加的是一個相同的地址,而該地址最終的值是3。
正確做法
解開代碼中的注釋// i := i
,每次循環(huán)時都重新創(chuàng)建一個新的i
變量。
再看一個比較隱秘的錯誤:
func main() { a1 := []int{1, 2, 3} a2 := make([]*int, len(a1)) for i, v := range a1 { a2[i] = &v } fmt.Println("值:", *a2[0], *a2[1], *a2[2]) fmt.Println("地址:", a2[0], a2[1], a2[2]) } // 輸出結(jié)果 // 值: 3 3 3 // 地址: 0xc000012090 0xc000012090 0xc000012090
分析
大多數(shù)人就是在range
這里給變量賦值的時候踩坑,因為比較隱秘,其實情況和上面的一樣,range
在遍歷值類型時,其中的v
是一個局部變量,只會聲明初始化一次,之后每次循環(huán)時重新賦值覆蓋前面的,所以給a2[i]
賦值的時候其實都是同一個地址&v
,而v
最終的值為a1
最后一個元素的值,也就是3。
正確做法
①a2[i]
賦值時傳遞原始指針,即a2[i] = &a1[i]
②創(chuàng)建臨時變量t := v
;a2[i] = &t
③閉包(與②原理一樣),func(v int) { a2[i] = &v }(v)
更為隱秘的還有:
func main() { var out [][]int for _, i := range [][1]int{{1}, {2}, {3}} { out = append(out, i[:]) } fmt.Println("Values:", out)}// 輸出結(jié)果// [[3] [3] [3]]
原理也是一樣的,不論遍歷多少次,i[:]
總是被本次遍歷的值所覆蓋
場景二,在循環(huán)體內(nèi)使用goroutines
func main() { values := []int{1, 2, 3} wg := sync.WaitGroup{} for _, val := range values { wg.Add(1) go func() { fmt.Println(val) wg.Done() }() } wg.Wait()}// 輸出結(jié)果// 3// 3// 3
分析
對于主協(xié)程來講,循環(huán)是很快就跑完的,而這個時候各個協(xié)程可能才開始跑,此時val
的值已經(jīng)遍歷到最后一個了,所以各協(xié)程都輸出了3
。(如果遍歷數(shù)據(jù)龐大,主協(xié)程遍歷耗時較久的話,goroutine的輸出會根據(jù)當時候的val
的值,所以每次的輸出結(jié)果不一定相同的。)
解決辦法
①使用臨時變量
for _, val := range values { wg.Add(1) val := val go func() { fmt.Println(val) wg.Done() }()}
②使用閉包
for _, val := range values { wg.Add(1) go func(val int) { fmt.Println(val) wg.Done() }(val)}