Golang?WaitGroup?底層原理及源碼解析

更新時(shí)間：2023年04月27日 09:21:21 作者：杜秉軒?lv-2

WaitGroup?是?Golang?中最常見(jiàn)的并發(fā)控制技術(shù)之一，它的作用我們可以簡(jiǎn)單類(lèi)比為其他語(yǔ)言中多線程并發(fā)控制中的?join()，這篇文章主要介紹了Golang?WaitGroup?底層原理及源碼詳解,需要的朋友可以參考下

0 知識(shí)背景

在進(jìn)入正文前，先對(duì) WaitGroup 及其相關(guān)背景知識(shí)做個(gè)簡(jiǎn)單的介紹，這里主要是 WaitGroup 的基本使用，以及系統(tǒng)信號(hào)量的基礎(chǔ)知識(shí)。對(duì)這些比較熟悉的小伙伴可以直接跳過(guò)這一節(jié)。

0.1 WaitGroup

WaitGroup 是 Golang 中最常見(jiàn)的并發(fā)控制技術(shù)之一，它的作用我們可以簡(jiǎn)單類(lèi)比為其他語(yǔ)言中多線程并發(fā)控制中的 join()，實(shí)例代碼如下：

package main
import (
	"fmt"
	"sync"
	"time"
)
func main() {
	fmt.Println("Main starts...")
	var wg sync.WaitGroup
	// 2 指的是下面有兩個(gè)協(xié)程需要等待
	wg.Add(2)
	go waitFunc(&wg, 3)
	go waitFunc(&wg, 1)
	// 阻塞等待
	wg.Wait()
	fmt.Println("Main ends...")
}
func waitFunc(wg *sync.WaitGroup, num int) {
	// 函數(shù)結(jié)束時(shí)告知 WaitGroup 自己已經(jīng)結(jié)束
	defer wg.Done()
	time.Sleep(time.Duration(num) * time.Second)
	fmt.Printf("Hello World from %v\n", num)
}
// 結(jié)果輸出：
Main starts...
Hello World from 1
Hello World from 3
Main ends...

如果這里沒(méi)有 WaitGroup，主協(xié)程（main 函數(shù)）會(huì)直接跑到最后的 Main ends...，而沒(méi)有中間兩個(gè) goroutine 的輸出，加了 WaitGroup 后，main 就會(huì)在 wg.Wait() 處阻塞等待兩個(gè)協(xié)程都結(jié)束后才繼續(xù)執(zhí)行。

上面我們看到的 WaitGroup 的三個(gè)方法：Wait()、Add(int) 和 Done() 也是 WaitGroup 對(duì)象僅有的三個(gè)方法。

0.2 信號(hào)量（Semaphore）

信號(hào)量（Semaphore）是一種用于實(shí)現(xiàn)多進(jìn)程或多線程之間同步和互斥的機(jī)制，也是 WaitGroup 中所采用的技術(shù)。并且 WaitGroup 自身的同步原理，也與信號(hào)量很相似。

由于翻譯問(wèn)題，不熟悉的小伙伴經(jīng)常將信號(hào)量（Semaphore）和信號(hào)（Signal）搞混，這倆實(shí)際上是兩個(gè)完全不同的東西。Semaphore 在英文中的本意是旗語(yǔ)，也就是航海領(lǐng)域的那個(gè)旗語(yǔ)，利用手旗或旗幟傳遞信號(hào)的溝通方式。在計(jì)算機(jī)領(lǐng)域，Semaphore，即信號(hào)量，在廣義上也可以理解為一種進(jìn)程、線程間的通信方式，但它的主要作用，正如前面所說(shuō)，是用于實(shí)現(xiàn)進(jìn)程、線程間的同步和互斥。

信號(hào)量本質(zhì)上可以簡(jiǎn)單理解為一個(gè)整型數(shù)，主要包含兩種操作：P（Proberen，測(cè)試）操作和 V（Verhogen，增加）操作。其中，P 操作會(huì)嘗試獲取一個(gè)信號(hào)量，如果信號(hào)量的值大于 0，則將信號(hào)量的值減 1 并繼續(xù)執(zhí)行；否則，當(dāng)前進(jìn)程或線程就會(huì)被阻塞，直到有其他進(jìn)程或線程釋放這個(gè)信號(hào)量為止。V 操作則是釋放一個(gè)信號(hào)量，將信號(hào)量的值加 1。

