ZooKeeper 分散式鎖 Curator 原始碼 02:可重入鎖重複加鎖和鎖釋放

程式設計師小航 發表於 2021-07-21
ZooKeeper

前言

加鎖邏輯已經介紹完畢,那當一個執行緒重複加鎖是如何處理的呢?

鎖重入

在上一小節中,可以看到加鎖的過程,再回頭看 internalLock 這個方法。

ZooKeeper 分散式鎖 Curator 原始碼 02:可重入鎖重複加鎖和鎖釋放

加鎖成功之後,將當前執行緒放到 threadData 中,threadData 是 ConcurrentMap<Thread, LockData> 型別的,不用擔心併發問題。

假如鎖重入了,直接就會在上一部分 lockData != null 被攔下,然後執行 lockData.lockCount.incrementAndGet();

對 lockCount 自增,代表了鎖重入。

這裡發現了吧!Curator 的鎖重入是在 Java 程式碼中實現的

鎖釋放

當鎖需要釋放的時候,只需要呼叫 lock.release() 進行釋放即可,具體是如何釋放的呢?

ZooKeeper 分散式鎖 Curator 原始碼 02:可重入鎖重複加鎖和鎖釋放

主要分為兩部分:

  1. 遞減 threadData 中當前執行緒的加鎖次數;
  2. 加鎖次數大於 0,說明還剩餘重入次數,直接返回;
  3. 加鎖次數等於 0,則 releaseLock 釋放鎖,並刪除 threadData 中當前執行緒 key。

releaseLock 方法中就沒有多少複雜邏輯了,就是移除監聽器,刪除臨時順序節點。 也就是 /locks/lock_01/_c_e855d232-c636-4241-bf8e-f047939a5833-lock-0000000001

總結

ZooKeeper 的 InterProcessMutex 鎖是通過 Java 程式碼中維護了一個 lockCount 來判斷是否重入的。

ZooKeeper 分散式鎖 Curator 原始碼 02:可重入鎖重複加鎖和鎖釋放

相關推薦