首先初始化WaitGroup,再通过Add增加计数,每个goroutine执行完调用Done,主线程调用Wait阻塞直至所有任务完成。
在Go语言中,
sync.WaitGroup
是一种常用的协程
同步机制
,用于等待一组并发的goroutine执行完成。它特别适用于主线程需要等待多个子任务结束的场景,比如批量发起网络请求、并行处理数据等。使用 WaitGroup 可以避免程序提前退出,确保所有协程都执行完毕。
WaitGroup 核心方法
WaitGroup 提供了三个主要方法:
Add(delta int)
:增加计数器的值,通常用于添加要等待的goroutine数量。
Done()
:将计数器减1,一般在goroutine结束时调用,相当于 Add(-1)。
Wait()
:阻塞当前协程,直到计数器变为0。
基本使用示例
下面是一个简单的例子,展示如何使用 WaitGroup 等待多个goroutine完成:
输出结果:
立即学习
“
go语言免费学习笔记(深入)
”;
go语言参考手册 中文CHM版
Go 是一个开源的编程语言,它能让构造简单、可靠且高效的软件变得容易。本文给大家带来Go参考手册,需要的可以来下载! Go是从2007年末由Robert Griesemer, Rob Pike, Ken Thompson主持开发,后来还加入了Ian Lance Taylor, Russ Cox等人,并最终于2009年11月开源,在2012年早些时候发布了Go 1稳定版本。现在Go的开发已经是完全开放的,并且拥有一个活跃的社区。 Go 语言特色 简洁、快速、安全 并行、有趣、开源 内存管理、v数组安全、编译
下载
Worker 1 startingWorker 2 startingWorker 3 startingWorker 1 doneWorker 2 doneWorker 3 doneAll workers finished
使用注意事项
虽然 WaitGroup 使用简单,但有几个关键点需要注意,避免常见错误:
避免Add负数或超出范围
:在调用 Add 时传入负数会引发 panic,必须确保 Add 的值合理。
确保 Done 被调用
:每个 Add 对应一次 Done,否则 Wait 会一直阻塞,导致死锁。
传递指针
:WaitGroup 应以指针形式传递给goroutine,否则值拷贝会导致各个协程操作的是不同的实例。
Add 应在 go 之前调用
:如果在goroutine内部才 Add,可能主线程的 Wait 已经开始,导致遗漏。
实际应用场景
WaitGroup 常用于并行任务处理。例如,从多个API获取数据:
这段代码并发请求多个URL,并等待全部完成后再继续。
基本上就这些。WaitGroup 是Go中最基础也最实用的同步工具之一,掌握它的正确用法对编写可靠的并发程序至关重要。不复杂但容易忽略细节。
package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done() // 任务完成,计数器减1
fmt.Printf("Worker %d starting\n", id)
time.Sleep(time.Second)
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 3; i++ {
wg.Add(1) // 每启动一个goroutine,计数器加1
go worker(i, &wg)
}
wg.Wait() // 等待所有goroutine完成
fmt.Println("All workers finished")
}
func fetchAllData() {
var wg sync.WaitGroup
urls := []string{"http://a.com", "http://b.com", "http://c.com"}
for _, url := range urls {
wg.Add(1)
go func(u string) {
defer wg.Done()
resp, err := http.Get(u)
if err != nil {
fmt.Printf("Error: %s\n", err)
return
}
defer resp.Body.Close()
fmt.Printf("Fetched %s\n", u)
}(url)
}
wg.Wait()
fmt.Println("All requests completed")
}
