golang中函數閉包遇到的坑

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

golang項目中用到kafka，有個組件需要拉取訊息；這個組件啟動是成功的，有的時候能拉取訊息；有的時候感覺像卡死了一樣，有訊息了，一條都沒有拉取到。kafka測試沒問題；最後定位到，原來是函數閉包用的有問題。

看下面這段代碼，看起來似乎是兩個goroutine，我們把rId作為goroutine的標記，結果想象應該是：

1 abcdef0 1234560 1234561 abcdef1 abcdef......

package mainimport (    "fmt"    "sync"    "time")func main(){    strSlice := make([]*string, 0)    s1 := "123456"    strSlice = append(strSlice, &s1)    s2 := "abcdef"    strSlice = append(strSlice, &s2)    var wg sync.WaitGroup    for i,s := range strSlice{        wg.Add(1)        go func(rId int){            for{                fmt.Println(rId, *s)                time.Sleep(1 * time.Second)            }            wg.Done()        }(i)    }    wg.Wait()}

而實際測試的結果是：

1 abcdef0 abcdef0 abcdef1 abcdef1 abcdef0 abcdef0 abcdef1 abcdef1 abcdef0 abcdef0 abcdef......

s變數在迴圈的過程中已經被修改了，因此結果變成了這樣。我們的項目中去擷取kafka partition然後每個partition建立一個goroutine去消費的時候，就這樣只會消費最後一個partition的資料，因此出現問題。

可以和rId一樣過將變數作為匿名函數的參數傳入進去，如下：

    for i,s := range strSlice{        wg.Add(1)        go func(rId int, rS *string){            for{                fmt.Println(rId, *rS)                time.Sleep(1 * time.Second)            }            wg.Done()        }(i, s)    }