Go语言之城 - 程序示例: 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 值 更多细节可以参考 the FAQ

 
        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

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

下一个例子: 速率限制

Go语言之城 | 闽ICP备20003902号-3