sync包中的WaitGroup实现了一个类似任务队列的结构,你可以向队列中加入任务,任务完成后就把任务从队列中移除,如果队列中的任务没有全部完成,队列就会触发阻塞以阻止程序继续运行。WaitGroup主要有以下三个方法
// 计数器增加 delta,delta 可以是负数。
func (wg *WaitGroup) Add(delta int)
// 计数器减少 1
func (wg *WaitGroup) Done()
// 等待直到计数器归零。如果计数器小于 0,则该操作会引发 panic。
func (wg *WaitGroup) Wait()
add 会给WaitGroup的counter加1,done会给WaitGroup的counter减1,当WaitGroup的counter的是0 则解除阻塞,否则一定要等任务都执行完毕后再继续执行接下来的操作。
import (
"fmt"
"math/rand"
"sync"
"time"
)
func work(name string,workTime time.Duration,sysGroup *sync.WaitGroup){
defer sysGroup.Done()
fmt.Printf("%s start work\n",name)
time.Sleep(time.Second*workTime)
fmt.Printf("After %d sec %s finid work\n",workTime,name)
}
func main() {
var sysGroup sync.WaitGroup
for i :=0;i<3;i++{
sysGroup.Add(1)
go work(fmt.Sprintf("Worker-%d",i), time.Duration(rand.Intn(10)),&sysGroup)
}
sysGroup.Wait()
fmt.Printf("all people finish work\n")
}