這幾種常見的“分布式鎖”寫法,搞懂再也不怕面試官,安排

什么是分布式鎖?

今天跟大家聊一聊分布式鎖。

首先說下什么是分布式鎖,當我們在進行下訂單減庫存,搶票,選課,搶紅包這些業務場景時,如果在此處沒有鎖的控制,會導致很嚴重的問題。

學過多線程的小伙們知道,為了防止多個線程同時執行同一段代碼,我們可以用 synchronized 關鍵字或 JUC 里面的 ReentrantLock 類來控制,但是目前幾乎任何一個系統都是部署多臺機器的,單機部署的應用很少,synchronized 和 ReentrantLock 發揮不出任何作用,此時就需要一把全局的鎖,來代替 JAVA 中的 synchronized 和 ReentrantLock。

分布式鎖的實現方式流行的主要有三種,分別是基于緩存 Redis 的實現方式,基于 zk 臨時順序節點的實現以及基于數據庫行鎖的實現。我們先來說下用 Jedis 中的 setnx 命令來構建這把鎖。

Jedis寫法

使用 Redis 做分布式鎖的思路是,在 redis 中設置一個值表示加了鎖,然后釋放鎖的時候就把這個 key 刪除。思路是很簡單,但是在使用過程中要避免一些坑,我們先看下加鎖的代碼:

    /**
     * 嘗試獲取分布式鎖
     *
     * @param jedis      Redis客戶端
     * @param lockKey    鎖
     * @param requestId  請求標識
     * @param expireTime 超期時間
     * @return 是否獲取成功
     */
    public static boolean tryGetDistributedLock(Jedis jedis, String lockKey, String requestId, int expireTime) {
        // set支持多個參數 NX(not exist) XX(exist) EX(seconds) PX(million seconds)
        String result = jedis.set(lockKey, requestId, "NX", "EX", expireTime);
        if (LOCK_SUCCESS.equals(result)) {
            return true;
        }
        return false;
    }

這段代碼很簡單,主要說下這里用的命令是 SET key value [EX seconds|PX milliseconds] [NX|XX] [KEEPTTL],而沒有使用 SETNX+EXPIRE 的命令,原因是 SETNX+EXPIRE 是兩條命令無法保證原子性,而 SET 是原子操作。

那這里為什么要設置超時時間呢?

原因是當一個客戶端獲得了鎖在執行任務的過程中掛掉了,來不及顯式地釋放鎖,這塊資源將會永遠被鎖住,這將會導致死鎖,所以必須設置一個超時時間。

釋放鎖的代碼如下:

    /**
     * 釋放分布式鎖
     *
     * @param jedis     Redis客戶端
     * @param lockKey   鎖
     * @param requestId 請求標識,當前工作線程線程的名稱
     * @return 是否釋放成功
     */
    public static boolean releaseDistributedLock(Jedis jedis, String lockKey, String requestId) {
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
        Object result = jedis.eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId));
        if (RELEASE_SUCCESS.equals(result)) {
            return true;
        }
        return false;
    }

這里也有兩點注意的地方,第一是解鈴還須系鈴人,怎么理解呢,就是 A 加的鎖 B 不能去 del 掉吧,不然豈不是全亂套了,誰加的鎖就誰去解,我們一般把 value 設為當前線程的 Id,Thread.currentThread().getId(),然后在刪的時候判斷下是不是當前線程。

第二點是驗證和釋放鎖是兩個獨立操作,不是原子性,這個怎么解決呢?使用 Lua 腳本,即 if redis.call('get', KEYS[1]) == ARGV[1] then returnredis.call('del', KEYS[1]) else return 0 end,它能給我們保證原子性。

Redisson寫法

Redisson 是 Java 的 Redis 客戶端之一,提供了一些 API 方便操作 Redis。但是 Redisson 這個客戶端可有點厲害,我們先打開官網看下https://github.com/redisson/redisson/wiki/%E7%9B%AE%E5%BD%95

file

