go随聊-sync.WaitGroup

最新推荐文章于 2025-07-07 08:34:02 发布

九品神元师

最新推荐文章于 2025-07-07 08:34:02 发布

阅读量468

点赞数

CC 4.0 BY-SA版权

分类专栏： Golang优雅中带着严谨文章标签： golang 并发

本文链接：https://blog.youkuaiyun.com/yimin_tank/article/details/83575344

Golang优雅中带着严谨专栏收录该内容

40 篇文章

订阅专栏

本文深入探讨了Go语言中sync.WaitGroup的使用方法，包括其三个关键函数：Add、Done和Wait的功能与应用场景。通过一个具体示例，演示了如何利用sync.WaitGroup来同步多个协程的执行，确保所有任务完成后再进行后续操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

GO提供了sync包和channel来解决协程同步和通讯。新手对channel通道操作起来更容易产生死锁，如果时缓冲的channel还要考虑channel放入和取出数据的速率问题。sync.WaitGroup是等待一组协程结束，它实现了一个类似任务队列的结构，你可以向队列中加入任务，任务完成后就把任务从队列中移除，如果队列中的任务没有全部完成，队列就会触发阻塞以阻止程序继续运行。本文主要介绍sync.WaitGroup。

package sync

type WaitGroup struct 三个重要函数：

//调用Add来设置等待的协程(goroutine)数量
func (wg *WaitGroup) Add(delta int)

//协程(goroutine)完成后调用Done，内部其实就是将值-1
func (wg *WaitGroup) Done() 

//Wait用来阻塞即等待wg值为0，直到所有协程(goroutine)完成才会向下执行
func (wg *WaitGroup) Wait()

使用例子

定义任务及WaitGroup

var wg sync.WaitGroup
//任务每个1秒打印一次，打印5次
func process(name string) {
    for i:=0;i<5;i++ {
        fmt.Println("process[",name,"] count[",i+1,"]")
        time.Sleep(time.Second)
    }
    //协程完成后调用Done即计数器-1
    wg.Done()
}

启动任务

fmt.Println("start")
wg.Add(1)
//启动任务worker1
go process("worker1")
wg.Add(1)
//启动任务worker2
go process("worker2")
//等待所有任务完成
wg.Wait()
fmt.Println("exit")
----------------------------------
start
process[ worker2 ] count[ 1 ]
process[ worker1 ] count[ 1 ]
process[ worker1 ] count[ 2 ]
process[ worker2 ] count[ 2 ]
process[ worker2 ] count[ 3 ]
process[ worker1 ] count[ 3 ]
process[ worker1 ] count[ 4 ]
process[ worker2 ] count[ 4 ]
process[ worker2 ] count[ 5 ]
process[ worker1 ] count[ 5 ]
exit