前言

waitgroup也是一個非常有用的并發工具，有點像是Java中的CyclicBarrier，只不過Go中的WaitGroup等待的是協程而已。 通常來說，WaitGroup是go并發中最常用的工具了，在起協程并發做一些事兒，我們可以通過WaitGroup了表達這一組協程的任務是否完成，已決定是否繼續往下走，或者取任務結果，下面來看一下WaitGroup的使用及實現。

語法基礎

WaitGroup的核心關注點是：Add()、Done()、Wait()三個函數 Add函數主要為WaitGroup的等待數+1或者+n Done函數調用的也是Add函數，主要用于-1操作 Wait函數是指阻塞當前協程，直到等待數歸為0才繼續向下執行 下面來看一個demo：

func main() {waitGroup := &sync.WaitGroup{}DoSomething(waitGroup)waitGroup.Wait() // 這里會阻塞main，直到所有的任務都完成fmt.Println("end") }func DoSomething(waitGroup *sync.WaitGroup) {for i:=0;i <10;i++ {waitGroup.Add(1)go func(waitGroup *sync.WaitGroup) {fmt.Print("1-")defer waitGroup.Done()}(waitGroup)} }

輸出：

實現原理

首先來看WaitGroup的結構體：

type WaitGroup struct {noCopy noCopystate1 [3]uint32 }

noCopy共64位，高32位是一個計數器，低32位代表有多少在等待，64位原子操作需要64位對齊，但32位編譯器不能確保對齊。因此，我們分配12個字節，然后使用其中對齊的8個字節作為狀態，另外4個字節作為SEMA的存儲。 state1 保存的是狀態信息 接下來看一下核心的兩個函數Add()、Wait() Add()： 1、添加將delta（可能為負）加到waitgroup計數器上。 2、如果計數器變為零，則所有在等待時阻塞的goroutine都被釋放。 3、如果計數器為負，則直接panic。

需要注意的是： 1、當計數器為零時發生的具有正增量的調用必須在等待之前發生，但任何時候都可能發生負增量的調用。這意味著要Add的函數調用應該在創建goroutine或等待的其他事件的語句之前執行。 2、如果waitgroup被重用以等待幾個獨立的事件集，則必須在返回所有以前的wait調用之后進行新的add調用。

func (wg *WaitGroup) Add(delta int) {statep, semap := wg.state()if race.Enabled {_ = *statep // trigger nil deref earlyif delta < 0 {// 與wait變為同步操作race.ReleaseMerge(unsafe.Pointer(wg))}race.Disable()defer race.Enable()}state := atomic.AddUint64(statep, uint64(delta)<<32)v := int32(state >> 32)w := uint32(state)if race.Enabled && delta > 0 && v == int32(delta) { //第一個增量必須與wait同步。 //需要將其變為為一個讀取，因為可以有幾個從0開始的并發wg.counter轉換。race.Read(unsafe.Pointer(semap))}if v < 0 {panic("sync: negative WaitGroup counter")}if w != 0 && delta > 0 && v == int32(delta) {panic("sync: WaitGroup misuse: Add called concurrently with Wait")}if v > 0 || w == 0 {return} //當waiters>0時，此goroutine已將counter設置為0。 //現在狀態不能同時突變： //-添加不能與等待同時發生， //-如果看到counter==0，則wait不會增加waiters。 //仍然執行簡單的檢查以檢測WaitGroup的誤用。if *statep != state {panic("sync: WaitGroup misuse: Add called concurrently with Wait")}// Reset waiters count to 0.*statep = 0for ; w != 0; w-- {runtime_Semrelease(semap, false)} }

Wait()： Wait函數主要用于阻塞那些調用Wait的goroutine，直到waitgroup的任務計數器為0，才會放行，下面來看一下源碼：

func (wg *WaitGroup) Wait() {statep, semap := wg.state()if race.Enabled {_ = *stateprace.Disable()}// 循環檢查計數器情況for {state := atomic.LoadUint64(statep)v := int32(state >> 32)w := uint32(state)if v == 0 {// 如果技術為0的話就不用等待了，完成放行if race.Enabled {race.Enable()race.Acquire(unsafe.Pointer(wg))}return}// 如果有有新的調用wait函數的，添加新的等待者，等待者計數器+1if atomic.CompareAndSwapUint64(statep, state, state+1) {if race.Enabled && w == 0 {// 等待和新添加的是同步的// 只能為第一個競爭者進行寫操作，否則的話會產生互相競爭race.Write(unsafe.Pointer(semap))}runtime_Semacquire(semap)if *statep != 0 {panic("sync: WaitGroup is reused before previous Wait has returned")}if race.Enabled {race.Enable()race.Acquire(unsafe.Pointer(wg))}return}} }

關于WaitGroup暫時介紹這么多。

總結

以上是生活随笔為你收集整理的go int32不能打印0_Go并发实战--sync WaitGroup的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。