Go WaitGroup及Cond底層實現原理
WaitGroup
概念
Go
標準庫提供瞭WaitGroup
原語, 可以用它來等待一批 Goroutine 結束
底層數據結構
// A WaitGroup must not be copied after first use. type WaitGroup struct { noCopy noCopy state1 [3]uint32 }
其中 noCopy
是 golang 源碼中檢測禁止拷貝的技術。如果程序中有 WaitGroup 的賦值行為,使用 go vet
檢查程序時,就會發現有報錯。但需要註意的是,noCopy 不會影響程序正常的編譯和運行。
state1
主要是存儲著狀態和信號量,狀態維護瞭 2 個計數器,一個是請求計數器counter ,另外一個是等待計數器waiter(已調用 WaitGroup.Wait
的 goroutine 的個數)
當數組的首地址是處於一個8
字節對齊的位置上時,那麼就將這個數組的前8
個字節作為64
位值使用表示狀態,後4
個字節作為32
位值表示信號量(semaphore
);同理如果首地址沒有處於8
字節對齊的位置上時,那麼就將前4
個字節作為semaphore
,後8
個字節作為64
位數值。
使用方法
在WaitGroup裡主要有3個方法:
WaitGroup.Add()
:可以添加或減少請求的goroutine數量,Add(n)
將會導致 counter += n
WaitGroup.Done()
:相當於Add(-1),Done()
將導致 counter -=1
,請求計數器counter為0 時通過信號量調用runtime_Semrelease
喚醒waiter線程
WaitGroup.Wait()
:會將 waiter++
,同時通過信號量調用 runtime_Semacquire(semap)
阻塞當前 goroutine
func main() { var wg sync.WaitGroup for i := 1; i <= 5; i++ { wg.Add(1) go func() { defer wg.Done() println("hello") }() } wg.Wait() }
Cond
概念
Go
標準庫提供瞭Cond
原語,可以讓 Goroutine 在滿足特定條件時被阻塞和喚醒
底層數據結構
type Cond struct { noCopy noCopy // L is held while observing or changing the condition L Locker notify notifyList checker copyChecker } type notifyList struct { wait uint32 notify uint32 lock uintptr // key field of the mutex head unsafe.Pointer tail unsafe.Pointer }
主要有4
個字段:
nocopy
: golang 源碼中檢測禁止拷貝的技術。如果程序中有 WaitGroup 的賦值行為,使用 go vet
檢查程序時,就會發現有報錯,但需要註意的是,noCopy 不會影響程序正常的編譯和運行
checker
:用於禁止運行期間發生拷貝,雙重檢查(Double check
)
L
:可以傳入一個讀寫鎖或互斥鎖,當修改條件或者調用Wait
方法時需要加鎖
notify
:通知鏈表,調用Wait()
方法的Goroutine
會放到這個鏈表中,從這裡獲取需被喚醒的Goroutine列表
使用方法
在Cond裡主要有3個方法:
sync.NewCond(l Locker)
: 新建一個 sync.Cond 變量,註意該函數需要一個 Locker 作為必填參數,這是因為在cond.Wait()
中底層會涉及到 Locker 的鎖操作Cond.Wait()
: 阻塞等待被喚醒,調用Wait函數前需要先加鎖;並且由於Wait函數被喚醒時存在虛假喚醒等情況,導致喚醒後發現,條件依舊不成立,因此需要使用 for 語句來循環地進行等待,直到條件成立為止Cond.Signal()
: 隻喚醒一個最先 Wait 的 goroutine,可以不用加鎖Cond.Broadcast()
: 喚醒所有Wait的goroutine,可以不用加鎖
package main import ( "fmt" "sync" "sync/atomic" "time" ) var status int64 func main() { c := sync.NewCond(&sync.Mutex{}) for i := 0; i < 10; i++ { go listen(c) } go broadcast(c) time.Sleep(1 * time.Second) } func broadcast(c *sync.Cond) { // 原子操作 atomic.StoreInt64(&status, 1) c.Broadcast() } func listen(c *sync.Cond) { c.L.Lock() for atomic.LoadInt64(&status) != 1 { c.Wait() // Wait 內部會先調用 c.L.Unlock(),來先釋放鎖,如果調用方不先加鎖的話,會報錯 } fmt.Println("listen") c.L.Unlock() }
以上就是Go WaitGroup及Cond底層實現原理的詳細內容,更多關於Go WaitGroup Cond原理的資料請關註WalkonNet其它相關文章!
推薦閱讀:
- Go並發控制WaitGroup的使用場景分析
- 一文帶你深入理解Go語言中的sync.Cond
- go並發編程sync.Cond使用場景及實現原理
- Go語言中sync.Cond使用詳解
- Go語言學習之WaitGroup用法詳解