這個目錄里面有很多的功能,Redisson 跟 Jedis 定位不同,它不是一個單純的 Redis 客戶端,而是基于 Redis 實現的分布式的服務,我們可以看到還有 JUC 包下面的類名,Redisson 幫我們搞了分布式的版本,比如 AtomicLong,直接用 RedissonAtomicLong 就行了。

鎖只是它的冰山一角,并且它對主從,哨兵,集群等模式都支持,當然了,單節點模式肯定是支持的。

在 Redisson 里面提供了更加簡單的分布式鎖的實現,我們來看下它的用法,相當的簡單,兩行代碼搞定,比 Jedis 要簡單的多,而且在 Jedis 里需要考慮的問題,它都已經幫我們封裝好了。

file

我們來看下,這里獲取鎖有很多種的方式,有公平鎖有讀寫鎖,我們使用的是 redissonClient.getLock, 這是一個可重入鎖。

file

現在我把程序啟動一下

file

打開 Redis Desktop Manager 工具,看下到底它存的是什么。原來在加鎖的時候,寫入了一個 HASH 類型的值,key 是鎖名稱 jackxu,field 是線程的名稱,而 value 是 1(即表示鎖的重入次數)。

file

小伙伴可能覺得我在一派胡言,沒關系,我們點進去看下它的源碼是具體實現的。

file

點進 tryLock() 方法的 tryAcquire() 方法,再到->tryAcquireAsync() 再到->tryLockInnerAsync(),終于見到廬山真面目了,原來它最終也是通過 Lua 腳本來實現的。

file

現在我把這段Lua腳本拉出來分析一下,很簡單。

// KEYS[1] 鎖名稱 updateAccount
// ARGV[1] key 過期時間 10000ms
// ARGV[2] 線程名稱
// 鎖名稱不存在
if (redis.call('exists', KEYS[1]) == 0) then
// 創建一個 hash,key=鎖名稱,field=線程名,value=1
redis.call('hset', KEYS[1], ARGV[2], 1);
// 設置 hash 的過期時間
redis.call('pexpire', KEYS[1], ARGV[1]);
return nil;
end;
// 鎖名稱存在,判斷是否當前線程持有的鎖
if (redis.call('hexists', KEYS[1], ARGV[2]) == 1) then
// 如果是,value+1,代表重入次數+1
redis.call('hincrby', KEYS[1], ARGV[2], 1);
// 重新獲得鎖,需要重新設置 Key 的過期時間
redis.call('pexpire', KEYS[1], ARGV[1]);
return nil;
end;
// 鎖存在,但是不是當前線程持有,返回過期時間(毫秒)
return redis.call('pttl', KEYS[1]);

unlock() 中的 unlockInnerAsync() 釋放鎖,同樣也是通過 Lua 腳本實現。

// KEYS[1] 鎖的名稱 updateAccount
// KEYS[2] 頻道名稱 redisson_lock__channel:{updateAccount}
// ARGV[1] 釋放鎖的消息 0
// ARGV[2] 鎖釋放時間 10000
// ARGV[3] 線程名稱
// 鎖不存在(過期或者已經釋放了)
if (redis.call('exists', KEYS[1]) == 0) then
// 發布鎖已經釋放的消息
redis.call('publish', KEYS[2], ARGV[1]);
return 1;
end;
// 鎖存在,但是不是當前線程加的鎖
if (redis.call('hexists', KEYS[1], ARGV[3]) == 0) then
return nil;
end;

// 鎖存在,是當前線程加的鎖
// 重入次數-1
local counter = redis.call('hincrby', KEYS[1], ARGV[3], -1);
// -1 后大于 0,說明這個線程持有這把鎖還有其他的任務需要執行
if (counter > 0) then
// 重新設置鎖的過期時間
redis.call('pexpire', KEYS[1], ARGV[2]);
return 0;
else
// -1 之后等于 0,現在可以刪除鎖了
redis.call('del', KEYS[1]);
// 刪除之后發布釋放鎖的消息
redis.call('publish', KEYS[2], ARGV[1]);
return 1;
end;

// 其他情況返回 nil
return nil;

看完它的使用后,我們發現真的使用起來像 JDK 中的 ReentrantLock 一樣絲滑。

