ZooKeeper 分布式鎖 Curator 源碼之二:可重入鎖重復加鎖和鎖釋放
前言
加鎖邏輯已經介紹完畢,那當一個線程重復加鎖是如何處理的呢?
1.鎖重入
在上一小節中,可以看到加鎖的過程,再回頭看 internalLock 這個方法。
加鎖成功之后,將當前線程放到 threadData 中,threadData 是 ConcurrentMap
假如鎖重入了,直接就會在上一部分 lockData != null 被攔下,然后執行 lockData.lockCount.incrementAndGet();。
對 lockCount 自增,代表了鎖重入。
這里發現了吧!Curator 的鎖重入是在 Java 代碼中實現的。
2.鎖釋放
當鎖需要釋放的時候,只需要調用 lock.release() 進行釋放即可,具體是如何釋放的呢?
主要分為兩部分:
- 遞減 threadData 中當前線程的加鎖次數;
- 加鎖次數大于 0,說明還剩余重入次數,直接返回;
- 加鎖次數等于 0,則 releaseLock 釋放鎖,并刪除 threadData 中當前線程 key。
releaseLock 方法中就沒有多少復雜邏輯了,就是移除監聽器,刪除臨時順序節點。也就是 /locks/lock_01/_c_e855d232-c636-4241-bf8e-f047939a5833-lock-0000000001。
3.總結
ZooKeeper 的 InterProcessMutex 鎖是通過 Java 代碼中維護了一個 lockCount 來判斷是否重入的。
本文轉載自微信公眾號「程序員小航」,可以通過以下二維碼關注。轉載本文請聯系程序員小航公眾號。