9.3. sync.RWMutex讀寫鎖
在100刀的存款消失時不做記錄多少還是會讓我們有一些恐慌,Bob寫了一個程序,每秒運行幾百次來檢查他的銀行餘額。他會在家,在工作中,甚至會在他的手機上來運行這個程序。銀行注意到這些陡增的流量使得存款和取款有了延時,因為所有的餘額查詢請求是順序執行的,這樣會互斥地獲得鎖,並且會暫時阻止其它的goroutine運行。
由於Balance函數只需要讀取變量的狀態,所以我們同時讓多個Balance調用併發運行事實上是安全的,只要在運行的時候沒有存款或者取款操作就行。在這種場景下我們需要一種特殊類型的鎖,其允許多個只讀操作並行執行,但寫操作會完全互斥。這種鎖叫作“多讀單寫”鎖(multiple readers, single writer lock),Go語言提供的這樣的鎖是sync.RWMutex:
var mu sync.RWMutex
var balance int
func Balance() int {
mu.RLock() // readers lock
defer mu.RUnlock()
return balance
}
Balance函數現在調用了RLock和RUnlock方法來獲取和釋放一個讀取或者共享鎖。Deposit函數沒有變化,會調用mu.Lock和mu.Unlock方法來獲取和釋放一個寫或互斥鎖。
在這次修改後,Bob的餘額查詢請求就可以彼此並行地執行並且會很快地完成了。鎖在更多的時間範圍可用,並且存款請求也能夠及時地被響應了。
RLock只能在臨界區共享變量沒有任何寫入操作時可用。一般來說,我們不應該假設邏輯上的只讀函數/方法也不會去更新某一些變量。比如一個方法功能是訪問一個變量,但它也有可能會同時去給一個內部的計數器+1(譯註:可能是記錄這個方法的訪問次數啥的),或者去更新緩存——使即時的調用能夠更快。如果有疑惑的話,請使用互斥鎖。
RWMutex只有當獲得鎖的大部分goroutine都是讀操作,而鎖在競爭條件下,也就是說,goroutine們必須等待才能獲取到鎖的時候,RWMutex才是最能帶來好處的。RWMutex需要更復雜的內部記錄,所以會讓它比一般的無競爭鎖的mutex慢一些。