首頁 > 後端開發 > Golang > 記錄Go的循環遍歷使用小坑

記錄Go的循環遍歷使用小坑

藏色散人
發布: 2021-02-19 17:38:23
轉載
2371 人瀏覽過
##上下上排符號為

golang#1欄位分享給大家的循環遍歷使用小坑 ,希望對需要的朋友有所幫助!

記錄Go的循環遍歷使用小坑

在Golang的流程控制中,迴圈語句有for和range兩種。

for語句

1.for 賦值表達式; 關係式表達式或邏輯表達式; 賦值表達式{ }

for i := 0; i < 10; i++ {}
登入後複製

2.for 關係式或邏輯運算式{ }

n := 10for n > 0 {
 n--}
登入後複製

#3.for { }

for {
 fmt.Println("hello world")
}
// 等价于
// for true {
//     fmt.Println("hello world")
// }
登入後複製

range語句

Golang range類似迭代器操作,可以對slice、map、陣列、字串等進行迭代循環。在字串、陣列和切片中它會傳回 (索引, 值) ,在集合中傳回 (鍵, 值),但若當只有一個傳回值時,第一個參數是索引或鍵。

str := "abc"
for i, char := range str {
    fmt.Printf("%d => %s\n", i, string(char))
}
for i := range str { //只有一个返回值
    fmt.Printf("%d\n", i)
}
nums := []int{1, 2, 3}
for i, num := range nums {
    fmt.Printf("%d => %d\n", i, num)
}
kvs := map[string]string{"a": "apple", "b": "banana"}
for k, v := range kvs {
    fmt.Printf("%s => %s\n", k, v)
}
for k := range kvs { //只有一个返回值
    fmt.Printf("%s\n", k)
}
// 输出结果
// 0 => a
// 1 => b
// 2 => c
// 0
// 1
// 2
// 0 => 1
// 1 => 2
// 2 => 3
// a => apple
// b => banana
// a
// b
登入後複製

for循環尤其是range語句,在平時開發過程中頻繁使用,但很多開發者(本人算一個)經常會在以下場景中踩坑。

場景一,使用循環迭代器的變數

先來看一個明顯的錯誤:

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])
}
// 输出结果
// 值: 3 3 3
// 地址: 0xc000012090 0xc000012090 0xc000012090
登入後複製

分析

out是一個整數指標數組變量,在for循環中,聲明了一個i變量,每次循環將i的位址追加到out切片中,但每次追加的其實都是i變量,因此我們追加的是一個相同的位址,而該位址最終的值是3。

正確做法

解開程式碼中的註解

// i := i,每次迴圈時都重新建立一個新的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])
}
// 输出结果
// 值: 3 3 3
// 地址: 0xc000012090 0xc000012090 0xc000012090
登入後複製

分析

大多數人就是在

range這裡給變數賦值的時候踩坑,因為比較隱秘,其實情況和上面的一樣,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)}
登入後複製

以上是記錄Go的循環遍歷使用小坑的詳細內容。更多資訊請關注PHP中文網其他相關文章!

相關標籤:
來源:learnku.com
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板