[[410320]]

前言

加鎖邏輯已經(jīng)介紹完畢，那當(dāng)一個(gè)線程重復(fù)加鎖是如何處理的呢?

1.鎖重入

在上一小節(jié)中，可以看到加鎖的過(guò)程，再回頭看 internalLock 這個(gè)方法。

加鎖成功之后，將當(dāng)前線程放到 threadData 中，threadData 是 ConcurrentMap

假如鎖重入了，直接就會(huì)在上一部分 lockData != null 被攔下，然后執(zhí)行 lockData.lockCount.incrementAndGet();。

對(duì) lockCount 自增，代表了鎖重入。

這里發(fā)現(xiàn)了吧!Curator 的鎖重入是在 Java 代碼中實(shí)現(xiàn)的。

2.鎖釋放

當(dāng)鎖需要釋放的時(shí)候，只需要調(diào)用 lock.release() 進(jìn)行釋放即可，具體是如何釋放的呢?

主要分為兩部分：

1. 遞減 threadData 中當(dāng)前線程的加鎖次數(shù);
2. 加鎖次數(shù)大于 0，說(shuō)明還剩余重入次數(shù)，直接返回;
3. 加鎖次數(shù)等于 0，則 releaseLock 釋放鎖，并刪除 threadData 中當(dāng)前線程 key。

releaseLock 方法中就沒(méi)有多少?gòu)?fù)雜邏輯了，就是移除監(jiān)聽(tīng)器，刪除臨時(shí)順序節(jié)點(diǎn)。也就是 /locks/lock_01/_c_e855d232-c636-4241-bf8e-f047939a5833-lock-0000000001。

3.總結(jié)

ZooKeeper 的 InterProcessMutex 鎖是通過(guò) Java 代碼中維護(hù)了一個(gè) lockCount 來(lái)判斷是否重入的。

本文轉(zhuǎn)載自微信公眾號(hào)「程序員小航」，可以通過(guò)以下二維碼關(guān)注。轉(zhuǎn)載本文請(qǐng)聯(lián)系程序員小航公眾號(hào)。