Go语言学习系列之协程：使用WaitGroup

之前介绍了 协程 和 信道 的内容，当时为了保证 main goroutine 在所有的 goroutine 都执行完毕后再退出，使用的是time.Sleep 这种方式。由于写的 demo 都是比较简单的， sleep 个 1 秒，我们主观上认为是够用的。

但在实际开发中，开发人员是无法预知所有的 goroutine 需要多长的时间才能执行完毕， sleep 少了吧有的子协程的任务就没法完成。

因此，使用time.Sleep 是一种在实际开发中极不推荐的方式，sleep（多了主程序就阻塞）（ sleep 少了子协程任务未完成）会引起各种问题，今天主要就要来介绍一下如何优雅的处理这种情况。

使用信道来标记完成

“不要通过共享内存来通信，要通过通信来共享内存”

信道可以实现多个协程间的通信，主协程通过监听获取信道是否为true（任务完成后j将true写入），来判断子协程任务是否完成

import "fmt"

func main() {
    done := make(chan bool)
    go func() {
        for i := 0; i < 5; i++ {
            fmt.Println(i)
        }
        done <- true
    }()
    <-done
}

输出如下：
0
1
2
3
4

使用 WaitGroup

上面使用信道的方法，在单个协程或者协程数少的时候，并不会有什么问题，但在协程数多的时候，代码就会显得非常复杂，那么有没有一种更加优雅的方式呢？

有，这就要说到 sync包 提供的 WaitGroup 类型。只要实例化了就能使用 WaitGroup ：

var 实例名 sync.WaitGroup 

实例化完成后，就可以使用它的几个方法：

• Add：初始值为0，你传入的值会往计数器上加，这里直接传入你子协程的数量

• Done：当某个子协程完成后，可调用此方法，会从计数器上减一，通常可以使用 defer 来调用。

• Wait：阻塞当前协程，直到实例里的计数器归零。

举一个栗子：

import (
    "fmt"
    "sync"
)

func worker(x int, wg *sync.WaitGroup) {
    defer wg.Done()
    for i := 0; i < 5; i++ {
        fmt.Printf("worker %d: %d\n", x, i)
    }
}

func main() {
    var wg sync.WaitGroup

    wg.Add(2)
    go worker(1, &wg)
    go worker(2, &wg)

    wg.Wait()
}

输出如下：
worker 2: 0
worker 2: 1
worker 2: 2
worker 2: 3
worker 2: 4
worker 1: 0
worker 1: 1
worker 1: 2
worker 1: 3
worker 1: 4

因此，Go 语言中实现一主多子的协程协作方式，推荐使用 sync.WaitGroup。