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
|
请发表评论