網站首頁 編程語言 正文
前言
當提到并發編程、多線程編程時,都會在第一時間想到鎖,鎖是并發編程中的同步原語,他可以保證多線程在訪問同一片內存時不會出現競爭來保證并發安全;在
Go
語言中更推崇由channel
通過通信的方式實現共享內存,這個設計點與許多主流編程語言不一致,但是Go
語言也在sync
包中提供了互斥鎖、讀寫鎖,畢竟channel
也不能滿足所有場景,互斥鎖、讀寫鎖的使用與我們是分不開的,所以接下來我會分兩篇來分享互斥鎖、讀寫鎖是怎么實現的,本文我們先來看看互斥鎖的實現。
本文基于Golang
版本:1.18
Go語言互斥鎖設計實現
mutex介紹
sync
包下的mutex
就是互斥鎖,其提供了三個公開方法:調用Lock()
獲得鎖,調用Unlock()
釋放鎖,在Go1.18
新提供了TryLock()
方法可以非阻塞式的取鎖操作:
-
Lock()
:調用Lock
方法進行加鎖操作,使用時應注意在同一個goroutine
中必須在鎖釋放時才能再次上鎖,否則會導致程序panic
。 -
Unlock()
:調用UnLock
方法進行解鎖操作,使用時應注意未加鎖的時候釋放鎖會引起程序panic
,已經鎖定的 Mutex 并不與特定的 goroutine 相關聯,這樣可以利用一個 goroutine 對其加鎖,再利用其他 goroutine 對其解鎖。 -
tryLock()
:調用TryLock
方法嘗試獲取鎖,當鎖被其他 goroutine 占有,或者當前鎖正處于饑餓模式,它將立即返回 false,當鎖可用時嘗試獲取鎖,獲取失敗不會自旋/阻塞,也會立即返回false;
mutex
的結構比較簡單只有兩個字段:
type Mutex struct { state int32 sema uint32 }
-
state
:表示當前互斥鎖的狀態,復合型字段; -
sema
:信號量變量,用來控制等待goroutine
的阻塞休眠和喚醒
初看結構你可能有點懵逼,互斥鎖應該是一個復雜東西,怎么就兩個字段就可以實現?那是因為設計使用了位的方式來做標志,state
的不同位分別表示了不同的狀態,使用最小的內存來表示更多的意義,其中低三位由低到高分別表示mutexed
、mutexWoken
和 mutexStarving
,剩下的位則用來表示當前共有多少個goroutine
在等待鎖:
const ( mutexLocked = 1 << iota // 表示互斥鎖的鎖定狀態 mutexWoken // 表示從正常模式被從喚醒 mutexStarving // 當前的互斥鎖進入饑餓狀態 mutexWaiterShift = iota // 當前互斥鎖上等待者的數量 )
mutex
最開始的實現只有正常模式,在正常模式下等待的線程按照先進先出的方式獲取鎖,但是新創建的gouroutine
會與剛被喚起的 goroutine
競爭,會導致剛被喚起的 goroutine
獲取不到鎖,這種情況的出現會導致線程長時間被阻塞下去,所以Go
語言在1.9
中進行了優化,引入了饑餓模式,當goroutine
超過1ms
沒有獲取到鎖,就會將當前互斥鎖切換到饑餓模式,在饑餓模式中,互斥鎖會直接交給等待隊列最前面的goroutine
,新的 goroutine 在該狀態下不能獲取鎖、也不會進入自旋狀態,它們只會在隊列的末尾等待。如果一個 goroutine 獲得了互斥鎖并且它在隊列的末尾或者它等待的時間少于 1ms,那么當前的互斥鎖就會切換回正常模式。
mutex
的基本情況大家都已經掌握了,接下來我們從加鎖到解鎖來分析mutex
是如何實現的;
Lock加鎖
從Lock
方法入手:
func (m *Mutex) Lock() { // 判斷當前鎖的狀態,如果鎖是完全空閑的,即m.state為0,則對其加鎖,將m.state的值賦為1 if atomic.CompareAndSwapInt32(&m.state, 0, mutexLocked) { if race.Enabled { race.Acquire(unsafe.Pointer(m)) } return } // Slow path (outlined so that the fast path can be inlined) m.lockSlow() }
上面的代碼主要兩部分邏輯:
- 通過
CAS
判斷當前鎖的狀態,也就是state
字段的低1位,如果鎖是完全空閑的,即m.state為0,則對其加鎖,將m.state的值賦為1 - 若當前鎖已經被其他
goroutine
加鎖,則進行lockSlow
方法嘗試通過自旋或饑餓狀態下饑餓goroutine
競爭方式等待鎖的釋放,我們在下面介紹lockSlow
方法;
lockSlow
代碼段有點長,主體是一個for
循環,其主要邏輯可以分為以下三部分:
- 狀態初始化
- 判斷是否符合自旋條件,符合條件進行自旋操作
- 搶鎖準備期望狀態
- 通過
CAS
操作更新期望狀態
初始化狀態
在locakSlow
方法內會先初始化5個字段:
func (m *Mutex) lockSlow() { var waitStartTime int64 starving := false awoke := false iter := 0 old := m.state ........ }
-
waitStartTime
用來計算waiter
的等待時間 -
starving
是饑餓模式標志,如果等待時長超過1ms,starving置為true,后續操作會把Mutex也標記為饑餓狀態。 -
awoke
表示協程是否喚醒,當goroutine
在自旋時,相當于CPU上已經有在等鎖的協程。為避免Mutex解鎖時再喚醒其他協程,自旋時要嘗試把Mutex置為喚醒狀態,Mutex處于喚醒狀態后 要把本協程的 awoke 也置為true。 -
iter
用于記錄協程的自旋次數, -
old
記錄當前鎖的狀態
自旋
自旋的判斷條件非常苛刻:
for { // 判斷是否允許進入自旋 兩個條件,條件1是當前鎖不能處于饑餓狀態 // 條件2是在runtime_canSpin內實現,其邏輯是在多核CPU運行,自旋的次數小于4 if old&(mutexLocked|mutexStarving) == mutexLocked && runtime_canSpin(iter) { // !awoke 判斷當前goroutine不是在喚醒狀態 // old&mutexWoken == 0 表示沒有其他正在喚醒的goroutine // old>>mutexWaiterShift != 0 表示等待隊列中有正在等待的goroutine // atomic.CompareAndSwapInt32(&m.state, old, old|mutexWoken) 嘗試將當前鎖的低2位的Woken狀態位設置為1,表示已被喚醒, 這是為了通知在解鎖Unlock()中不要再喚醒其他的waiter了 if !awoke && old&mutexWoken == 0 && old>>mutexWaiterShift != 0 && atomic.CompareAndSwapInt32(&m.state, old, old|mutexWoken) { // 設置當前goroutine喚醒成功 awoke = true } // 進行自旋 runtime_doSpin() // 自旋次數 iter++ // 記錄當前鎖的狀態 old = m.state continue } }
自旋這里的條件還是很復雜的,我們想讓當前goroutine
進入自旋轉的原因是我們樂觀的認為當前正在持有鎖的goroutine能在較短的時間內歸還鎖,所以我們需要一些條件來判斷,mutex
的判斷條件我們在文字描述一下:
old&(mutexLocked|mutexStarving) == mutexLocked
用來判斷鎖是否處于正常模式且加鎖,為什么要這么判斷呢?
mutexLocked
二進制表示為 0001
mutexStarving
二進制表示為 0100
mutexLocked|mutexStarving
二進制為 0101. 使用0101在當前狀態做 &
操作,如果當前處于饑餓模式,低三位一定會是1,如果當前處于加鎖模式,低1位一定會是1,所以使用該方法就可以判斷出當前鎖是否處于正常模式且加鎖;
runtime_canSpin()
方法用來判斷是否符合自旋條件:
// / go/go1.18/src/runtime/proc.go const active_spin = 4 func sync_runtime_canSpin(i int) bool { if i >= active_spin || ncpu <= 1 || gomaxprocs <= int32(sched.npidle+sched.nmspinning)+1 { return false } if p := getg().m.p.ptr(); !runqempty(p) { return false } return true }
自旋條件如下:
- 自旋的次數要在4次以內
-
CPU
必須為多核 GOMAXPROCS>1
- 當前機器上至少存在一個正在運行的處理器 P 并且處理的運行隊列為空;
判斷當前goroutine
可以進自旋后,調用runtime_doSpin
方法進行自旋:
const active_spin_cnt = 30 func sync_runtime_doSpin() { procyield(active_spin_cnt) } // asm_amd64.s TEXT runtime·procyield(SB),NOSPLIT,$0-0 MOVL cycles+0(FP), AX again: PAUSE SUBL $1, AX JNZ again RET
循環次數被設置為30
次,自旋操作就是執行30次PAUSE
指令,通過該指令占用CPU
并消費CPU
時間,進行忙等待;
這就是整個自旋操作的邏輯,這個就是為了優化 等待阻塞->喚醒->參與搶占鎖這個過程不高效,所以使用自旋進行優化,在期望在這個過程中鎖被釋放。
搶鎖準備期望狀態
自旋邏輯處理好后開始根據上下文計算當前互斥鎖最新的狀態,根據不同的條件來計算mutexLocked
、mutexStarving
、mutexWoken
和 mutexWaiterShift
:
首先計算mutexLocked
的值:
// 基于old狀態聲明到一個新狀態 new := old // 新狀態處于非饑餓的條件下才可以加鎖 if old&mutexStarving == 0 { new |= mutexLocked }
計算mutexWaiterShift
的值:
//如果old已經處于加鎖或者饑餓狀態,則等待者按照FIFO的順序排隊 if old&(mutexLocked|mutexStarving) != 0 { new += 1 << mutexWaiterShift }
計算mutexStarving
的值:
// 如果當前鎖處于饑餓模式,并且已被加鎖,則將低3位的Starving狀態位設置為1,表示饑餓 if starving && old&mutexLocked != 0 { new |= mutexStarving }
計算mutexWoken
的值:
// 當前goroutine的waiter被喚醒,則重置flag if awoke { // 喚醒狀態不一致,直接拋出異常 if new&mutexWoken == 0 { throw("sync: inconsistent mutex state") } // 新狀態清除喚醒標記,因為后面的goroutine只會阻塞或者搶鎖成功 // 如果是掛起狀態,那就需要等待其他釋放鎖的goroutine來喚醒。 // 假如其他goroutine在unlock的時候發現Woken的位置不是0,則就不會去喚醒,那該goroutine就無法在被喚醒后加鎖 new &^= mutexWoken }
通過CAS
操作更新期望狀態
上面我們已經得到了鎖的期望狀態,接下來通過CAS
將鎖的狀態進行更新:
// 這里嘗試將鎖的狀態更新為期望狀態 if atomic.CompareAndSwapInt32(&m.state, old, new) { // 如果原來鎖的狀態是沒有加鎖的并且不處于饑餓狀態,則表示當前goroutine已經獲取到鎖了,直接推出即可 if old&(mutexLocked|mutexStarving) == 0 { break // locked the mutex with CAS } // 到這里就表示goroutine還沒有獲取到鎖,waitStartTime是goroutine開始等待的時間,waitStartTime != 0就表示當前goroutine已經等待過了,則需要將其放置在等待隊列隊頭,否則就排到隊列隊尾 queueLifo := waitStartTime != 0 if waitStartTime == 0 { waitStartTime = runtime_nanotime() } // 阻塞等待 runtime_SemacquireMutex(&m.sema, queueLifo, 1) // 被信號量喚醒后檢查當前goroutine是否應該表示為饑餓 // 1. 當前goroutine已經饑餓 // 2. goroutine已經等待了1ms以上 starving = starving || runtime_nanotime()-waitStartTime > starvationThresholdNs // 再次獲取當前鎖的狀態 old = m.state // 如果當前處于饑餓模式, if old&mutexStarving != 0 { // 如果當前鎖既不是被獲取也不是被喚醒狀態,或者等待隊列為空 這代表鎖狀態產生了不一致的問題 if old&(mutexLocked|mutexWoken) != 0 || old>>mutexWaiterShift == 0 { throw("sync: inconsistent mutex state") } // 當前goroutine已經獲取了鎖,等待隊列-1 delta := int32(mutexLocked - 1<<mutexWaiterShift // 當前goroutine非饑餓狀態 或者 等待隊列只剩下一個waiter,則退出饑餓模式(清除饑餓標識位) if !starving || old>>mutexWaiterShift == 1 { delta -= mutexStarving } // 更新狀態值并中止for循環,拿到鎖退出 atomic.AddInt32(&m.state, delta) break } // 設置當前goroutine為喚醒狀態,且重置自璇次數 awoke = true iter = 0 } else { // 鎖被其他goroutine占用了,還原狀態繼續for循環 old = m.state }
這塊的邏輯很復雜,通過CAS
來判斷是否獲取到鎖,沒有通過 CAS 獲得鎖,會調用 runtime.sync_runtime_SemacquireMutex
通過信號量保證資源不會被兩個 goroutine
獲取,runtime.sync_runtime_SemacquireMutex
會在方法中不斷嘗試獲取鎖并陷入休眠等待信號量的釋放,一旦當前 goroutine
可以獲取信號量,它就會立刻返回,如果是新來的goroutine
,就需要放在隊尾;如果是被喚醒的等待鎖的goroutine
,就放在隊頭,整個過程還需要啃代碼來加深理解。
解鎖
相對于加鎖操作,解鎖的邏輯就沒有那么復雜了,接下來我們來看一看UnLock
的邏輯:
func (m *Mutex) Unlock() { // Fast path: drop lock bit. new := atomic.AddInt32(&m.state, -mutexLocked) if new != 0 { // Outlined slow path to allow inlining the fast path. // To hide unlockSlow during tracing we skip one extra frame when tracing GoUnblock. m.unlockSlow(new) } }
使用AddInt32
方法快速進行解鎖,將m.state的低1位置為0,然后判斷新的m.state值,如果值為0,則代表當前鎖已經完全空閑了,結束解鎖,不等于0
說明當前鎖沒有被占用,會有等待的goroutine
還未被喚醒,需要進行一系列喚醒操作,這部分邏輯就在unlockSlow
方法內:
func (m *Mutex) unlockSlow(new int32) { // 這里表示解鎖了一個沒有上鎖的鎖,則直接發生panic if (new+mutexLocked)&mutexLocked == 0 { throw("sync: unlock of unlocked mutex") } // 正常模式的釋放鎖邏輯 if new&mutexStarving == 0 { old := new for { // 如果沒有等待者則直接返回即可 // 如果鎖處于加鎖的狀態,表示已經有goroutine獲取到了鎖,可以返回 // 如果鎖處于喚醒狀態,這表明有等待的goroutine被喚醒了,不用嘗試獲取其他goroutine了 // 如果鎖處于饑餓模式,鎖之后會直接給等待隊頭goroutine if old>>mutexWaiterShift == 0 || old&(mutexLocked|mutexWoken|mutexStarving) != 0 { return } // 搶占喚醒標志位,這里是想要把鎖的狀態設置為被喚醒,然后waiter隊列-1 new = (old - 1<<mutexWaiterShift) | mutexWoken if atomic.CompareAndSwapInt32(&m.state, old, new) { // 搶占成功喚醒一個goroutine runtime_Semrelease(&m.sema, false, 1) return } // 執行搶占不成功時重新更新一下狀態信息,下次for循環繼續處理 old = m.state } } else { // 饑餓模式釋放鎖邏輯,直接喚醒等待隊列goroutine runtime_Semrelease(&m.sema, true, 1) } }
我們在喚醒goroutine
時正常模式/饑餓模式都調用func runtime_Semrelease(s *uint32, handoff bool, skipframes int)
,這兩種模式在第二個參數的傳參上不同,如果handoff is true, pass count directly to the first waiter.
。
非阻塞加鎖
Go
語言在1.18
版本中引入了非阻塞加鎖的方法TryLock()
,其實現就很簡潔:
func (m *Mutex) TryLock() bool { // 記錄當前狀態 old := m.state // 處于加鎖狀態/饑餓狀態直接獲取鎖失敗 if old&(mutexLocked|mutexStarving) != 0 { return false } // 嘗試獲取鎖,獲取失敗直接獲取失敗 if !atomic.CompareAndSwapInt32(&m.state, old, old|mutexLocked) { return false } return true }
TryLock
的實現就比較簡單了,主要就是兩個判斷邏輯:
- 判斷當前鎖的狀態,如果鎖處于加鎖狀態或饑餓狀態直接獲取鎖失敗
- 嘗試獲取鎖,獲取失敗直接獲取鎖失敗
TryLock
并不被鼓勵使用,至少我還沒想到有什么場景可以使用到它。
總結
通讀源碼后你會發現互斥鎖的邏輯真的十分復雜,代碼量雖然不多,但是很難以理解,一些細節點還需要大家多看看幾遍才能理解其為什么這樣做,文末我們再總結一下互斥鎖的知識點:
- 互斥鎖有兩種模式:正常模式、饑餓模式,饑餓模式的出現是為了優化正常模式下剛被喚起的
goroutine
與新創建的goroutine
競爭時長時間獲取不到鎖,在Go1.9
時引入饑餓模式,如果一個goroutine
獲取鎖失敗超過1ms
,則會將Mutex
切換為饑餓模式,如果一個goroutine
獲得了鎖,并且他在等待隊列隊尾 或者 他等待小于1ms
,則會將Mutex
的模式切換回正常模式 - 加鎖的過程:
- 鎖處于完全空閑狀態,通過CAS直接加鎖
- 當鎖處于正常模式、加鎖狀態下,并且符合自旋條件,則會嘗試最多4次的自旋
- 若當前
goroutine
不滿足自旋條件時,計算當前goroutine的鎖期望狀態 - 嘗試使用CAS更新鎖狀態,若更新鎖狀態成功判斷當前
goroutine
是否可以獲取到鎖,獲取到鎖直接退出即可,若獲取不到鎖則陷入睡眠,等待被喚醒 - goroutine被喚醒后,如果鎖處于饑餓模式,則直接拿到鎖,否則重置自旋次數、標志喚醒位,重新走for循環自旋、獲取鎖邏輯;
- 解鎖的過程
- 原子操作mutexLocked,如果鎖為完全空閑狀態,直接解鎖成功
- 如果鎖不是完全空閑狀態,,那么進入
unlockedslow
邏輯 - 如果解鎖一個未上鎖的鎖直接panic,因為沒加鎖
mutexLocked
的值為0,解鎖時進行mutexLocked - 1操作,這個操作會讓整個互斥鎖混亂,所以需要有這個判斷 - 如果鎖處于饑餓模式直接喚醒等待隊列隊頭的waiter
- 如果鎖處于正常模式下,沒有等待的goroutine可以直接退出,如果鎖已經處于鎖定狀態、喚醒狀態、饑餓模式則可以直接退出,因為已經有被喚醒的
goroutine
獲得了鎖.
- 使用互斥鎖時切記拷貝
Mutex
,因為拷貝Mutex
時會連帶狀態一起拷貝,因為Lock
時只有鎖在完全空閑時才會獲取鎖成功,拷貝時連帶狀態一起拷貝后,會造成死鎖 - TryLock的實現邏輯很簡單,主要判斷當前鎖處于加鎖狀態、饑餓模式就會直接獲取鎖失敗,嘗試獲取鎖失敗直接返回;
原文鏈接:https://juejin.cn/post/7113931205308841997
相關推薦
- 2022-01-27 workerman執行busy,http請求不返回導致阻塞
- 2022-06-12 詳解python的字典及相關操作_python
- 2022-12-26 使用python如何對圖片進行壓縮_python
- 2022-05-15 C++設計模式中的觀察者模式一起來看看_C 語言
- 2023-12-22 獲取微信小程序版本號,uni
- 2022-04-01 關于python中if __name=‘__main__‘的理解
- 2024-03-08 SpringBoot 項目啟動報錯 Failed to configure a DataSource
- 2022-06-06 MybatisPlus二級緩存體系探究分析_相關技巧
- 最近更新
-
- window11 系統安裝 yarn
- 超詳細win安裝深度學習環境2025年最新版(
- Linux 中運行的top命令 怎么退出?
- MySQL 中decimal 的用法? 存儲小
- get 、set 、toString 方法的使
- @Resource和 @Autowired注解
- Java基礎操作-- 運算符,流程控制 Flo
- 1. Int 和Integer 的區別,Jav
- spring @retryable不生效的一種
- Spring Security之認證信息的處理
- Spring Security之認證過濾器
- Spring Security概述快速入門
- Spring Security之配置體系
- 【SpringBoot】SpringCache
- Spring Security之基于方法配置權
- redisson分布式鎖中waittime的設
- maven:解決release錯誤:Artif
- restTemplate使用總結
- Spring Security之安全異常處理
- MybatisPlus優雅實現加密?
- Spring ioc容器與Bean的生命周期。
- 【探索SpringCloud】服務發現-Nac
- Spring Security之基于HttpR
- Redis 底層數據結構-簡單動態字符串(SD
- arthas操作spring被代理目標對象命令
- Spring中的單例模式應用詳解
- 聊聊消息隊列,發送消息的4種方式
- bootspring第三方資源配置管理
- GIT同步修改后的遠程分支