golang中的sync.WaitGroup

Golang的sync的包有一个并发原语WaitGroup，在日常开发中比较的有用。

WaitGroup的用途：它能够一直等到所有的goroutine执行完成，在其期间会会阻塞主线程的执行，直到所有的goroutine执行完成。

这里要注意一下，在其中的多个goroutine 的执行结果是没有顺序的，调度器不能保证多个 goroutine 执行次序，且进程退出时不会等待它们结束。

WaitGroup总共有三个方法：Add(delta int), Done() 和 Wait()。简单的说一下这三个方法的作用。

Add: 添加或者减少等待goroutine的数量

Done: 相当于Add(-1)

Wait: 执行阻塞，直到所有的WaitGroup数量变成0

举个例子

package main

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

func main() {
    var wg sync.WaitGroup

    for i := 0; i < 5; i++ {
        wg.Add(1)
        go func(n int) {
            defer wg.Done(),注意这个Done的位置，是另一个函数
            // 也可以将 defer wg.Done() 换成 defer wg.Add(-1)
            EchoNumber(n)
        }(i)
    }

    wg.Wait()
}

func EchoNumber(i int) {
    time.Sleep(3e9)
    fmt.Println(i)
}

golang中的同步是通过 sync.WaitGroup 来实现的。它实现了一个类似队列的结构，可以一直向队列中添加任务，当任务完成后便从队列中删除，如果队列中的任务没有完全完成，可以通过 Wait() 函数来出发阻塞，防止程序继续进行，直到所有的队列任务都完成为止．

WaitGroup的特点是Wait()可以用来阻塞直到队列中的所有任务都完成时才解除阻塞，而不需要sleep一个固定的时间来等待．但是其缺点是无法指定固定的goroutine数目．不过可以通过使用channel解决此问题。

另一个例子

package main

import (
    "fmt"
    "sync"
)

//声明一个全局变量
var waitgroup sync.WaitGroup

func Afunction(shownum int) {
    fmt.Println(shownum)
    defer waitgroup.Done() //任务完成，将任务队列中的任务数量-1，其实.Done就是.Add(-1)
}

func main() {
    for i := 0; i < 10; i++ {
        waitgroup.Add(1) //每创建一个goroutine，就把任务队列中任务的数量+1
        go Afunction(i)
    }
    waitgroup.Wait() //.Wait()这里会发生阻塞，直到队列中所有的任务结束就会解除阻塞
}