使用 Redis 實現分散式鎖

發表於2015-11-26

分散式鎖是一個在很多環境中非常有用的原語,它是不同程式互斥操作共享資源的唯一方法。有很多的開發庫和部落格描述如何使用Redis實現DLM(Distributed Lock Manager),但是每個開發庫使用不同的方式,而且相比更復雜的設計與實現,很多庫使用一些簡單低可靠的方式來實現。

這篇文章嘗試提供更標準的演算法來使用Redis實現分散式鎖。我們提出一種演算法,叫做Relock,它實現了我們認為比vanilla單一例項方式更安全的DLM(分散式鎖管理)。我們希望社群分析它並提供反饋,以做為更加複雜或替代設計的一個實現。

實現

在說具體演算法之前,下面有一些具體的實現可供參考.

安全和活躍性保證

從有效分散式鎖的最小保證粒度來說,我們的模型裡面只用了3個屬性,具體如下:

1. 屬性安全: 互斥行.在任何時候,只有一個客戶端可以獲得鎖.

2. 活躍屬性A: 死鎖自由. 即使一個客戶端已經擁用了已損壞或已被分割資源的鎖,但它也有可能請求其他的鎖.

3. 活躍屬性B:容錯. 只要大部分Redis節點可用, 客戶端就可以獲得和釋放鎖.

為何基於容錯的實現還不夠

要理解我們所做的改進,就要先分析下當前基於Redis的分散式鎖的做法。

使用Redis鎖住資源的最簡單的方法是建立一對key-value值。利用Redis的超時機制,key被建立為有一定的生存期,因此它最終會被釋放。而當客戶端想要釋放時,直接刪除key就行了。

一般來說這工作得很好,但有個問題: 這是系統的一個單點。如果Redis主節點掛了呢?當然,我們可以加個子節點,主節點出問題時可以切換過來。不過很可惜,這種方案不可行,因為Redis的主-從複製是非同步的,我們無法用其實現互斥的安全特性。

這明顯是該模型的一種競態條件:

  1. 客戶端A在主節點獲得了一個鎖。
  2. 主節點掛了,而到從節點的寫同步還沒完成。
  3. 從節點被提升為主節點。
  4. 客戶端B獲得和A相同的鎖。注意,安全性被破壞了!

有時候,在某些情況下這反而工作得很好,例如在出錯時,多個客戶端可以獲得同一個鎖。如果這正好是你想要的,那就可以使用主-從複製的方案。否則,我們建議使用這篇文章中描述的方法。

單例項的正確實現方案

在嘗試解決上文描述的單例項方案的缺陷之前,先讓我們確保針對這種簡單的情況,怎麼做才是無誤的,因為這種方案對某些程式而言也是可以接受的,而且這也是我們即將描述的分散式方案的基礎。

為了獲取鎖,方法是這樣的:

這條指令將設定key的值,僅當其不存在時生效(NX選項), 且設定其生存期為30000毫秒(PX選項)。和key關聯的value值是”my_random_value”。這個值在所有客戶端和所有加鎖請求中是必須是唯一的。

使用隨機值主要是為了能夠安全地釋放鎖,這要同時結合這麼個處理邏輯:刪除key值當且僅當其已存在並且其value值是我們所期待的。看看以下lua程式碼:

這麼做很重要,可以避免誤刪其他客戶端建立的鎖。例如某個客戶端獲得了一個鎖,但它的處理時長超過了鎖的有效時長,之後它刪除了這個鎖,而此時這個鎖可能又被其他客戶端給獲得了。僅僅做刪除是不夠安全的,很可能會把其他客戶端的鎖給刪了。結合上面的程式碼,每個鎖都有個唯一的隨機值,因此僅當這個值依舊是客戶端所設定的值時,才會去刪除它。

那麼應該怎樣生成這個隨機值呢?我們使用的是從/dev/urandom讀取的20個位元組,但你也可以找個更簡單的方法,只要能滿足任務就行。例如,可以使用/dev/urandom初始化RC4演算法,然後用其產生隨機數流。更簡單的方法是組合unix時間戳和客戶端ID, 這並不安全,但對很多環境而言也夠用了。

我們所說的key的時間,是指”鎖的有效時長“. 它代表兩種情況,一種是指鎖的自動釋放時長,另一種是指在另一個客戶端獲取鎖之前某個客戶端佔用這個鎖的時長,這被限制在從鎖獲取後開始的一段時間視窗內。

現在我們已經有好的辦法獲取和釋放鎖了。在單例項非分散式系統中,只要保證節點沒掛掉,這個方法就是安全的。那麼讓我們把這個概念擴充套件到分散式的系統中吧,那裡可沒有這種保證。

Redlock 演算法

