巧妙运用Go语言sync.WaitGroup避免并发陷阱
Go语言的并发特性让高效编程成为可能，而sync.WaitGroup是实现高效并发同步的关键工具。它能够等待一组goroutine完成任务后，再执行后续操作。本文将深入讲解sync.WaitGroup的用法，并分析一个常见的错误示例。
示例代码中，Wg()函数启动了三个goroutine，每个goroutine都执行wgCore()函数，模拟一个需要等待完成的任务。初始代码中，wg.Add(1)位于wgCore()函数内部，这意味着每个goroutine只增加计数器1，但wg.Wait()并不知道需要等待多少goroutine完成。sync.WaitGroup的Wait()方法会阻塞直到计数器归零，由于计数未预先设置，Wait()可能在goroutine未完成时提前返回，导致程序结果错误。
改进后的代码将wg.Add(3)放在Wg()函数中，显式地将计数器设置为3，表示需要等待三个goroutine。这样，wg.Wait()就能准确等待所有goroutine完成再继续执行。 关键在于，sync.WaitGroup的计数器必须在启动goroutine之前预先设置，它不会自动跟踪goroutine数量。wg.Add()必须在启动goroutine前调用，告知WaitGroup需要等待的goroutine数量；defer wg.Done()则确保每个goroutine结束后计数器减1，最终使计数器归零，解除wg.Wait()的阻塞。 只有预先设置正确的计数，才能确保sync.WaitGroup的正确使用，避免意外结果。
立即学习“go语言免费学习笔记（深入）”；
以上就是Go语言sync.WaitGroup如何正确使用才能避免并发编程中的陷阱？的详细内容，更多请关注php中文网其它相关文章！
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号
977
收藏
