首页 > 文章列表 > Go语言sync.WaitGroup如何正确使用才能避免并发编程中的陷阱?

Go语言sync.WaitGroup如何正确使用才能避免并发编程中的陷阱?

156 2025-03-13

Go语言sync.WaitGroup如何正确使用才能避免并发编程中的陷阱?

巧妙运用Go语言sync.WaitGroup避免并发陷阱

Go语言的并发特性让高效编程成为可能,而sync.WaitGroup是实现高效并发同步的关键工具。它能够等待一组goroutine完成任务后,再执行后续操作。本文将深入讲解sync.WaitGroup的用法,并分析一个常见的错误示例。

示例代码中,Wg()函数启动了三个goroutine,每个goroutine都执行wgCore()函数,模拟一个需要等待完成的任务。初始代码中,wg.Add(1)位于wgCore()函数内部,这意味着每个goroutine只增加计数器1,但wg.Wait()并不知道需要等待多少goroutine完成。sync.WaitGroupWait()方法会阻塞直到计数器归零,由于计数未预先设置,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的正确使用,避免意外结果。

来源:1741115618