golang中函数闭包遇到的坑

golang项目中用到kafka，有个组件需要拉取消息；这个组件启动是成功的，有的时候能拉取消息；有的时候感觉像卡死了一样，有消息了，一条都没有拉取到。kafka测试没问题；最后定位到，原来是函数闭包用的有问题。

看下面这段代码，看起来似乎是两个goroutine，我们把rId作为goroutine的标记，结果想象应该是：

1 abcdef
0 123456
0 123456
1 abcdef
1 abcdef
......

package main

import (
"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 abcdef
0 abcdef
0 abcdef
1 abcdef
1 abcdef
0 abcdef
0 abcdef
1 abcdef
1 abcdef
0 abcdef
0 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)
}