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)
}
有疑问加站长微信联系(非本文作者)