文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

如何解决Golang中使用WaitGroup的问题

2023-06-14 20:37

关注

这篇文章主要介绍了如何解决Golang中使用WaitGroup的问题,具有一定借鉴价值,感兴趣的朋友可以参考下,希望大家阅读完这篇文章之后大有收获,下面让小编带着大家一起了解一下。

golang的优点

golang是一种编译语言,可以将代码编译为机器代码,编译后的二进制文件可以直接部署到目标机器而无需额外的依赖,所以golang的性能优于其他的解释性语言,且可以在golang中使用goroutine来实现并发性,它提供了一个非常优雅的goroutine调度程序系统,可以很容易地生成数百万个goroutine。

sync.WaitGroup对于Golang开发者来说并不陌生,其经常作为多协程之间同步的一种机制。用好它势必会让你事半功倍,但是一旦错用将引发问题。

关于WaitGroup的使用网上有很多例子,在此就不做介绍了,我想说的是我在项目中使用WaitGroup遇到的坑。

在项目中,因为服务器有同步需求, 所以直接使用了WaitGroup,但是未考虑使用场景,结果在项目上线之后,高峰期的时候客户端经常出现卡顿,经过多方查找,才发现如果使用WaitGroup的时候,未启动单独的goroutine,那么极有可能造成主线程的阻塞

所以我做了下面的测试(测试中,我把WaitGroup置于协程内):

import ( "fmt" "sync" "time") func main() {    fmt.Println("main-1") testW() fmt.Println("main-2") time.Sleep(time.Duration(15) * time.Second) } func testW() { fmt.Println("testW-1") go func() {  var wg sync.WaitGroup  fmt.Println("testW-2")  testW1(&wg)  fmt.Println("testW-5")  wg.Wait()  fmt.Println("testW-6") }()} func testW1(wg *sync.WaitGroup) { wg.Add(1) fmt.Println("testW-3") time.AfterFunc(time.Second*5, func() {  wg.Done() }) fmt.Println("testW-4") }

输出为:

main-1

testchan-1

main-2

testchan-2

testchan-3

testchan-4

testchan-5

// 过5秒

testchan-6

总结:

将WaitGroup用于goroutine内,不会导致主线程的阻塞,同样可以实现同步的效果。

补充:WaitGroup的简单用法(等待组)

你品一下人家这名字,等待组。等待什么,等待goroutine完成啊。有些时候,我们启动多个goroutine去执行任务,我举个例子

listip := []string{"10.0.9.11","10.0.9.22","10.0.9.33"}for _, ip := range(listip) {    //假设我们执行一个ping ip 的逻辑    go PingIPWork(ip)}

我这里执行了一个多ip去ping的逻辑,一般这种时候,你要是执行一波,人家肯定毛都不会返回给你,为什么?因为人家主线程直接就退出了,还是那句话,你又没告诉人家主线程要等这ip全部都ping 完,所以你必须要加个等待,等着Goroutine完成,这里我再举一个网上的例子

package mainimport (    "fmt")func main() {    go func() {        fmt.Println("Goroutine 1")    }()    go func() {        fmt.Println("Goroutine 2")    }()    //来个睡眠,等Goroutine结束    time.Sleep(time.Second * 1)}

看到了么,加了一个sleep,用sleep去等着Goroutine跑完,上面我举的那个例子也可以这么来

listip := []string{"10.0.9.11","10.0.9.22","10.0.9.33"}for _, ip := range(listip) {    //假设我们执行一个ping ip 的逻辑    go PingIPWork(ip)}time.Sleep(time.Second * 1)

加个sleep可以等待完成,但是万一啊,Goroutine有的跑的快,有的慢,你那sleep就一秒,要是有的Goroutine没跑完不就白瞎了吗,所以咱们需要一个机制,这个机制可以帮助咱们去管理Goroutine,让我们知道Goroutine这东西什么时候停,什么时候完成。

所以,WaitGroup这个东西,就可以帮助我们解决这个问题,还是老样子,我举一个简单的例子来说明我的想法。

package mainimport (    "fmt" "sync")func PingIPWork(ip string) { fmt.Println(ip)}func main() {    //定义一个等待阿祖 var wg sync.WaitGroup wg.Add(3) // 因为有3个Ip,咱们定义三个动作,所以来三个计数 listip := []string{"10.0.9.11","10.0.9.22","10.0.9.33"} for _, ip := range(listip) {  //假设我们执行一个ping ip 的逻辑  go func(ip string) {            //执行一个work            PingIPWork(ip)            //操作完成之后,done一个计数,也就是3-1   wg.Done()  }(ip) }    //等待 wg.Wait() // 等待,直到计数为0}

这里我举了一个简单的例子,其实wg的用法较为简单,在这个例子里面我们用到了

wg.wait等待Goroutine结束之后退出主进程wg.Add添加Goroutine,其实你可以把它想成,可添加的最大Goroutine数wg.Done想象成销毁参数,当Goroutine结束之后调用,意思就是,你没了,我减1

WaitGroup的其他注意事项

将Wg作为参数进行传递的时候,需要使用指针

有些时候,咱们不想写的这么麻烦,就寻思怎么才能简单一点,或者可变性稍微强一点,有些时候我们要把wg最为参数,在函数内部调用,我们该怎么写呢?

package mainimport ( "fmt" "sync")func PingIPWork(ip string, wg *sync.WaitGroup) { fmt.Println(ip) wg.Done()}func main() { var wg sync.WaitGroup wg.Add(3) // 因为有两个动作,所以增加2个计数 listip := []string{"10.0.9.11","10.0.9.22","10.0.9.33"} for _, ip := range(listip) {  //假设我们执行一个ping ip 的逻辑  go PingIPWork(ip, &wg)  } wg.Wait() // 等待,直到计数为0}

看到了么,如果你把Wg作为参数进行传递,你得要用指针的形式传值,否则就会死锁!!!!!!!!

Wg.Add的数值不能为负

wg.Add()的数值必须为正数,如果为负数,将会抛出异常。

panic: sync: negative WaitGroup countergoroutine 1 [running]:sync.(*WaitGroup).Add(0xc042008230, 0xffffffffffffff9c)    D:/Go/src/sync/waitgroup.go:75 +0x1d0main.main()    D:/code/go/src/test-src/2-Package/sync/waitgroup/main.go:10 +0x54

感谢你能够认真阅读完这篇文章,希望小编分享的“如何解决Golang中使用WaitGroup的问题”这篇文章对大家有帮助,同时也希望大家多多支持编程网,关注编程网行业资讯频道,更多相关知识等着你来学习!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     801人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     348人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     311人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     432人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     220人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