分佈式鎖初窺-分佈式鎖的三種實現方式

  • 2020 年 8 月 25 日
  • 筆記

分佈式鎖應該具備哪些條件

在分析分佈式鎖的三種實現方式之前，先了解一下分佈式鎖應該具備哪些條件：

  1. 在分佈式系統環境下，一個方法在同一時間只能被一個機器的一個線程執行；
  2. 高可用的獲取鎖與釋放鎖；
  3. 高性能的獲取鎖與釋放鎖；
  4. 具備可重入特性；
  5. 具備鎖失效機制，防止死鎖；
  6. 具備非阻塞鎖特性，即沒有獲取到鎖將直接返回獲取鎖失敗。

分佈式鎖的三種實現方式

目前幾乎很多大型網站及應用都是分佈式部署的，分佈式場景中的數據一致性問題一直是一個比較重要的話題。分佈式的CAP理論告訴我們「任何一個分佈式系統都無法同時滿足一致性（Consistency）、可用性（Availability）和分區容錯性（Partition tolerance），最多只能同時滿足兩項。」所以，很多系統在設計之初就要對這三者做出取捨。在互聯網領域的絕大多數的場景中，都需要犧牲強一致性來換取系統的高可用性，系統往往只需要保證「最終一致性」，只要這個最終時間是在用戶可以接受的範圍內即可。

在很多場景中，我們為了保證數據的最終一致性，需要很多的技術方案來支持，比如分佈式事務、分佈式鎖等。有的時候，我們需要保證一個方法在同一時間內只能被同一個線程執行。

基於數據庫實現分佈式鎖；
基於緩存（Redis等）實現分佈式鎖；
基於Zookeeper實現分佈式鎖；

基於數據庫的實現方式

基於數據庫的實現方式的核心思想是：在數據庫中創建一個表，表中包含方法名等字段，並在方法名字段上創建唯一索引，想要執行某個方法，就使用這個方法名向表中插入數據，成功插入則獲取鎖，執行完成後刪除對應的行數據釋放鎖。

  1. 創建一個表
DROP TABLE IF EXISTS `method_lock`;
CREATE TABLE `method_lock` (
  `id` int(11) unsigned NOT NULL AUTO_INCREMENT COMMENT '主鍵',
  `method_name` varchar(64) NOT NULL COMMENT '鎖定的方法名',
  `desc` varchar(255) NOT NULL COMMENT '備註信息',
  `update_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
  PRIMARY KEY (`id`),
  UNIQUE KEY `uidx_method_name` (`method_name`) USING BTREE
) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8 COMMENT='鎖定中的方法';
  1. 想要執行某個方法，就使用這個方法名向表中插入數據：
INSERT INTO method_lock (method_name, desc) VALUES ('methodName', '測試的methodName');

因為我們對method_name做了唯一性約束，這裡如果有多個請求同時提交到數據庫的話，數據庫會保證只有一個操作可以成功，那麼我們就可以認為操作成功的那個線程獲得了該方法的鎖，可以執行方法體內容。

  1. 成功插入則獲取鎖，執行完成後刪除對應的行數據釋放鎖：
delete from method_lock where method_name ='methodName';

注意：這只是使用基於數據庫的一種方法，使用數據庫實現分佈式鎖還有很多其他的玩法！

使用基於數據庫的這種實現方式很簡單，但是對於分佈式鎖應該具備的條件來說，它有一些問題需要解決及優化：

  1. 因為是基於數據庫實現的，數據庫的可用性和性能將直接影響分佈式鎖的可用性及性能，所以，數據庫需要雙機部署、數據同步、主備切換；

  2. 不具備可重入的特性，因為同一個線程在釋放鎖之前，行數據一直存在，無法再次成功插入數據，所以，需要在表中新增一列，用於記錄當前獲取到鎖的機器和線程信息，在再次獲取鎖的時候，先查詢表中機器和線程信息是否和當前機器和線程相同，若相同則直接獲取鎖；

  3. 沒有鎖失效機制，因為有可能出現成功插入數據後，服務器宕機了，對應的數據沒有被刪除，當服務恢復後一直獲取不到鎖，所以，需要在表中新增一列，用於記錄失效時間，並且需要有定時任務清除這些失效的數據；

  4. 不具備阻塞鎖特性，獲取不到鎖直接返回失敗，所以需要優化獲取邏輯，循環多次去獲取。

  5. 在實施的過程中會遇到各種不同的問題，為了解決這些問題，實現方式將會越來越複雜；依賴數據庫需要一定的資源開銷，性能問題需要考慮。

基於Redis實現分佈式鎖

  1. 選用Redis實現分佈式鎖原因：
  • Redis有很高的性能；
  • Redis命令對此支持較好，實現起來比較方便
  1. 使用命令介紹：
  • SETNX

SETNX key val：當且僅當key不存在時，set一個key為val的字符串，返回1；若key存在，則什麼都不做，返回0。

  • expire

expire key timeout：為key設置一個超時時間，單位為second，超過這個時間鎖會自動釋放，避免死鎖。

  • delete

delete key：刪除key

在使用Redis實現分佈式鎖的時候，主要就會使用到這三個命令。

  1. 實現思想：
  • 獲取鎖的時候，使用setnx加鎖，並使用expire命令為鎖添加一個超時時間，超過該時間則自動釋放鎖，鎖的value值為一個隨機生成的UUID，通過此在釋放鎖的時候進行判斷。
  • 獲取鎖的時候還設置一個獲取的超時時間，若超過這個時間則放棄獲取鎖。
  • 釋放鎖的時候，通過UUID判斷是不是該鎖，若是該鎖，則執行delete進行鎖釋放。

基於Zookeeper實現分佈式鎖

ZooKeeper是一個為分佈式應用提供一致性服務的開源組件，它內部是一個分層的文件系統目錄樹結構，規定同一個目錄下只能有一個唯一文件名。基於ZooKeeper實現分佈式鎖的步驟如下：

（1）創建一個目錄mylock；
（2）線程A想獲取鎖就在mylock目錄下創建臨時順序節點；
（3）獲取mylock目錄下所有的子節點，然後獲取比自己小的兄弟節點，如果不存在，則說明當前線程順序號最小，獲得鎖；
（4）線程B獲取所有節點，判斷自己不是最小節點，設置監聽比自己次小的節點；
（5）線程A處理完，刪除自己的節點，線程B監聽到變更事件，判斷自己是不是最小的節點，如果是則獲得鎖。

這裡推薦一個Apache的開源庫Curator，它是一個ZooKeeper客戶端，Curator提供的InterProcessMutex是分佈式鎖的實現，acquire方法用於獲取鎖，release方法用於釋放鎖。

優點：具備高可用、可重入、阻塞鎖特性，可解決失效死鎖問題。

缺點：因為需要頻繁的創建和刪除節點，性能上不如Redis方式。

結語

歡迎關注微信公眾號『碼仔zonE』，專註於分享Java、雲計算相關內容，包括SpringBoot、SpringCloud、微服務、Docker、Kubernetes、Python等領域相關技術乾貨，期待與您相遇！

VirMach 便宜 VPS

QNews

QNews