golang中函数闭包遇到的坑

qq_33204116 · · 1358 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

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

有疑问加站长微信联系(非本文作者)

本文来自:CSDN博客

感谢作者:qq_33204116

查看原文:golang中函数闭包遇到的坑

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

1358 次点击  
加入收藏 微博
下一篇:GO面向对象
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传