前言
加鎖邏輯已經介紹完畢,那當一個執行緒重複加鎖是如何處理的呢?
鎖重入
在上一小節中,可以看到加鎖的過程,再回頭看 internalLock
這個方法。
加鎖成功之後,將當前執行緒放到 threadData
中,threadData 是 ConcurrentMap<Thread, LockData>
型別的,不用擔心併發問題。
假如鎖重入了,直接就會在上一部分 lockData != null
被攔下,然後執行 lockData.lockCount.incrementAndGet();
。
對 lockCount 自增,代表了鎖重入。
這裡發現了吧!Curator 的鎖重入是在 Java 程式碼中實現的。
鎖釋放
當鎖需要釋放的時候,只需要呼叫 lock.release() 進行釋放即可,具體是如何釋放的呢?
主要分為兩部分:
- 遞減 threadData 中當前執行緒的加鎖次數;
- 加鎖次數大於 0,說明還剩餘重入次數,直接返回;
- 加鎖次數等於 0,則 releaseLock 釋放鎖,並刪除 threadData 中當前執行緒 key。
releaseLock 方法中就沒有多少複雜邏輯了,就是移除監聽器,刪除臨時順序節點。 也就是 /locks/lock_01/_c_e855d232-c636-4241-bf8e-f047939a5833-lock-0000000001
。
總結
ZooKeeper 的 InterProcessMutex 鎖是通過 Java 程式碼中維護了一個 lockCount 來判斷是否重入的。