在golang中使用Sync.WaitGroup解决等待的问题
面对goroutine我们都需要等待它完成交给它的事情，等待它计算完成或是执行完毕，所以不得不在程序需要等待的地方使用time.Sleep()来睡眠一段时间，等待其他goroytine执行完毕，下面的代码打印1到100的for循环可以在很快的时间内运行完毕，但是我们必须添加time.Sleep()来等待其打印完毕,如果我们不等待仿佛什么也没有发生一样。。。。。这肯定不是我们想要的！
func main(){
for i := 0; i < 100 ; i++{
go fmt.Println(i)
}
time.Sleep(time.Second)
}
这就是我最开始学习goroutine想到的笨方法。。。。死等！ 等不到 time.Second*2 在等不到*3
但是对于我们实际的使用场景来说，1秒也许等待的时间有点长，而有时候是远远不够的，最主要的我们无法预知gorotine的运行时间长短，所以也无法指定Sleep睡眠的时间。
正好前几天学习了channel，那么现在我们就可以
使用channel来达到我们的目的了
func main() {
c := make(chan bool, 100)
for i := 0; i < 100; i++ {
go func(i int) {
fmt.Println(i)
c <- true
}(i)
}
for i := 0; i < 100; i++ {
<-c
}
}
完美输出！这里我们可以从控制台看到我们的胜利成果，但是我们有更好的处理方式：sync.WaitGroup
WaitGroup对象内部有一个计数器，最初从0开始，它有三个方法：Add() Done() Wait() 用来控制计数器的数量从而达到等待的目的。
func main() {
wg := sync.WaitGroup{}
wg.Add(100)
for i := 0; i < 100; i++ {
go func(i int) {
defer wg.Done()
fmt.Print(i)
}(i)
}
wg.Wait()
}
首先使用 add(100) 把计数器的数量设置为100，每次循环运行完毕都把计数器减1，主函数使用Wait(）
阻塞，直到计数器归0结束主线程结束阻塞。
这里会阻塞主线程！！！需要注意。
另外：以上三种方式打印出的数字皆无序
type WaitGroup struct {
noCopy noCopy
state1 [3]uint32
}
它只是一个struct所以当参数传递的时候需要传递指针。
补充：sync.WaitGroup的错误用法与纠正
本文主要是记录在学习golang语言过程中错误使用sync.WaitGroup以及指针引起的报错，并且纠正过程。
1.错误使用场景
main包的实现内容：
package main
import (
"fmt"
"pressure/game"
"sync"
)
var waitGroup sync.WaitGroup //定义一个同步等待的组
func main(){
waitGroup.Add(1) //添加一个计数
go game.ConnSocket(serverAddr, waitGroup) //调用其他包的方法执行任务
waitGroup.Wait() //阻塞直到所有任务完成
fmt.Println("main DONE!!!")
}
如下是game包的内容：
package game
import (
"fmt"
"net"
"strconv"
"strings"
"sync"
)
var gameWait sync.WaitGroup
func ConnSocket(serverAddr string, wait sync.WaitGroup) {
var err error
Conn, err = net.Dial("tcp", serverAddr)
if err != nil {
fmt.Println("Error connected:", err.Error())
wait.Done()
return
}
gameWait = wait
fmt.Println("connected OK:", Conn.RemoteAddr())
go readMessage()
}
func readMessage() {
fmt.Println("readMessage:")
for {
if OK:=dosomething(),OK{
gameWait.Done()
}
}
}
然后执行如上的代码，会出现报错：
panic: sync: negative WaitGroup counter
goroutine 12 [running]:
sync.(*WaitGroup).Add(0x7b8b90, 0xffffffffffffffff)
D:/Program Files/Go/src/sync/waitgroup.go:75 +0x1d0
sync.(*WaitGroup).Done(0x7b8b90)
原因：
func ConnSocket(serverAddr string, wait sync.WaitGroup)只是把sync.WaitGroup的值传递过来，而非内存地址，所以在赋值给gameWait的时候也只是赋值了值，gameWait并非指向的main里的waitGroup 的内存地址，导致 gameWait.Done()的时候并没有获取到main里的计数1，所以才会报负数。究其原因还是没有对指针的理解。所以针对上面的问题做了修改。
2.正确的使用
main包的实现内容：
package main
import (
"fmt"
"pressure/game"
"sync"
)
var waitGroup sync.WaitGroup //定义一个同步等待的组
func main(){
waitGroup.Add(1) //添加一个计数
//这里传递waitGroup的内存地址
go game.ConnSocket(serverAddr, &waitGroup) //调用其他包的方法执行任务
waitGroup.Wait() //阻塞直到所有任务完成
fmt.Println("main DONE!!!")
}
如下是game包的内容：
package game
import (
"fmt"
"net"
"strconv"
"strings"
"sync"
)
var gameWait *sync.WaitGroup//此处也申明为指针变量
//参数传递为指针传递*sync.WaitGroup，即wait 指向的是main里的waitGroup的内存地址
func ConnSocket(serverAddr string, wait *sync.WaitGroup) {
var err error
Conn, err = net.Dial("tcp", serverAddr)
if err != nil {
fmt.Println("Error connected:", err.Error())
wait.Done()
return
}
gameWait = wait//指针指向另一个指针，这样就做到了gameWait指向的是wait的地址，所以gameWait修改wait的内存地址锁存贮的值也会跟随改变
fmt.Println("connected OK:", Conn.RemoteAddr())
go readMessage()
}
func readMessage() {
fmt.Println("readMessage:")
for {
if OK:=dosomething(),OK{
gameWait.Done()
}
}
}
3.结果
还是对指针传递以及值传递理解不够透彻才会出现如此问题，因此还需要多针对指针进行学习与理解。以防在以后的工作中出现类似问题。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。
相关文章
这篇文章主要为大家介绍了k8s容器互联flannel vxlan通信原理详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-04-04
这篇文章主要为大家介绍了执行go vendor时，第三方包go版本冲突问题的解决方法，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-07-07
这篇文章主要为大家介绍了Golang对struct字段重新排序优化数据结构性能实践,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-12-12
本文主要介绍了Go语言遍历结构体、切片和字典的方法，对大家的学习具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-06-06
这篇文章主要为大家介绍了Go语句与表达式案例手册深度解析,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-09-09
Golang使用pprof和trace进行诊断和修复性能问题
在 Go 中,开发人员可以使用强大的内置工具来帮助诊断和修复性能问题,其中两个工具是 pprof 和 trace 包,下面就跟随小编一起来了解下如何使用pprof和trace进行诊断和修复性能问题吧2024-01-01
本文主要介绍了Go语言单向通道的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-04-04
这篇文章主要为大家介绍了go 原子操作的方式及实现原理深入解析，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-04-04
目标检测是计算机视觉领域的重要任务,它不仅可以识别图像中的物体,还可以标记出物体的位置和边界框,YOLO是一种先进的目标检测算法,以其高精度和实时性而闻名,本文将介绍如何使用Golang实现YOLO目标检测算法,文中有相关的代码示例供大家参考,需要的朋友可以参考下2023-11-11
new 只分配内存，make 用于初始化 slice、map 和 channel，本文详细的介绍了go中new和make的区别小结，感兴趣的可以了解一下2023-05-05
最新评论