詳解Golang互斥鎖內(nèi)部實(shí)現(xiàn)
go語言提供了一種開箱即用的共享資源的方式,互斥鎖(sync.Mutex), sync.Mutex的零值表示一個(gè)沒有被鎖的,可以直接使用的,一個(gè)goroutine獲得互斥鎖后其他的goroutine只能等到這個(gè)gorutine釋放該互斥鎖,在Mutex結(jié)構(gòu)中只公開了兩個(gè)函數(shù),分別是Lock和Unlock,在使用互斥鎖的時(shí)候非常簡(jiǎn)單,本文并不闡述使用。
在使用sync.Mutex的時(shí)候千萬不要做值拷貝,因?yàn)檫@樣可能會(huì)導(dǎo)致鎖失效。當(dāng)我們打開我們的IDE時(shí)候跳到我們的sync.Mutex 代碼中會(huì)發(fā)現(xiàn)它有如下的結(jié)構(gòu):
type Mutex struct { state int32 //互斥鎖上鎖狀態(tài)枚舉值如下所示 sema uint32 //信號(hào)量,向處于Gwaitting的G發(fā)送信號(hào) } const ( mutexLocked = 1 << iota // 1 互斥鎖是鎖定的 mutexWoken // 2 喚醒鎖 mutexWaiterShift = iota // 2 統(tǒng)計(jì)阻塞在這個(gè)互斥鎖上的goroutine數(shù)目需要移位的數(shù)值 )
上面的state值分別為 0(可用) 1(被鎖) 2~31等待隊(duì)列計(jì)數(shù)
下面是互斥鎖的源碼,這里會(huì)有四個(gè)比較重要的方法需要提前解釋,分別是runtime_canSpin,runtime_doSpin,runtime_SemacquireMutex,runtime_Semrelease,
1、runtime_canSpin:比較保守的自旋,golang中自旋鎖并不會(huì)一直自旋下去,在runtime包中runtime_canSpin方法做了一些限制, 傳遞過來的iter大等于4或者cpu核數(shù)小等于1,最大邏輯處理器大于1,至少有個(gè)本地的P隊(duì)列,并且本地的P隊(duì)列可運(yùn)行G隊(duì)列為空。
//go:linkname sync_runtime_canSpin sync.runtime_canSpin 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 }
2、 runtime_doSpin:會(huì)調(diào)用procyield函數(shù),該函數(shù)也是匯編語言實(shí)現(xiàn)。函數(shù)內(nèi)部循環(huán)調(diào)用PAUSE指令。PAUSE指令什么都不做,但是會(huì)消耗CPU時(shí)間,在執(zhí)行PAUSE指令時(shí),CPU不會(huì)對(duì)它做不必要的優(yōu)化。
//go:linkname sync_runtime_doSpin sync.runtime_doSpin func sync_runtime_doSpin() { procyield(active_spin_cnt) }
3、runtime_SemacquireMutex:
//go:linkname sync_runtime_SemacquireMutex sync.runtime_SemacquireMutex func sync_runtime_SemacquireMutex(addr *uint32) { semacquire(addr, semaBlockProfile|semaMutexProfile) }
4、runtime_Semrelease:
//go:linkname sync_runtime_Semrelease sync.runtime_Semrelease func sync_runtime_Semrelease(addr *uint32) { semrelease(addr) } Mutex的Lock函數(shù)定義如下 func (m *Mutex) Lock() { //先使用CAS嘗試獲取鎖 if atomic.CompareAndSwapInt32(&m.state, 0, mutexLocked) { //這里是-race不需要管它 if race.Enabled { race.Acquire(unsafe.Pointer(m)) } //成功獲取返回 return } awoke := false //循環(huán)標(biāo)記 iter := 0 //循環(huán)計(jì)數(shù)器 for { old := m.state //獲取當(dāng)前鎖狀態(tài) new := old | mutexLocked //將當(dāng)前狀態(tài)最后一位指定1 if old&mutexLocked != 0 { //如果所以被占用 if runtime_canSpin(iter) { //檢查是否可以進(jìn)入自旋鎖 if !awoke && old&mutexWoken == 0 && old>>mutexWaiterShift != 0 && atomic.CompareAndSwapInt32(&m.state, old, old|mutexWoken) { //awoke標(biāo)記為true awoke = true } //進(jìn)入自旋狀態(tài) runtime_doSpin() iter++ continue } //沒有獲取到鎖,當(dāng)前G進(jìn)入Gwaitting狀態(tài) new = old + 1<<mutexWaiterShift } if awoke { if new&mutexWoken == 0 { throw("sync: inconsistent mutex state") } //清除標(biāo)記 new &^= mutexWoken } //更新狀態(tài) if atomic.CompareAndSwapInt32(&m.state, old, new) { if old&mutexLocked == 0 { break } // 鎖請(qǐng)求失敗,進(jìn)入休眠狀態(tài),等待信號(hào)喚醒后重新開始循環(huán) runtime_SemacquireMutex(&m.sema) awoke = true iter = 0 } } if race.Enabled { race.Acquire(unsafe.Pointer(m)) } } Mutex的Unlock函數(shù)定義如下 func (m *Mutex) Unlock() { if race.Enabled { _ = m.state race.Release(unsafe.Pointer(m)) } // 移除標(biāo)記 new := atomic.AddInt32(&m.state, -mutexLocked) if (new+mutexLocked)&mutexLocked == 0 { throw("sync: unlock of unlocked mutex") } old := new for { //當(dāng)休眠隊(duì)列內(nèi)的等待計(jì)數(shù)為0或者自旋狀態(tài)計(jì)數(shù)器為0,退出 if old>>mutexWaiterShift == 0 || old&(mutexLocked|mutexWoken) != 0 { return } // 減少等待次數(shù),添加清除標(biāo)記 new = (old - 1<<mutexWaiterShift) | mutexWoken if atomic.CompareAndSwapInt32(&m.state, old, new) { // 釋放鎖,發(fā)送釋放信號(hào) runtime_Semrelease(&m.sema) return } old = m.state } }
互斥鎖無沖突是最簡(jiǎn)單的情況了,有沖突時(shí),首先進(jìn)行自旋,,因?yàn)榇蠖鄶?shù)的Mutex保護(hù)的代碼段都很短,經(jīng)過短暫的自旋就可以獲得;如果自旋等待無果,就只好通過信號(hào)量來讓當(dāng)前Goroutine進(jìn)入Gwaitting狀態(tài)。
以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
GoLang協(xié)程庫libtask學(xué)習(xí)筆記
libtask一個(gè)C語言的協(xié)程庫,是go語言的前身很早期的原型. 測(cè)試機(jī)器是我的mac air 安裝的centos虛擬機(jī)(只有一個(gè)核), 代碼沒有采用任何優(yōu)化,只是使用默認(rèn)配置2022-12-12Golang發(fā)送Get和Post請(qǐng)求的實(shí)現(xiàn)
做第三方接口有時(shí)需要用Get或者Post請(qǐng)求訪問,本文主要介紹了Golang發(fā)送Get和Post請(qǐng)求的實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2024-05-05golang中sync.Once只執(zhí)行一次的原理解析
在某些場(chǎng)景下,我們希望某個(gè)操作或者函數(shù)僅被執(zhí)行一次,比如單例模式的初始化,一些資源配置的加載等,golang中的sync.Once就實(shí)現(xiàn)了這個(gè)功能,本文就和大家一起解析sync.Once只執(zhí)行一次的原理,需要的朋友可以參考下2023-09-09Go語言中io.Reader和io.Writer的詳解與實(shí)現(xiàn)
在Go語言的實(shí)際編程中,幾乎所有的數(shù)據(jù)結(jié)構(gòu)都圍繞接口展開,接口是Go語言中所有數(shù)據(jù)結(jié)構(gòu)的核心。在使用Go語言的過程中,無論你是實(shí)現(xiàn)web應(yīng)用程序,還是控制臺(tái)輸入輸出,又或者是網(wǎng)絡(luò)操作,不可避免的會(huì)遇到IO操作,使用到io.Reader和io.Writer接口。下面來詳細(xì)看看。2016-09-09