WaitGroup

如果你不关心并发操作的结果，或者有其他方式收集结果，那么WaitGroup是等待一组并发操作完成的好方法；否则还是建议你改用channel和select语句。可以把WaitGroup视作一个安全的并发计数器：调用Add增加计数，调用Done减少计数。调用Wait会阻塞并等待至计数器归零。注意，Add的调用是在goroutines之外完成的。如果没有这样做，我们会引入一个数据竞争条件，因为我们没有对goroutine做任何调度顺序上的保证;我们可能在任何一个goroutines开始前触发Wait调用。如果Add的调用被放置在goroutines的闭包中，对Wait的调用可能完全没有阻塞地返回，因为Add没有被执行。

var wg sync.WaitGroup

wg.Add(1) //1
go func() {
	defer wg.Done() //2
	fmt.Println("1st goroutine sleeping...")
	time.Sleep(1)
}()

wg.Add(1) //1
go func() {
	defer wg.Done() //2
	fmt.Println("2nd goroutine sleeping...")
	time.Sleep(2)
}()

wg.Wait() //3
fmt.Println("All goroutines complete.")

通常情况下，尽可能与要跟踪的goroutine就近且成对的调用Add，但有时候会一次性调用Add来跟踪一组goroutine。我通常会做这样的循环：

hello := func(wg *sync.WaitGroup, id int) {
	defer wg.Done()
	fmt.Printf("Hello from %v!\n", id)
}

const numGreeters = 5
var wg sync.WaitGroup
wg.Add(numGreeters)
for i := 0; i < numGreeters; i++ {
	go hello(&wg, i+1)
}
wg.Wait()

WaitGroup

WaitGroup

Links