在此演算法的分散式版本中,我們假設有N個Redis主節點。這些節點是相互獨立的,因此我們不使用複製或其他隱式同步機制。我們已經描述過在單例項情況下如何安全地獲取鎖。我們也指出此演算法將使用這種方法從單例項獲取和釋放鎖。在以下示例中,我們設定N=5(這是個比較適中的值),這樣我們需要在不同物理機或虛擬機器上執行5個Redis主節點,以確保它們的出錯是儘可能獨立的。

為了獲取鎖,客戶端執行以下操作:

  1. 獲取當前時間,以毫秒為單位。
  2. 以序列的方式嘗試從所有的N個例項中獲取鎖,使用的是相同的key值和相同的隨機value值。在從每個例項獲取鎖時,客戶端會設定一個連線超時,其時長相比鎖的自動釋放時間要短得多。例如,若鎖的自動釋放時間是10秒,那麼連線超時大概設在5到50毫秒之間。這可以避免當Redis節點掛掉時,會長時間堵住客戶端:如果某個節點沒及時響應,就應該儘快轉到下個節點。
  3. 客戶端計算獲取所有鎖耗費的時長,方法是使用當前時間減去步驟1中的時間戳。當且僅當客戶端能從多數節點(至少3個)中獲得鎖,並且耗費的時長小於鎖的有效期時,可認為鎖已經獲得了。
  4. 如果鎖獲得了,它的最終有效時長將重新計算為其原時長減去步驟3中獲取鎖耗費的時長。
  5. 如果鎖獲取失敗了(要麼是沒有鎖住N/2+1個節點,要麼是鎖的最終有效時長為負數),客戶端會對所有例項進行解鎖操作(即使對那些沒有加鎖成功的例項也一樣)。

演算法是非同步的?

演算法依賴於這樣一個假定,它在處理的時候不是(基於)同步時鐘的,每個處理中仍然使用的是本地的時間,它只是大致地以同樣地速率執行,這樣它就會有一個小的錯誤,與之相比會有一個小的自動開合的時鐘時間。這個假設很像真正世界的電腦:每一臺電腦有一個本地時鐘,通常我們使用不同的電腦會有一個很小的時鐘差。

基於這個觀點,我們需要更好地指明我們共同的互斥法則:這是保證客戶端能長時間保持狀態鎖定,其將會終止它們在有效時間內的工作(在步驟3中獲得),減去一些時間(在處理時時間差時減去了一些毫秒用來補償)。

想要了解關於系統需要一個範圍的時間差的內容可以獲取更多的資訊,這篇論文是很好的參考: Leases: an efficient fault-tolerant mechanism for distributed file cache consistency.

失敗時重試

當客戶端無法獲取鎖時,它應該在一個隨機延遲後重試,從而避免多個客戶端同時試圖獲取鎖,相對應同一的同時請求(這可能會導致崩潰,沒人會勝出)。同樣的,客戶端在大多數場合下嘗試獲取鎖的速度越快,崩潰的視窗就越少(重試的需要也越少),所以實際情況下客戶端應嘗試採用複用方式傳送SET命令到多個例項。

強調客戶在獲取主鎖失敗是值得的,釋放(或部分)以儘快獲得鎖,這樣沒有必要為獲取鎖鎖而去等待鍵到期(但是如果網路分割槽發生變化時客戶端不能與Redis通訊的情況下,需要顯性提示和等待超時)。

釋放鎖

釋放鎖是簡單的,只需要釋放所有例項的鎖即可,儘管客戶端認為有能力成功鎖住一個給出的例項。

安全引數

要問一個演算法是安全的麼?那麼可以嘗試著去理解在不同的情景下發生了什麼。我們以假設客戶端在大多數情況下都能獲得鎖來開始,所有的例項都包含相同生存週期的鍵。由於鍵是在不同的時間設定的,所以鍵也將在不同的時間超時。然而,如果第一個節點最遲在t1時刻建立樣品接觸的第一伺服器之前,上一個鍵最遲在T2時刻建立(從上一個伺服器獲得回覆的時間)。可以確定的是第一個鍵在超時之前將生存至少MIN_VALIDITY=TTL-(T2-T1)-CLOCK_DRIFT。所有其他的鑰匙將到期後,鑰匙將至少在這一次同時設定

在過半的鍵被設定這段時間裡,另一個客戶端無法獲得鎖,如果N/2+1個鍵已經存在,N/2+1 SET NX操作將不能成功。所以一個鎖被獲取,同一時刻被重複獲取是不可能的(違反互斥性)。

然而我們還想讓多個客戶端在獲取鎖的時候不能同時成功。

如果一個客戶端鎖定大部分例項的時間超過了鎖的最大有效時間(TTL基本設定) ,它將考慮鎖無效了,並解鎖。所以我們僅考慮在有效時間內大部分例項獲得鎖的情況。這種情況已經在上文中討論過, 對於MIN_VALIDITY沒有客戶端會重新獲取鎖。所以只有當鎖大多數例項的時間超過TTL時間時,多客戶端才能同時鎖住N/2+1個例項(在步驟2的“時間”即將結束時),讓鎖失效。

