Go by Example 中文版: WaitGroup
|
想要等待多个协程完成，我们可以使用 wait group 。 |
|
 
package main |
|
import ( "fmt" "sync" "time" ) |
|
|
每个协程都会运行该函数。 |
func worker(id int) { fmt.Printf("Worker %d starting\n", id) |
|
睡眠一秒钟，以此来模拟耗时的任务。 |
time.Sleep(time.Second)
fmt.Printf("Worker %d done\n", id)
}
|
func main() { |
|
|
这个 WaitGroup 用于等待这里启动的所有协程完成。 注意：如果 WaitGroup 显式传递到函数中，则应使用 指针 。 |
var wg sync.WaitGroup
|
|
启动几个协程，并为其递增 WaitGroup 的计数器。 |
for i := 1; i <= 5; i++ {
wg.Add(1)
|
|
避免在每个协程闭包中重复利用相同的 |
i := i
|
|
将 worker 调用包装在一个闭包中，可以确保通知 WaitGroup 此工作线程已完成。 这样，worker 线程本身就不必知道其执行中涉及的并发原语。 |
go func() {
defer wg.Done()
worker(i)
}()
}
|
|
阻塞，直到 WaitGroup 计数器恢复为 0； 即所有协程的工作都已经完成。 |
wg.Wait()
|
|
请注意，WaitGroup 这种使用方式没有直观的办法传递来自 worker 的错误。 更高级的用例，请参见 errgroup package |
}
|
$ go run waitgroups.go Worker 5 starting Worker 3 starting Worker 4 starting Worker 1 starting Worker 2 starting Worker 4 done Worker 1 done Worker 2 done Worker 5 done Worker 3 done |
|
|
每次运行，各个协程开启和完成的时间可能是不同的。 |
下一个例子: 速率限制