可以把信號(hào)量看作是一種類(lèi)似鎖的東西，P 操作相當(dāng)于獲取鎖，而 V 操作相當(dāng)于釋放鎖。由于信號(hào)量是一種操作系統(tǒng)級(jí)別的機(jī)制，通常由內(nèi)核提供支持，因此我們不用擔(dān)心上述對(duì)信號(hào)量的操作本身會(huì)產(chǎn)生競(jìng)態(tài)條件，相信內(nèi)核能搞定這種東西。

本文的重點(diǎn)不是信號(hào)量，因此不會(huì)過(guò)多展開(kāi)關(guān)于信號(hào)量的技術(shù)細(xì)節(jié)，有興趣的小伙伴可以查閱相關(guān)資料。

最后提一嘴技術(shù)之外的東西，Proberen 和 Verhogen 這倆單詞眼生吧？因?yàn)樗鼈兪呛商m語(yǔ)，不是英語(yǔ)。為啥是荷蘭語(yǔ)嘞？因?yàn)榘l(fā)明信號(hào)量的人，是上古計(jì)算機(jī)大神，來(lái)自荷蘭的計(jì)算機(jī)先驅(qū) Edsger W. Dijkstra 先生。嗯，對(duì)，就是那個(gè) Dijkstra。

1 WaitGroup 底層原理

聲明：本文所用源碼均基于 Go 1.20.3 版本，不同版本 Go 的 WaitGroup 源碼可能略有不同，但設(shè)計(jì)思想基本是一致的。

WaitGroup 相關(guān)源碼非常短，加上注釋和空行也只有 120 多行，它們?nèi)荚?src/sync/waitgroup.go 中。

1.1 定義

先來(lái)看 WaitGroup 的定義，這里我把源文件中的注釋都簡(jiǎn)單翻譯了一下：

// WaitGroup 等待一組 Goroutine 完成。
// 主 Goroutine 調(diào)用 Add 方法設(shè)置要等待的 Goroutine 數(shù)量，
// 然后每個(gè) Goroutine 運(yùn)行并在完成后調(diào)用 Done 方法。
// 同時(shí)，可以使用 Wait 方法阻塞，直到所有 Goroutine 完成。
//
// WaitGroup 在第一次使用后不能被復(fù)制。
//
// 根據(jù) Go 內(nèi)存模型的術(shù)語(yǔ)，Done 調(diào)用“同步于”任何它解除阻塞的 Wait 調(diào)用的返回。
type WaitGroup struct {
	noCopy noCopy
	state atomic.Uint64 // 高 32 位是計(jì)數(shù)器, 低 32 位是等待者數(shù)量（后文解釋?zhuān)?
	sema  uint32
}

WaitGroup 類(lèi)型是一個(gè)結(jié)構(gòu)體，它有三個(gè)私有成員，我們一個(gè)一個(gè)來(lái)看。

1.1.1 noCopy

首先是 noCopy，這個(gè)東西是為了告訴編譯器，WaitGroup 結(jié)構(gòu)體對(duì)象不可復(fù)制，即 wg2 := wg 是非法的。之所以禁止復(fù)制，是為了防止可能發(fā)生的死鎖。但實(shí)際上如果我們對(duì) WaitGroup 對(duì)象進(jìn)行復(fù)制后，至少在 1.20 版本下，Go 的編譯器只是發(fā)出警告，沒(méi)有阻止編譯過(guò)程，我們依然可以編譯成功。警告的內(nèi)容如下：

assignment copies lock value to wg2: sync.WaitGroup contains sync.noCopy

為什么編譯器沒(méi)有編譯失敗，我猜應(yīng)該是 Go 官方想盡量減少編譯器對(duì)程序的干預(yù)，而更多地交給程序員自己去處理（此時(shí) Rust 發(fā)出了一陣笑聲）?？傊覀?cè)谑褂?WaitGroup 的過(guò)程中，不要去復(fù)制它就對(duì)了，不然非常容易產(chǎn)生死鎖（其實(shí)結(jié)構(gòu)體注釋上也說(shuō)了，WaitGroup 在第一次使用后不能被復(fù)制）。譬如我將文章開(kāi)頭代碼中的 main 函數(shù)稍微改了改：

