抛砖引玉

假设，在我们的程序中启用了4个Goroutine，分别是G1、G2、G3和G4。其中，G2、G3和G4是由G1中的代码启用并被用于执行某些特定任务的。G1在启用这3个Goroutine之后要等待这些特定任务的完成。在这种情况下，我们有两个方案。

方案1. 通道解决

例如，我们在启用G2、G3和G4之前声明这样一个通道：

  1. sign := make(chan byte, 3)

然后，在G2、G3和G4执行的任务完成之后立即向该通道发送代表了某个任务已被执行完成的元素值：

  1. go func() { // G2
  2. // 省略若干条语句
  3. sign }()
  5. go func() { // G3
  6. // 省略若干条语句
  7. sign }()
  9. go func() { // G4
  10. // 省略若干条语句
  11. sign }()

最后，在启用这几个Goroutine之后，我们还要在G1执行的函数中添加类似这样的代码以等待相关的任务完成信号：

  1. for i := 0; i < 3; i++ {
  2.     fmt.Printf("G%d is ended.\n", <-sign)
  3. }

这样的方法固然是有效的。上面的这条for语句会等到G2、G3和G4都被运行结束之后才会被执行结束，继而其后面的语句才会得以执行。sign通道起到了协调这4个Goroutine的运行的作用。

不过，对于这样一个简单的协调工作来说，使用通道是否过重了？或者说，通道sign是否被大材小用了？通道的实现中包含了很多专为并发安全的数据而建立的数据结构和算法。原则上说，我们不应该把通道当做互斥锁或信号灯来说用。在这里使用它并没有体现出它的优势，反而会在代码易读性和程序性能方面打一些折扣。

方案2. WaitGroup

该需求的第二个方案就是使用 sync.WaitGroup 等待组进行多个任务的同步。

  1. sync.WaitGroup类型的值也是开箱即用的。该类型有三个指针方法，即
    1. Add
      1. 等待组的计数器 +1，虽然Add方法接受一个int类型的值，并且我们也可以通过该方法减少计数值，但是我们一定不要让计数值变为负数。因为这样会立即引发一个运行恐慌。
    2. Done
      1. 完成一个Goroutine，等待组的计数器 -1
    3. Wait。
      1. 阻塞等待所有的Goroutine完成，当等待组计数器不等于 0 时阻塞直到变 0完成。
  2. 简单使用就是在创建一个任务的时候wg.Add(1), 任务完成的时候使用wg.Done()来将任务减一。使用wg.Wait()来阻塞等待所有任务完成。
  3. 值得说明的是，在sync.WaitGroup类型及其方法中也用到了互斥锁、原子操作和信号灯机制。这使得我们总是可以在任意个Goroutine中并发的调用同一个sync.WaitGroup类型值的那些方法。也就是说，它们都是并发安全的。
  1. func main() {
  2.     var wg sync.WaitGroup
  3.     for i := 0; i < 5; i++ {
  4.         wg.Add(1)
  5.         // 方法传递的不是地址，那么就会做一个拷贝，要不调用的wg根本就不是同一个对象，所以需要传递地址
  6.         go func(a int, wg1 *sync.WaitGroup) {
  7.             defer wg1.Done()
  8.             fmt.Println("run-start:",a)
  9.             time.Sleep(3 * time.Second)
  10.             fmt.Println("run-ok:",a)
  11.         }(i,&wg)
  12.     }
  14.     // 阻塞，直到所有wg.Done
  15.     wg.Wait()
  16.     fmt.Println("run-all-ok!")
  17. }
  1. run-start: 1
  2. run-start: 0
  3. run-start: 3
  4. run-start: 4
  5. run-start: 2
  6. run-ok: 2
  7. run-ok: 3
  8. run-ok: 4
  9. run-ok: 1
  10. run-ok: 0
  11. run-all-ok!

感谢

http://ifeve.com/go-concurrent-waitgroup/