WaitGroup
用于线程同步,WaitGroup等待一组线程集合完成,才会继续向下执行。 主线程 (goroutine) 调用 Add
来设置等待的线程(goroutine)数量。 然后每个线程(goroutine)运行,并在完成后调用 Done
。 同时,Wait
用来阻塞,直到所有线程(goroutine)完成才会向下执行。
对官方的代码做简单修改:
package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
var urls = []string{
"http://www.golang.org/",
"http://www.google.com/",
"http://www.somestupidname.com/",
}
for _, url := range urls {
// Increment the WaitGroup counter.
wg.Add(1)
go func(url string) {
// Launch a goroutine to fetch the URL.
defer wg.Done()
// Fetch the URL.
fmt.Println(url)
}(url)
}
// Wait for all goroutines to finish.
wg.Wait()
fmt.Println("Game Over")
}
执行结果为:
http://www.somestupidname.com/
http://www.golang.org/
http://www.google.com/
Game Over
可以看出3个线程(goroutine)全部执行完成后,wg.Wait()
才停止等待,程序继续往下执行,输出Game Over。