调用sync.WaitGroup
后可以Wait()
重复使用吗?
func worker(who string, in <-chan int, wg *sync.WaitGroup) {
for i := range in {
fmt.Println(who, i)
wg.Done()
}
}
func main() {
var wg sync.WaitGroup
AIn := make(chan int, 1)
BIn := make(chan int, 1)
go worker("a:", AIn, &wg)
go worker("b:", BIn, &wg)
for i := 0; i < 4; i++ {
wg.Add(2)
AIn <- i
BIn <- i
wg.Wait()
fmt.Println("main:", i)
}
}
此play.golang.org/p/QLsvA-b4Ae按预期运行,但保证是否安全?文件没有这么说,但也许我只是偏执狂。
答案 0 :(得分:16)
是的,这是安全的。事实上,它甚至比那更安全。您可以同时从多个goroutine Wait
,并根据您的用例交换Add
和Done
次调用。只要Add
发生在Wait
之前,您应该是安全的。
出于好奇,现在WaitGroup
使用互斥锁,两个int32s计数器和一个信号量来实现:
type WaitGroup struct {
m Mutex
counter int32
waiters int32
sema *uint32
}
这也是一个有趣的测试:
var wg1, wg2 sync.WaitGroup
wg1.Add(1)
wg1.Done()
wg1.Wait()
fmt.Println(wg1 == wg2) // Prints true
最后,如果您确实发现有任何问题,请报告,因为这将是一个错误。