前言

加锁逻辑已经介绍完毕，那当一个线程重复加锁是如何处理的呢？

锁重入

在上一小节中，可以看到加锁的过程，再回头看 internalLock 这个方法。
2-可重入锁重入 - 图1
加锁成功之后，将当前线程放到 threadData 中，threadData 是 ConcurrentMap<Thread, LockData> 类型的，不用担心并发问题。
假如锁重入了，直接就会在上一部分 lockData != null 被拦下，然后执行 lockData.lockCount.incrementAndGet();。
对 lockCount 自增，代表了锁重入。
这里发现了吧！Curator 的锁重入是在 Java 代码中实现的。

锁释放

当锁需要释放的时候，只需要调用 lock.release() 进行释放即可，具体是如何释放的呢？
2-可重入锁重入 - 图2
主要分为两部分：

  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 来判断是否重入的。
2-可重入锁重入 - 图3

相关推荐