在使用 go 的 goroutine 时主线程不需要等待其它线程结束就可以继续执行后续操作,但有时有些业务逻辑的处理需要等待所有线程结束后才能执行,这个时候可以使用 sync.WaitGroup。
直接上代码:
import (
    "fmt"
    "sync"
)
var wg sync.WaitGroup
wg.Add(2)
go func(){
	time.Sleep(time.Duration(1) * time.Second)
	fmt.Println("test1")
	wg.Done() // wg.Add(-1)
}()
go func(){
	time.Sleep(time.Duration(3) * time.Second)
	fmt.Println("test2")
	wg.Done() // wg.Add(-1)
}()
wg.Wait()
fmt.Println("end.")
	
执行结果:
test1
test2
end.
如果不启用sync.WaitGroup执行结果会是:
end.
test1
test2
End.