Go sync WaitGroup使用深入理解

func (wg *WaitGroup) Add(delta int)
func (wg *WaitGroup) Done()
func (wg *WaitGroup) Wait()

type WaitGroup struct {
   // 避免复制
   noCopy noCopy
   // 64位环境下，高32位是计数值，低32位记录waiter的数量
   state1 uint64
   // 用于信号量
   state2 uint32
}

func (wg *WaitGroup) Add(delta int) {
   // 获取状态值，信号量
   statep, semap := wg.state()
   // 将参数delta左32位，加到statep中，即给计数值加上delta
   state := atomic.AddUint64(statep, uint64(delta)<<32)
   // 加后的计数值
   v := int32(state >> 32)
   // waiter的数量
   w := uint32(state)
   // 加后不能是负值
   if v < 0 {
      panic( "sync: negative WaitGroup counter" )
   }
   // 有waiter的情况下，当前协程又加了计数值，panic
   // 即有waiter的情况下，不能再给waitgroup增加计数值了
   if w != 0 && delta > 0 && v == int32(delta) {
      panic( "sync: WaitGroup misuse: Add called concurrently with Wait" )
   }
   // 如果加完后v大于0，或者加完后v等于0，但没有等待者，直接返回
   if v > 0 || w == 0 {
      return
   }
   // 接下来就是v等于0，且w大于0的情况
   // 再次检查是否有Add和Wait并发调用的情况
   if *statep != state {
      panic( "sync: WaitGroup misuse: Add called concurrently with Wait" )
   }
   // 将计数值和waiter数量清0
   *statep = 0
   // 唤醒所有的waiter
   for ; w != 0; w-- {
      runtime_Semrelease(semap, false, 0)
   }
}

func (wg *WaitGroup) Done() {
   wg.Add(-1)
}

func (wg *WaitGroup) Wait() {
   statep, semap := wg.state()
   for {
      state := atomic.LoadUint64(statep)
      // v：计数值
      v := int32(state >> 32)
      w := uint32(state)
      // 如果计数值为0，自己不需要等到，直接返回
      if v == 0 {
         return
   }
      // 增加waiter计数值
 if atomic.CompareAndSwapUint64(statep, state, state+1) {
         // 自己在信号量上阻塞
         runtime_Semacquire(semap)
         // 检查Waitgroup是否在wait返回前被重用
         if *statep != 0 {
            panic( "sync: WaitGroup is reused before previous Wait has returned" )
         }
         return
      }
   }
}