解决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
猜你喜欢
- 我就废话不多说了,大家还是直接看代码吧~lt=client.fangjia.district_stat_all_0416dl = dt.fi
- 南京的炎炎夏日也未能挡住书友会朋友们的参与,下午两点半我们正式开始了这个月的话题:“浏览器”。一开始大家大致上说了一些自己认为各浏览器中各自
- 一般事件事件浏览器支持描述onClickHTML: 2 | 3 | 3.2 |
- 前言:在日常开发的过程中,有时候会临时用到一些简单的函数,她们的业务逻辑会相对简单,简单到并不值得让他们留下名字,这个时候,就很值得使用匿名
- 上一篇博客selenium+python自动化测试(二)–使用webdriver操作浏览器讲解了使用webdriver操作浏览器的各种方法,
- 计算机之所以能做很多自动化的任务,因为它可以自己做条件判断。比如,输入用户的年龄,根据年龄打印不同的内容。。。Python程序中,能让计算机
- 没什么实际用途,纯属消遣Quick Click<html><head><title>Quick_Clic
- python2和python3对于字符串的处理有很大的区别熟悉了python2的写法用python3时真的会遇到很多问题啊……区别pytho
- 从事API相关的工作很有挑战性,在高峰期保持系统的稳定及健壮性就是其中之一,这也是我们在Mailgun做很多压力测试的原因。这么久以来,我们
- 使用Python 2.7 + pywin32 + wxpython开发每隔一段时间检测一下服务是否停止,如果停止尝试启动服务。进行服务停止日
- 本文较为详细的分析了了Python的对象体系。分享给大家供大家参考。具体如下:Guido用C语言创造了Python,在Python的世界中一
- #!/usr/bin/env python3# -*- coding: utf-8 -*-import globfrom os import
- 常用分支首先分享一下我们的分支规范,然后再介绍摸索出的打tag的规范。mastermaster : 主分支 , 最终在master分支对外发
- 在Dreamweaver 4.0中,我们就已接触了模板与库的概念,知道它们是批量生成风格类似的网页的好工具。如今在Dreamweaver M
- 本文记录了Windows10安装Python2.7的详细步骤,分享给大家。一、下载软件Python的官方地址点击Downloads找到进行下
- 关于asp缓存函数,类什么的,在网上可以说笔笔皆是,为啥我要不辞辛苦去写一个呢?大概看了下,各有各的优点吧,可是大部分好像不可以缓存数据额,
- 语法: ROW_NUMBER() OVER([ <partition_by_clause>] <order_by_clau
- 一、旧版本(1.0以下)的卷积函数:tf.nn.conv2d在tf1.0中,对卷积层重新进行了封装,比原来版本的卷积层有了很大的简化。con
- 简介在逛github时发现一个好玩的Go项目,彩色输出文本说明支持Linux彩色输出支持Windows彩色输出Golang IDE输出是不支
- 同级目录(兄弟目录)调用看书看得好好的,一写代码就出错!!!这个问题是大家初学Python的时候会遇到的一个很常见的问题,然后我们去搜网上的