sync.WaitGroup的基本使用

主要方法如下

方法名	功能
(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")
}