巧妙运用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
的正确使用,避免意外结果。