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)
    }

你可能感兴趣的:(golang)