RedLock

RedLock 的中文是直譯過來的,就叫紅鎖。紅鎖并非是一個工具,而是 Redis 官方提出的一種分布式鎖的算法。

我們知道如果采用單機部署模式,會存在單點問題,只要 redis 故障了,加鎖就不行了。如果采用 master-slave 模式,加鎖的時候只對一個節點加鎖,即便通過 sentinel 做了高可用,但是如果 master 節點故障了,發生主從切換,此時就會有可能出現鎖丟失的問題。

基于以上的考慮,其實 redis 的作者 Antirez 也考慮到這個問題,他提出了一個 RedLock 的算法。

我在這里畫了一個圖,圖中這五個實例都是獨自部署的,沒有主從關系,它們就是5個 master 節點。

file

通過以下步驟獲取一把鎖:

  • 獲取當前時間戳,單位是毫秒
  • 輪流嘗試在每個 master 節點上創建鎖,過期時間設置較短,一般就幾十毫秒
  • 嘗試在大多數節點上建立一個鎖,比如5個節點就要求是3個節點(n / 2 +1)
  • 客戶端計算建立好鎖的時間,如果建立鎖的時間小于超時時間,就算建立成功了
  • 要是鎖建立失敗了,那么就依次刪除這個鎖
  • 只要別人建立了一把分布式鎖,你就得不斷輪詢去嘗試獲取鎖

但是這樣的這種算法還是頗具爭議的,可能還會存在不少的問題,無法保證加鎖的過程一定正確。Martin Kleppmann 針對這個算法提出了質疑,接著 antirez 又回復了 Martin Kleppmann 的質疑。一個是很有資歷的分布式架構師,一個是 Redis 之父,這個就是著名的關于紅鎖的神仙打架事件。

最后 Redisson 官網上也給出了如何使用紅鎖 redlock,幾行代碼搞定,依然很絲滑,感興趣的小伙伴可以看下。

file

Zookeeper寫法

在介紹 zookeeper 實現分布式鎖的機制之前,先粗略介紹一下 zk 是什么東西:
zk 是一種提供配置管理、分布式協同以及命名的中心化服務。

它的模型是這樣的:包含一系列的節點,叫做znode,就好像文件系統一樣每個 znode 表示一個目錄,然后 znode 有一些特性,我們可以把它們分為四類:

  • 持久化節點(zk斷開節點還在)
  • 持久化順序節點(如果是第一個創建的子節點,那么生成的子節點為/lock/node-0000000000,下一個節點則為/lock/node-0000000001,依次類推)
  • 臨時節點(客戶端斷開后節點就刪除了)
  • 臨時順序節點

zookeeper分布式鎖恰恰應用了臨時順序節點,下面我們就用圖解的方式來看下是怎么實現的。

1、獲取鎖

首先,在 Zookeeper 當中創建一個持久節點 ParentLock。當第一個客戶端想要獲得鎖時,需要在ParentLock這個節點下面創建一個臨時順序節點 Lock1。

file

之后,Client1 查找 ParentLock 下面所有的臨時順序節點并排序,判斷自己所創建的節點 Lock1 是不是順序最靠前的一個。如果是第一個節點,則成功獲得鎖。

file

這時候,如果再有一個客戶端 Client2 前來獲取鎖,則在 ParentLock 下再創建一個臨時順序節點Lock2。

file

Client2 查找 ParentLock 下面所有的臨時順序節點并排序,判斷自己所創建的節點 Lock2 是不是順序最靠前的一個,結果發現節點 Lock2 并不是最小的。

于是,Client2 向排序僅比它靠前的節點 Lock1 注冊 Watcher,用于監聽 Lock1 節點是否存在。這意味著 Client2 搶鎖失敗,進入了等待狀態。

file

這時候,如果又有一個客戶端 Client3 前來獲取鎖,則在ParentLock下載再創建一個臨時順序節點Lock3。

file

Client3 查找 ParentLock 下面所有的臨時順序節點并排序,判斷自己所創建的節點 Lock3 是不是順序最靠前的一個,結果同樣發現節點 Lock3 并不是最小的。