func main() {
	fmt.Println("Main starts...")
	var wg sync.WaitGroup
	// 2 指的是下面有兩個(gè)協(xié)程需要等待
	wg.Add(1)
	wg2 := wg
	wg2.Add(1)
	go waitFunc(&wg, 3)
	go waitFunc(&wg2, 1)
	// 阻塞等待
	wg.Wait()
	wg2.Wait()
	fmt.Println("Main ends...")
}
// 輸出結(jié)果
Main starts...
Hello World from 1
Hello World from 3
fatal error: all goroutines are asleep - deadlock!
goroutine 1 [semacquire]:
sync.runtime_Semacquire(0xc000042060?)
        C:/Program Files/Go/src/runtime/sema.go:62 +0x27
sync.(*WaitGroup).Wait(0xe76b28?)
        C:/Program Files/Go/src/sync/waitgroup.go:116 +0x4b
main.main()
        D:/Codes/Golang/waitgroup/main.go:23 +0x139
exit status 2

為什么會(huì)這樣？因?yàn)?wg 已經(jīng) Add(1) 了，這時(shí)我們復(fù)制了 wg 給 wg2，并且是個(gè)淺拷貝，意味著 wg2 內(nèi)實(shí)際上已經(jīng)是 Add(1) 后的狀態(tài)了（state 成員保存的狀態(tài)，即它的值），此時(shí)我們?cè)賵?zhí)行 wg2.Add(1)，其實(shí)相當(dāng)于執(zhí)行了兩次 wg2.Add(1)。而后面 waitFunc() 中對(duì) wg2 只進(jìn)行了一次 Done() 釋放操作，main 函數(shù)在 wg2.Wait() 時(shí)就陷入了無(wú)限等待，即 all goroutines are asleep。等看了后面 Add() 和 Done() 的原理后，再回頭來(lái)看這段死鎖的代碼，會(huì)更加清晰。

那么這段代碼能既復(fù)制，又不死鎖嗎？當(dāng)然可以，只需要把 wg2 := wg 提到 wg.Add(1) 前面即可。

1.1.2 state atomic.Uint64

state 是 WaitGroup 的核心，它是一個(gè)無(wú)符號(hào)的 64 位整型，并且用的是 atomic 包中的 Uint64，所以 state 本身是線程安全的。至于 atomic.Uint64 為什么能保證線程安全，因?yàn)樗褂昧?CompareAndSwap(CAS) 操作，而這個(gè)操作依賴(lài)于 CPU 提供的原子性指令，是 CPU 級(jí)的原子操作。

state 的高 32 位是計(jì)數(shù)器（counter），低 32 位是等待者數(shù)量（waiters）。其中計(jì)數(shù)器其實(shí)就是 Add(int) 數(shù)量的總和，譬如 Add(1) 后再 Add(2)，那么這個(gè)計(jì)數(shù)器就是 1 + 2 = 3；而等待數(shù)量就是現(xiàn)在有多少 goroutine 在執(zhí)行 Wait() 等待 WaitGroup 被釋放。

1.1.3 sema uint32

這玩意兒就是信號(hào)量，它的用法我們到后文結(jié)合代碼再講。

1.2 Add(delta int)

首先是 Add(delta int) 方法。WaitGroup 所有三個(gè)方法都沒(méi)有返回值，并且只有 Add 擁有參數(shù)，整個(gè)設(shè)計(jì)可謂簡(jiǎn)潔到了極點(diǎn)。

Add 方法的第一句代碼是：

if race.Enabled {
	if delta < 0 {
		// Synchronize decrements with Wait.
		race.ReleaseMerge(unsafe.Pointer(wg))
	}
	race.Disable()
	defer race.Enable()
}