你是否能提供一個形式化的證明,指出現存的足夠相似的演算法,或找出些bug? 那我們將感激不盡。

存活性證明

系統的存活性基於以下三個主要特性:

  1. 鎖的自動釋放(key會到期): 最終所有的key將可以被重新鎖住;
  2. 一般來說,客戶端如果沒有成功獲得鎖,或者獲得了鎖並且完成了工作,都會及時釋放鎖,使得我們無需等待key自動釋放以重新獲得。
  3. 當客戶端重新獲取鎖之前,它會等待一段時間,這段時間比獲取鎖本身要長得多,這是為了儘量降低資源競爭引起的腦裂條件的概率。

然而,在網路割裂的情況下,我們得付出等同於”TTL”時間的可用性代價,如果網路持續割裂,我們就得無限的付出這個代價。這發生於當客戶端獲取了一個鎖,而在刪除鎖之前網路斷開了。

基本上,如果網路無限期地持續割裂,那系統將無限期地不可用。

 

效能、故障恢復和檔案同步

許多使用者使用Redis作為一個需要高效能的加鎖伺服器,可以根據延遲動態的獲取和釋放鎖,每秒可以成功執行大量的獲取/釋放鎖操作。為了滿足這些需求,一種多路複用策略是協同N臺 Redis伺服器減少延遲(或者叫做窮人的互助,也就是說,將埠置為non-blocking模式,傳送所有的命令,延遲讀出所有的命令,假定客戶端和每個Redis例項的往返時間是相似的)。

然而,如果我們旨在實現一種故障系統的恢復模式,這裡有另一種與永續性相關的思路。

考慮這個基本問題,假定我們完全沒有配置Redis的永續性。一個客戶端需要鎖定5個例項中的3個。其中一個允許客戶端獲取的鎖重新啟動,雖然我們可以再次為一些資源鎖定3個例項,但其它的客戶端同樣可以鎖定它,違反了排他鎖安全性。

如果我們啟用AOF永續性,情況就會得到相當的改善。例如我們可以通過傳送 SHUTDOWN升級一個伺服器並且重啟它。因為Redis的期限是通過語義設定的,所以伺服器關閉的情況下虛擬時間仍然會流逝,我們所有的需求都得到了滿足。不管怎樣所有事務都會正常運轉只要伺服器完全關閉。如果電源中斷會怎樣?如果Redis進行了相關配置,預設情況下每秒檔案都會同步寫入磁碟,很有可能在重啟後我們的資料會丟失。理論上,如果我們想在任何一種例項重啟後保證鎖的安全性,我們需要確保在永續性配置中設定fsync=always。這將會在同等級別的CP系統上損失效能,傳統上這種方式用來更安全的分配鎖。

不管怎樣事情比我們初次瞥見他們看起來好些。基本上演算法的安全性得到保留,就算是當一個例項在故障後重啟,它也將不再參與任何當前活躍的鎖的分配。因此當例項重啟時,當前所有活動鎖的設定將從鎖定的例項中獲取除它重新加入系統。

為了保證這一點,我們只需要做一個例項,在超過最大TTL後,崩潰,不可用,那麼就需要時間去獲取所有存在著的鎖的鑰匙,當例項崩潰時,其就會變得無效,會被自動釋放。

使用延時重啟可以基本上實現安全,甚至不需要利用任何Redis的持久化特性,但是這存在著另外的副作用。舉例來說,如果大量的例項崩潰,系統變得全域性不可用,那麼TTL(這裡的全域性意味著根本就沒有資源可用,在這個時間內所有的資源都會被鎖定)。

讓演算法更可靠: 擴充套件鎖

如果客戶工作的執行是由小步驟組成,那麼它就可以在預設時間裡預設使用更小的鎖,並擴充套件了演算法去實現的一個鎖的擴充套件機制。當鎖的有效性接近於一個低值,那麼通常是客戶端在運算中處於居中位置。當鎖被取得時,可能擴充套件的鎖通過傳送一個Lua指令碼到所有的例項,這個例項是擴充套件TTL的鑰匙,如果鑰匙存在,那麼它的值就是客戶端複製的隨機值。

客戶端應該僅考慮鎖的重新取得,如果它可以被擴充套件,鎖就會在有效時間內進入大量例項(基本的演算法使用非常類似於獲取鎖的使用)。

雖然這不是從技術上去改變演算法,但是無論如何嘗試獲取鎖的最大次數是需要限制的,否則的話會違反活躍性中的一個屬性。

希望對您有所幫助

如果你用的是分散式系統,對於你的意見或是分析都將是非常有用的。同時用其他語言實現也是很有意義的參考。

先謝謝了!

相關文章