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)
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值