race.Enabled 是判斷當(dāng)前程序是否開(kāi)啟了競(jìng)態(tài)條件檢查，這個(gè)檢查是在編譯時(shí)需要我們手動(dòng)指定的：go build -race main.go，默認(rèn)情況下并不開(kāi)啟，即 race.Enabled 在默認(rèn)情況下就是 false。這段代碼里如果程序開(kāi)啟了競(jìng)態(tài)條件檢查，會(huì)將其關(guān)閉，最后再重新打開(kāi)。其他有關(guān) race 的細(xì)節(jié)本文不再討論，這對(duì)我們理解 WaitGroup 也沒(méi)有太大影響，將其考慮進(jìn)去反而會(huì)增加我們理解 WaitGroup 核心機(jī)制的復(fù)雜度，因此后續(xù)代碼中也會(huì)忽略所有與 race 相關(guān)的部分。

Add 方法整理后的代碼如下：

// Add 方法將 delta 值加上計(jì)數(shù)器，delta 可以為負(fù)數(shù)。如果計(jì)數(shù)器變?yōu)?0，
// 則所有在 Wait 上阻塞的 Goroutine 都會(huì)被釋放。
// 如果計(jì)數(shù)器變?yōu)樨?fù)數(shù)，則 Add 方法會(huì) panic。
//
// 注意：當(dāng)計(jì)數(shù)器為 0 時(shí)調(diào)用 delta 值為正數(shù)的 Add 方法必須在 Wait 方法之前執(zhí)行。
// 而 delta 值為負(fù)數(shù)或者 delta 值為正數(shù)但計(jì)數(shù)器大于 0 時(shí)，則可以在任何時(shí)間點(diǎn)執(zhí)行。
// 通常情況下，這意味著應(yīng)該在創(chuàng)建 Goroutine 或其他等待事件的語(yǔ)句之前執(zhí)行 Add 方法。
// 如果一個(gè) WaitGroup 用于等待多組獨(dú)立的事件，
// 那么必須在所有先前的 Wait 調(diào)用返回之后再進(jìn)行新的 Add 調(diào)用。
// 詳見(jiàn) WaitGroup 示例代碼。
func (wg *WaitGroup) Add(delta int) {
	// 將 int32 的 delta 變成 unint64 后左移 32 位再與 state 累加。
	// 相當(dāng)于將 delta 與 state 的高 32 位累加。
	state := wg.state.Add(uint64(delta) << 32)
	// 高 32 位，就是 counter，計(jì)數(shù)器
	v := int32(state >> 32)
	// 低 32 位，就是 waiters，等待者數(shù)量
	w := uint32(state)
	// 計(jì)數(shù)器為負(fù)數(shù)時(shí)直接 panic
	if v < 0 {
		panic("sync: negative WaitGroup counter")
	}
	// 當(dāng) Wait 和 Add 并發(fā)執(zhí)行時(shí)，會(huì)有概率觸發(fā)下面的 panic
	if w != 0 && delta > 0 && v == int32(delta) {
		panic("sync: WaitGroup misuse: Add called concurrently with Wait")
	}
	// 如果計(jì)數(shù)器大于 0，或者沒(méi)有任何等待者，即沒(méi)有任何 goroutine 在 Wait()，那么就直接返回
	if v > 0 || w == 0 {
		return
	}
	// 當(dāng) waiters > 0 時(shí)，這個(gè) Goroutine 將計(jì)數(shù)器設(shè)置為 0。
	// 現(xiàn)在不可能有對(duì)狀態(tài)的并發(fā)修改：
	// - Add 方法不能與 Wait 方法同時(shí)執(zhí)行，
	// - Wait 不會(huì)在看到計(jì)數(shù)器為 0 時(shí)增加等待者。
	// 仍然需要進(jìn)行簡(jiǎn)單的健全性檢查來(lái)檢測(cè) WaitGroup 的誤用情況。
	if wg.state.Load() != state {
		panic("sync: WaitGroup misuse: Add called concurrently with Wait")
	}
	// 重置 state 為 0
	wg.state.Store(0)
	// 喚醒所有等待者
	for ; w != 0; w-- {
		// 使用信號(hào)量控制喚醒等待者
		runtime_Semrelease(&wg.sema, false, 0)
	}
}

這里我將原代碼中的注釋翻譯成了中文，并且自己在每句代碼前也都加了注釋。

一開(kāi)始，方法將參數(shù) delta 變成 uint64 后左移 32 位，和 state 相加。因?yàn)?state 的高 32 位是這個(gè) WaitGroup 的計(jì)數(shù)器，所以這里其實(shí)就是把計(jì)數(shù)器進(jìn)行了累加操作：

