go通过waitgroup等待协程的执行

在go中假如我起了一个goroutine,在主协程中为了让子协程运行完,可以使用 time.Sleep 的方式,主动的等到子协程运行完之后再退出。

但是假如我不知道这个子协程运行的具体时间,这个时候为了让主协程可以等待协程运行完之后退出可以使用 sync.WaitGroup

sync.WaitGroup 提供了三个方法:

Add(), Done(), Wait()

第一个Add,可以添加需要等待的协程数量,第二个Done可以让等待的协程数量减一,最后一个Wait可以放在main函数中,效果是当等待的协程数量清零之后才会运行后面的内容。例如:

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 10; i ++ {
        wg.Add(1)
        go func(i int) {
            defer wg.Done()
            fmt.Println(i)
        }(i)
    }
    wg.Wait()
    fmt.Print("全部协程都运行完了呢!")
}

运行之后的结果应该类似于:

0
9
3
2
4
7
6
8
5
1
全部协程都运行完了呢!
添加新评论