golang的WaitGroup解决了什么问题

    设想这么一种场景，起多个goroutine同时去做事，等他们都做完了程序才能继续往下走。由于不同的goroutine完成工作的时间不同，怎么来等待他们都完成呢？sync包里的WaitGroup可以解决这个问题。示例如下：

 

package main

import (
        "fmt"
        "sync"
        "time"
)

var wg sync.WaitGroup

func main() {
        fmt.Println("start", time.Now().Unix())

        wg.Add(1)
        go func() {
                time.Sleep(time.Second * 2)
                fmt.Println("goroutine1 ready!", time.Now().Unix())

                wg.Done()
        }()

        wg.Add(1)
        go func() {
                time.Sleep(time.Second * 5)
                fmt.Println("goroutine2 ready!", time.Now().Unix())

                wg.Done()
        }()

        fmt.Println("waiting...", time.Now().Unix())
        wg.Wait()

        fmt.Println("all ready!", time.Now().Unix())
}

1. 定义一个全局的WaitGroup对象：var wg sync.WaitGroup

2. 在起goroutine之前调用Add方法加1，表示需要等待一个goroutine：wg.Add(1)；在goroutine退出前调用Done方法减1（源码中Done方法实际上是wg.Add(-1)），表示goroutine退出

3. 以同样的方式起了两个goroutine，第一需要2s才能退出，第二个需要5s才能退出

运行结果如下：

start 1605797142
waiting... 1605797142
goroutine1 ready! 1605797144
goroutine2 ready! 1605797147
all ready! 1605797147

可以看到程序启动后迅速的执行到了wg.Wait()，阻塞住了没有继续打印“all ready!”。当2s后goroutine1完成，又过了3s goroutine2完成后，程序才继续执行打印出了“all ready!”。

总结：WaitGroup可以很好的解决等待多个goroutine都完成任务后再继续执行的问题，实践中可以实现进程的优雅退出，等进程内的goroutine都退出后，进程才退出。