WaitGroup用于线程同步,WaitGroup等待一组线程集合完成,才会继续向下执行。
示例中 main 等待 job1 job2 job3结束后, wg.Wait 才返回
wg.Add(3) 表明要等待3个 goroutines
每调用一次 Done ,减少一个等待的goroutine计数
如果我不小心, wg.Add(4), 那么就程序就会永远无法结束
如果wg.Add(2) , 那么只要等到 wg.Done 两次 , main wg.Wait() 就返回了,
package main
import (
"sync"
"time"
"log"// "github.com/sirupsen/logrus"
)
func job(wg *sync.WaitGroup, name string, sleepSecond time.Duration) {
defer wg.Done()
time.Sleep(time.Second*sleepSecond)
log.Print(name + " done")
}
func main() {
log.Print("hello world")
var wg sync.WaitGroup
wg.Add(3)
go job(&wg, "job1", 1)
go job(&wg, "job2", 2)
go job(&wg, "job3", 1)
wg.Wait()
log.Print("all job done")
}