跳转到主内容
websoft网络软件专家 - 深耕网络技术,打造实用软件!

Golang协程同步方法 sync.WaitGroup实践

首先初始化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完成:
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") }
输出结果: 立即学习 “ 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获取数据:
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") }
这段代码并发请求多个URL,并等待全部完成后再继续。 基本上就这些。WaitGroup 是Go中最基础也最实用的同步工具之一,掌握它的正确用法对编写可靠的并发程序至关重要。不复杂但容易忽略细节。

相关文章