解决Golang 中使用WaitGroup的那点坑

作者:he_silong 时间:2024-04-28 09:11:08 

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 main
import (
   "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 main
import (
   "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 main
import (
"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 counter
goroutine 1 [running]:
sync.(*WaitGroup).Add(0xc042008230, 0xffffffffffffff9c)
   D:/Go/src/sync/waitgroup.go:75 +0x1d0
main.main()
   D:/code/go/src/test-src/2-Package/sync/waitgroup/main.go:10 +0x54

以上为个人经验,希望能给大家一个参考,也希望大家多多支持asp之家。如有错误或未考虑完全的地方,望不吝赐教。

来源:https://blog.csdn.net/he_silong/article/details/82459868

标签:Golang,WaitGroup
0
投稿

猜你喜欢

  • PyCharm 2020.2.2 x64 下载并安装的详细教程

    2023-11-29 13:36:32
  • 一个轻量级的XHTML右键菜单[支持IE和firefox]

    2024-04-19 10:58:32
  • Python中变量交换的例子

    2021-04-02 13:28:36
  • 解决python 文本过滤和清理问题

    2023-08-31 08:18:38
  • python实现简单购物商城

    2022-01-23 06:47:39
  • Python3 批量扫描端口的例子

    2021-12-20 05:17:37
  • Python中运行并行任务技巧

    2022-12-30 01:13:33
  • 微信公众平台实现获取用户OpenID的方法

    2023-11-17 05:55:56
  • 页面重构应注意的repaint和reflow

    2011-03-31 17:11:00
  • vue项目打包后怎样优雅的解决跨域

    2024-04-29 13:11:03
  • Python中 传递值 和 传递引用 的区别解析

    2023-12-26 07:05:22
  • python数据操作之lambda表达式详情

    2022-08-19 21:21:32
  • python pycharm的安装及其使用

    2023-11-04 20:27:06
  • keras 解决加载lstm+crf模型出错的问题

    2022-06-11 01:27:27
  • 详解python读写json文件

    2022-11-01 16:18:53
  • 深入理解go slice结构

    2024-04-26 17:27:07
  • 使用Dreamweaver便捷技巧方法十六则

    2010-07-02 16:28:00
  • pythotn条件分支与循环详解

    2021-12-02 12:20:08
  • Python使用Pandas库常见操作详解

    2022-12-10 09:03:13
  • 如何利用Python打开txt格式的文件

    2022-06-01 02:08:36
  • asp之家 网络编程 m.aspxhome.com