于是,Client3 向排序僅比它靠前的節點 Lock2 注冊 Watcher,用于監聽 Lock2 節點是否存在。這意味著 Client3 同樣搶鎖失敗,進入了等待狀態。

file

這樣一來,Client1 得到了鎖,Client2 監聽了 Lock1,Client3 監聽了 Lock2。這恰恰形成了一個等待隊列,很像是 Java 當中 ReentrantLock 所依賴的 AQS(AbstractQueuedSynchronizer)。

2、釋放鎖

釋放鎖分為兩種情況:

1)任務完成,客戶端顯示釋放

當任務完成時,Client1 會顯示調用刪除節點 Lock1 的指令。

file

2)任務執行過程中,客戶端崩潰

獲得鎖的 Client1 在任務執行過程中,如果 Duang 的一聲崩潰,則會斷開與 Zookeeper 服務端的鏈接。根據臨時節點的特性,相關聯的節點 Lock1 會隨之自動刪除。

file

由于 Client2 一直監聽著 Lock1 的存在狀態,當 Lock1 節點被刪除,Client2 會立刻收到通知。這時候 Client2 會再次查詢 ParentLock 下面的所有節點,確認自己創建的節點 Lock2 是不是目前最小的節點。如果是最小,則 Client2 順理成章獲得了鎖。

file

同理,如果 Client2 也因為任務完成或者節點崩潰而刪除了節點 Lock2,那么 Client3 就會接到通知。

最終,Client3 成功得到了鎖。

file

3、Curator

在 Apache 的開源框架 Apache Curator 中,包含了對 Zookeeper 分布式鎖的實現。

https://github.com/apache/curator/

它的使用方式也很簡單,如下所示:

file

我們看了下依然絲滑,源碼我就不分析了。

總結

zookeeper 天生設計定位就是分布式協調,強一致性,鎖很健壯。如果獲取不到鎖,只需要添加一個監聽器就可以了,不用一直輪詢,性能消耗較小。缺點: 在高請求高并發下,系統瘋狂的加鎖釋放鎖,最后 zk 承受不住這么大的壓力可能會存在宕機的風險。

在這里簡單的提一下,zk 鎖性能比 redis 低的原因:

zk 中的角色分為 leader,flower,每次寫請求只能請求 leader,leader 會把寫請求廣播到所有 flower,如果 flower 都成功才會提交給 leader,其實這里相當于一個 2PC 的過程。在加鎖的時候是一個寫請求,當寫請求很多時,zk 會有很大的壓力,最后導致服務器響應很慢。

redis 鎖實現簡單,理解邏輯簡單,性能好,可以支撐高并發的獲取、釋放鎖操作。

缺點: Redis 容易單點故障,集群部署,并不是強一致性的,鎖的不夠健壯; key 的過期時間設置多少不明確,只能根據實際情況調整;需要自己不斷去嘗試獲取鎖,比較消耗性能。

最后不管 redis 還是 zookeeper,它們都應滿足分布式鎖的特性:

  • 具備可重入特性(已經獲得鎖的線程在執行的過程中不需要再次獲得鎖)
  • 異常或者超時自動刪除,避免死鎖
  • 互斥(在分布式環境下同一時刻只能被單個線程獲取)
  • 分布式環境下高性能、高可用、容錯機制

作者:jack_xu

鏈接:https://juejin.im/post/5ee227b46fb9a047a86237cf

文源網絡,僅供學習之用,如有侵權請聯系刪除。

我將面試題和答案都整理成了PDF文檔,還有一套學習資料,涵蓋Java虛擬機、spring框架、Java線程、數據結構、設計模式等等,但不僅限于此。

關注公眾號【java圈子】獲取資料,還有優質文章每日送達。

file
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,156評論 6 531
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,401評論 3 415
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,069評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,873評論 1 309
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,635評論 6 408
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,128評論 1 323
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,203評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,365評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,881評論 1 334
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,733評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,935評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,475評論 5 358
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,172評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,582評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,821評論 1 282
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,595評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,908評論 2 372