state := wg.state.Add(uint64(delta) << 32)

接著，程序會(huì)分別取出已經(jīng)累加后的計(jì)數(shù)器 v，和當(dāng)前的等待者數(shù)量 w：

v := int32(state >> 32)
w := uint32(state)

然后是幾個(gè)判斷：

// 計(jì)數(shù)器為負(fù)數(shù)時(shí)直接 panic
if v < 0 {
	panic("sync: negative WaitGroup counter")
}
// 當(dāng) Wait 和 Add 并發(fā)執(zhí)行時(shí)，會(huì)有概率觸發(fā)下面的 panic
if w != 0 && delta > 0 && v == int32(delta) {
	panic("sync: WaitGroup misuse: Add called concurrently with Wait")
}
// 如果計(jì)數(shù)器大于 0，或者沒(méi)有任何等待者，
// 即沒(méi)有任何 goroutine 在 Wait()，那么就直接返回
if v > 0 || w == 0 {
	return
}

注釋已經(jīng)比較清晰了，這里主要展開(kāi)解釋一下第二個(gè) if：if w != 0 && delta > 0 && v == int32(delta)。

w != 0 意味著當(dāng)前有 goroutine 在 Wait()；
delta > 0 意味著 Add() 傳入的是正整數(shù)，也就是正常調(diào)用；
v == int32(delta) 意味著累加后的計(jì)數(shù)器等于傳入的 delta，這里最容易想到的符合這個(gè)等式的場(chǎng)景是：原計(jì)數(shù)器等于 0 時(shí)，也就是 wg 第一次使用，或前面的 Wait() 已經(jīng)全部結(jié)束時(shí)。

上述三個(gè)條件看上去有些沖突：w != 0 表示存在 Wait()，而 v == int32(delta) 按照分析應(yīng)該不存在 Wait()。再往下分析，其實(shí)應(yīng)該是 v 在獲取的時(shí)候不存在 Wait()，而 w 在獲取的時(shí)候存在 Wait()。會(huì)有這種可能嗎？會(huì)！就是并發(fā)的時(shí)候：當(dāng)前 goroutine 獲取了 v，然后另一個(gè) goroutine 立刻進(jìn)行了 Wait()，接著本 goroutine 又獲取了 w，過(guò)程如下：

我們可以用下面這段代碼來(lái)復(fù)現(xiàn)這個(gè) panic：

func main() {
	var wg sync.WaitGroup
	// 并發(fā)問(wèn)題不易復(fù)現(xiàn)，所以循環(huán)多次
	for i := 0; i < 100000; i++ {
		go addDoneFunc(&wg)
		go waitFunc(&wg)
	}
	wg.Wait()
}
func addDoneFunc(wg *sync.WaitGroup) {
	wg.Add(1)
	wg.Done()
}
func waitFunc(wg *sync.WaitGroup) {
	wg.Wait()
}
// 輸出結(jié)果
panic: sync: WaitGroup misuse: Add called concurrently with Wait
goroutine 71350 [running]:
sync.(*WaitGroup).Add(0x0?, 0xbf8aa5?)
        C:/Program Files/Go/src/sync/waitgroup.go:65 +0xce      
main.addDoneFunc(0xc1cf66?, 0x0?)
        D:/Codes/Golang/waitgroup/main.go:19 +0x1e
created by main.main
        D:/Codes/Golang/waitgroup/main.go:11 +0x8f
exit status 2

這段代碼可能要多運(yùn)行幾次才會(huì)看到上述效果，因?yàn)檫@種并發(fā)操作在整個(gè) WaitGroup 的生命周期中會(huì)造成好幾種 panic，包括 Wait() 方法中的。

因此，我們?cè)谑褂?WaitGroup 的時(shí)候應(yīng)當(dāng)注意一點(diǎn)：不要在被調(diào)用的 goroutine 內(nèi)部使用 Add，而應(yīng)當(dāng)在外面使用，也就是：

// 正確
wg.Add(1)
go func(wg *sync.WaitGroup) {
	defer wg.Done()
}(&wg)
wg.Wait()
// 錯(cuò)誤
go func(wg *sync.WaitGroup) {
	wg.Add(1)
	defer wg.Done()
}(&wg)
wg.Wait()

