golang 迴圈變數
阿新 • • 發佈:2022-04-09
下面例子:
package main import "fmt" func test1() { 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]) } func main() { test1() }
輸出:
值: 3 3 3 地址: 0xc000016098 0xc000016098 0xc000016098 原因:,range
v
是一個區域性變數,只會宣告初始化一次,之後每次迴圈時重新賦值覆蓋前面的,所以給a2[i]
賦值的時候其實都是同一個地址&v
,而v
最終的值為a1
最後一個元素的值,也就是3。
正確做法:
①a2[i]
賦值時傳遞原始指標,即a2[i] = &a1[i]
②建立臨時變數
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)}// 輸出結果// [[3] [3] [3]]
原理也是一樣的,不論遍歷多少次,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()}// 輸出結果// 3// 3// 3
分析
對於主協程來講,迴圈是很快就跑完的,而這個時候各個協程可能才開始跑,此時val的值已經遍歷到最後一個了,所以各協程都輸出了3。(如果遍歷資料龐大,主協程遍歷耗時較久的話,goroutine的輸出會根據當時候的val的值,所以每次的輸出結果不一定相同的。)
解決辦法
①使用臨時變數 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)}
上面的錯誤程式碼vscode外掛會提示:loop variable xx captured by func literal loopclosure
協程引用迴圈變數的坑
迴圈體中啟動協程非同步執行,這個時候就容易出現問題了,比如下面這樣一段程式碼就會出現我們不期望的結果。
for i := 0; i < 10; i++ {
go func() {
fmt.Println(i)
} ()
}
我們期望他能亂序輸出0到9這幾個數,但是他的執行結果並非如此。實際的執行結果如下:
7
10
10
10
10
10
10
10
10
7
可以看到他的執行結果大家基本都輸出10。其實原因也很容易解釋:
主協程的迴圈很快就跑完了,而各個協程才開始跑,此時i的值已經是10了,所以各協程都輸出了10。(輸出7的兩個協程,在開始輸出的時候主協程的i值剛好是7,這個結果每次執行輸出都不一樣)
參考:https://www.cnblogs.com/wscsq789/p/15388804.html