Go by Example 中文版: WaitGroup

想要等待多个协程完成,我们可以使用 *wait group*。

package main
import (
    "fmt"
    "sync"
    "time"
)

每个协程都会运行该函数。 注意,WaitGroup 必须通过指针传递给函数。

func worker(id int, wg *sync.WaitGroup) {

return 时,通知 WaitGroup,当前协程的工作已经完成。

    defer wg.Done()
    fmt.Printf("Worker %d starting\n", id)

睡眠一秒钟,以此来模拟耗时的任务。

    time.Sleep(time.Second)
    fmt.Printf("Worker %d done\n", id)
}
func main() {

WaitGroup 用于等待该函数启动的所有协程。

    var wg sync.WaitGroup

启动几个协程,并为其递增 WaitGroup 的计数器。

    for i := 1; i <= 5; i++ {
        wg.Add(1)
        go worker(i, &wg)
    }

阻塞,直到 WaitGroup 计数器恢复为 0; 即所有协程的工作都已经完成。

    wg.Wait()
}
$ 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

每次运行,各个协程开启和完成的时间可能是不同的。

下一个例子: 速率限制