從而避免并發(fā)導(dǎo)致的異常。

上面三個(gè) if 都結(jié)束后，會(huì)再次對(duì) state 的一致性進(jìn)行判斷，防止并發(fā)異常：

if wg.state.Load() != state {
	panic("sync: WaitGroup misuse: Add called concurrently with Wait")
}

這里 state.Load() 包括后面會(huì)出現(xiàn)的 Store() 都是 atomic.Uint64 的原子操作。

根據(jù)前面代碼的邏輯，當(dāng)程序運(yùn)行到這里時(shí)，計(jì)數(shù)器一定為 0，而等待者則可能 >= 0，于是代碼會(huì)執(zhí)行一次 wg.state.Store(0) 將 state 設(shè)為 0，接著執(zhí)行通知等待者結(jié)束等待的操作：

wg.state.Store(0)
for ; w != 0; w-- {
	runtime_Semrelease(&wg.sema, false, 0)
}

好了，這里又是讓人迷惑的地方，我第一次看到這段代碼時(shí)產(chǎn)生了下面幾個(gè)疑問(wèn)：

為什么 Add 方法會(huì)有計(jì)數(shù)器為 0 的分支邏輯？計(jì)數(shù)器不是累加的嗎？
為什么要在 Add 中通知等待者結(jié)束，不應(yīng)該是 Done 方法嗎？
那個(gè) runtime_Semrelease(&wg.sema, false, 0) 為什么需要循環(huán) w 次？

一個(gè)一個(gè)來(lái)看。

為什么 Add 方法會(huì)有計(jì)數(shù)器為 0 的分支邏輯？

首先，按照前面代碼的邏輯，只有計(jì)數(shù)器 v 為 0 的時(shí)候，代碼才會(huì)走到最后兩句，而之所以為 0，是因?yàn)?Add(delta int) 的參數(shù) delta 是一個(gè) int，也就是說(shuō)，delta 可以為負(fù)數(shù)！那什么時(shí)候會(huì)傳入負(fù)數(shù)進(jìn)來(lái)呢？Done 的時(shí)候。我們?nèi)タ?Done() 的代碼，會(huì)發(fā)現(xiàn)它非常簡(jiǎn)單：

// Done 給 WaitGroup 的計(jì)數(shù)器減 1。
func (wg *WaitGroup) Done() {
	wg.Add(-1)
}

所以，Done 操作或是我們手動(dòng)給 Add 傳入負(fù)數(shù)時(shí)，就會(huì)進(jìn)入到 Add 最后幾行邏輯，而 Done 本身也意味著當(dāng)前 goroutine 的 WaitGroup 結(jié)束，需要同步給外部的 Wait 讓它不再阻塞。

為什么要在 Add 中通知等待者結(jié)束，不應(yīng)該是 Done 方法嗎？

嗯，這個(gè)問(wèn)題其實(shí)在上一個(gè)問(wèn)題已經(jīng)一起解決了，因?yàn)?Done() 實(shí)際上調(diào)用了 Add(-1)。

那個(gè) runtime_Semrelease(&wg.sema, false, 0) 為什么需要循環(huán) w 次？

這個(gè)函數(shù)按照字面意思，就是釋放信號(hào)量。源碼在 src/sync/runtime.go 中，函數(shù)聲明如下：

// Semrelease 函數(shù)用于原子地增加 *s 的值，
// 并在有等待 Semacquire 函數(shù)被阻塞的協(xié)程時(shí)通知它們繼續(xù)執(zhí)行。
// 它旨在作為同步庫(kù)使用的簡(jiǎn)單喚醒基元，不應(yīng)直接使用。
// 如果 handoff 參數(shù)為 true，則將 count 直接傳遞給第一個(gè)等待者。
// skipframes 參數(shù)表示在跟蹤時(shí)要忽略的幀數(shù)，從 runtime_Semrelease 的調(diào)用者開(kāi)始計(jì)數(shù)。
func runtime_Semrelease(s *uint32, handoff bool, skipframes int)

第一個(gè)參數(shù)就是信號(hào)量的值本身，釋放時(shí)會(huì) +1。

