前言
加鎖邏輯已經介紹完畢,那當一個線程重復加鎖是如何處理的呢?
鎖重入
在上一小節中,可以看到加鎖的過程,再回頭看 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 來判斷是否重入的。