go随聊-sync.WaitGroup

        GO提供了sync包和channel来解决协程同步和通讯。新手对channel通道操作起来更容易产生死锁，如果时缓冲的channel还要考虑channel放入和取出数据的速率问题。sync.WaitGroup是等待一组协程结束，它实现了一个类似任务队列的结构，你可以向队列中加入任务，任务完成后就把任务从队列中移除，如果队列中的任务没有全部完成，队列就会触发阻塞以阻止程序继续运行。本文主要介绍sync.WaitGroup。

package sync

type WaitGroup struct  三个重要函数：

//调用Add来设置等待的协程(goroutine)数量
func (wg *WaitGroup) Add(delta int)

//协程(goroutine)完成后调用Done，内部其实就是将值-1
func (wg *WaitGroup) Done() 

//Wait用来阻塞即等待wg值为0，直到所有协程(goroutine)完成才会向下执行
func (wg *WaitGroup) Wait() 

使用例子

定义任务及WaitGroup

var wg sync.WaitGroup
//任务每个1秒打印一次，打印5次
func process(name string) {
    for i:=0;i<5;i++ {
        fmt.Println("process[",name,"] count[",i+1,"]")
        time.Sleep(time.Second)
    }
    //协程完成后调用Done即计数器-1
    wg.Done()
}

启动任务

fmt.Println("start")
wg.Add(1)
//启动任务worker1
go process("worker1")
wg.Add(1)
//启动任务worker2
go process("worker2")
//等待所有任务完成
wg.Wait()
fmt.Println("exit")
----------------------------------
start
process[ worker2 ] count[ 1 ]
process[ worker1 ] count[ 1 ]
process[ worker1 ] count[ 2 ]
process[ worker2 ] count[ 2 ]
process[ worker2 ] count[ 3 ]
process[ worker1 ] count[ 3 ]
process[ worker1 ] count[ 4 ]
process[ worker2 ] count[ 4 ]
process[ worker2 ] count[ 5 ]
process[ worker1 ] count[ 5 ]
exit