package main
import("fmt""sync""time")funcmain(){var wg sync.WaitGroup
// 设置计数器,数值即为goroutine的个数
wg.Add(2)gofunc(){// Do some work
time.Sleep(1* time.Second)
fmt.Println("Goroutine 1 finished!")// goroutine 执行结束后计数器减一
wg.Done()}()gofunc(){// Do some work
time.Sleep(2* time.Second)
fmt.Println("Goroutine 2 finished!")
wg.Done()}()// 主goroutine阻塞等待计数器变为0
wg.Wait()
fmt.Printf("All Goroutine finished!")}
简单的说
上面程序中wg内部维护了一个计数器:
启动goroutine前将计数器通过Add(2)将计数器设置为待启动的goroutine个数
启动goroutine后,使用Wait()方法阻塞自己,等待计数器变为0
每个goroutine执行结束通过Done()方法将计数器减1
计数器变为0后,阻塞的goroutine被唤醒
其实WaitGroup也可以实现一组goroutine等待另一组goroutine
这有点像玩杂技
很容出错
如果不了解其实现原理更是如此
实际上,WaitGroup的实现源码非常简单
基础知识
信号量
信号量是Unix系统提供的一种保护共享资源的机制
用于防止多个线程同时访问某个资源
可简单理解为信号量为一个数值:
当信号量>0时
表示资源可用,获取信号量时系统自动将信号量减1
当信号量==0时
表示资源暂不可用,获取信号量时,当前线程会进入睡眠
当信号量为正时被唤醒
由于WaitGroup实现中也使用了信号量
在此做个简单介绍
WaitGroup
数据结构
state1是个长度为3的数组
其中包含了state和一个信号量
而state实际上是两个计数器:
counter: 当前还未执行结束的goroutine计数器
waiter count: 等待goroutine-group结束的goroutine数量
即有多少个等候者
semaphore: 信号量
考虑到字节是否对齐
三者出现的位置不同
为简单起见
依照字节已对齐情况下
三者在内存中的位置如下所示:
WaitGroup对外提供三个接口:
Add(delta int): 将delta值加到counter中
Wait(): waiter递增1
并阻塞等待信号量semaphore
Done(): counter递减1
按照waiter数值释放相应次数信号量
下面分别介绍这三个函数的实现细节
Add(delta int)
Add()做了两件事
一是把delta值累加到counter中
因为delta可以为负值
也就是说counter有可能变成0或 负值
所以第二件事就是当counter值变为0时
跟据waiter数值释放等量的信号量
把等待的goroutine全部唤醒
如果counter变为负值
则panic
Add()伪代码如下:
func(wg *WaitGroup)Add(delta int){//获取state和semaphore地址指针
statep, semap := wg.state()//把delta左移32位累加到state,即累加到counter中
state := atomic.AddUint64(statep,uint64(delta)<<32)//获取counter值
v :=int32(state >>32)//获取waiter值
w :=uint32(state)//经过累加后counter值变为负值,panicif v <0{panic("sync: negative WaitGroup counter")}//经过累加后,此时,counter >= 0//如果counter为正,说明不需要释放信号量,直接退出//如果waiter为零,说明没有等待者,也不需要释放信号量,直接退出if v >0|| w ==0{return}//此时,counter一定等于0,而waiter一定大于0(内部维护waiter,不会出现小于0的情况),//先把counter置为0,再释放waiter个数的信号量*statep =0for; w !=0; w--{//释放信号量,执行一次释放一个,唤醒一个等待者runtime_Semrelease(semap,false)}}
wait()
Wait()方法也做了两件事
一是累加waiter
二是阻塞等待信号量
func(wg *WaitGroup)Wait(){
statep, semap := wg.state()//获取state和semaphore地址指针for{
state := atomic.LoadUint64(statep)//获取state值
v :=int32(state >>32)//获取counter值
w :=uint32(state)//获取waiter值if v ==0{//如果counter值为0,说明所有goroutine都退出了,不需要待待,直接返回return}// 使用CAS(比较交换算法)累加waiter,累加可能会失败,失败后通过for loop下次重试if atomic.CompareAndSwapUint64(statep, state, state+1){runtime_Semacquire(semap)//累加成功后,等待信号量唤醒自己return}}}