主要方法如下
方法名 | 功能 |
---|---|
(wg * WaitGroup) Add(delta int) | 计算器增加/减少一定次数 |
(wg * WaitGroup) Done() | 计数器减1,等价于wg.Add(-1) |
(wg * WaitGroup) Wait() | 阻塞程序,直到计算器归零 |
注意,如果调佣wg.Add(delta) 或者 wg.Done() 将 wg 维护的计数更改成一个负数,会发生panic。
package main
import (
"fmt"
"sync"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done()
fmt.Printf("Worker %d starting\n", id)
// do some work
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 5; i++ {
wg.Add(1)
go worker(i, &wg)
}
wg.Wait()
fmt.Println("All workers done")
}