第二個(gè)參數(shù) handoff 在我查閱了資料后，根據(jù)我的理解，應(yīng)該是：當(dāng) handoff 為 false 時(shí)，僅正常喚醒其他等待的協(xié)程，但是不會(huì)立即調(diào)度被喚醒的協(xié)程；而當(dāng) handoff 為 true 時(shí)，會(huì)立刻調(diào)度被喚醒的協(xié)程。

第三個(gè)參數(shù) skipframes，看上去應(yīng)當(dāng)也和調(diào)度有關(guān)，但具體含義我不太確定，這里就不猜了（水平有限，見(jiàn)諒哈）。

按照信號(hào)量本身的機(jī)制，這里釋放時(shí)會(huì) +1，同理還存在一個(gè)信號(hào)量獲取函數(shù) runtime_Semacquire(s *uint32) 會(huì)在信號(hào)量 > 0 時(shí)將信號(hào)量 -1，否則等待，它會(huì)在 Wait() 中被調(diào)用。這也是 runtime_Semrelease 需要循環(huán) w 次的原因：因?yàn)槟?w 個(gè) Wait() 中會(huì)調(diào)用 runtime_Semacquire 并不斷將信號(hào)量 -1，也就是減了 w 次，所以?xún)蓚€(gè)地方需要對(duì)沖一下嘛。

信號(hào)量和 WaitGroup 的機(jī)制很像，但計(jì)數(shù)器又是反的，所以這里再多嘴補(bǔ)充幾句：

信號(hào)量獲取時(shí)（runtime_Semacquire），其實(shí)就是在阻塞等待，P（Proberen，測(cè)試）操作，如果此時(shí)信號(hào)量 > 0，則獲取成功，并將信號(hào)量 -1，否則繼續(xù)等待；

信號(hào)量釋放時(shí)（runtime_Semrelease），會(huì)把信號(hào)量 +1，也就是 V（Verhogen，增加）操作。

1.2 Done()

Done() 方法我們?cè)谏厦嬉呀?jīng)看到過(guò)了：

// Done 給 WaitGroup 的計(jì)數(shù)器減 1。
func (wg *WaitGroup) Done() {
	wg.Add(-1)
}

1.3 Wait()

同樣的，這里我會(huì)把與 race 相關(guān)的代碼都刪掉：

// Wait 會(huì)阻塞，直到計(jì)數(shù)器為 0。
func (wg *WaitGroup) Wait() {
	for {
		state := wg.state.Load()
		v := int32(state >> 32)  // 計(jì)數(shù)器
		w := uint32(state)       // 等待者數(shù)量
		if v == 0 {
			// 計(jì)數(shù)器為 0，直接返回。
			return
		}
		// 增加等待者數(shù)量
		if wg.state.CompareAndSwap(state, state+1) {
			// 獲取信號(hào)量
			runtime_Semacquire(&wg.sema)
			// 這里依然是為了防止并發(fā)問(wèn)題
			if wg.state.Load() != 0 {
				panic("sync: WaitGroup is reused before previous Wait has returned")
			}
			return
		}
	}
}

比 Add 簡(jiǎn)單多了，而且有了前面 Add 的長(zhǎng)篇大論為基礎(chǔ)，Wait 的代碼看上去一目了然。

當(dāng)計(jì)數(shù)器為 0，即沒(méi)有任何 goroutine 調(diào)用 Add 時(shí)，直接調(diào)用 Wait，沒(méi)有任何意義，因此直接返回，也不操作信號(hào)量。

最后 Wait 也有一個(gè)防止并發(fā)問(wèn)題的判斷，而這個(gè) panic 同樣可以用前面 Add 中的那段并發(fā)問(wèn)題代碼復(fù)現(xiàn)，大家可以試試。

Wait 中唯一不同的是，它用了一個(gè)無(wú)限循環(huán) for{}，為什么？這是因?yàn)椋?code>wg.state.CompareAndSwap(state, state+1) 這個(gè)原子操作因?yàn)椴l(fā)等原因有可能失敗，此時(shí)就需要重新獲取 state，把整個(gè)過(guò)程再走一遍。而一旦操作成功，Wait 會(huì)在 runtime_Semacquire(&wg.sema) 處阻塞，直到 Done 操作將計(jì)數(shù)器減為 0，Add 中釋放了信號(hào)量。