WaitGroup并發(fā)控制

sync.WaitGroup內(nèi)部維護著一個計數(shù)器，計數(shù)器的值可以增加和減少。

例如當(dāng)我們啟動了N 個并發(fā)任務(wù)時，就通過Add()將計數(shù)器值增加N。

每個任務(wù)完成時通過調(diào)用Done()方法將計數(shù)器減1。

通過調(diào)用Wait()來等待并發(fā)任務(wù)執(zhí)行完，當(dāng)計數(shù)器值為0時，表示所有并發(fā)任務(wù)已經(jīng)完成。

const N = 10

var wg = &sync.WaitGroup{}

func main() {

    for i := 0; i < N; i++ {
        //wg.Add(1) 正確寫法
        go func(i int) {
            wg.Add(1)
            println(i)
            defer wg.Done()
        }(i)
    }
    wg.Wait()

}

結(jié)果

結(jié)果不唯一，代碼存在風(fēng)險, 所有g(shù)o未必都能執(zhí)行到

這是使用WaitGroup經(jīng)常犯下的錯誤！請各位同學(xué)多次運行就會發(fā)現(xiàn)輸出都會不同甚至又出現(xiàn)報錯的問題。 這是因為go執(zhí)行太快了，導(dǎo)致wg.Add(1)還沒有執(zhí)行main函數(shù)就執(zhí)行完畢了。

底層原理

type WaitGroup struct {
   noCopy noCopy

   state1 [3]uint32  // uint32數(shù)組一共12個字節(jié)，前8個即uint64記錄 高8位記錄需要等待的數(shù)量 低8位正在等待                       的數(shù)量  ，后4個字節(jié)存儲信號量，用于喚醒
}

1.核心原理就是通過之前說的64位的uint64來進(jìn)行計數(shù)，采用高位記錄需要Done的數(shù)量，低位記錄Wait的數(shù)量，然后排隊休眠等待喚醒

2.如果發(fā)現(xiàn)當(dāng)前count>0則 Wait的goroutine會進(jìn)行排隊

3.任務(wù)完成后的goroutine則進(jìn)行Done操作，直到count==0,則完成，就喚醒所有因為wait操作睡眠的goroutine