ZooKeeper 分布式锁 Curator 源码 02:可重入锁重复加锁和锁释放
前言
加锁逻辑已经介绍完毕,那当一个线程重复加锁是如何处理的呢?
锁重入
在上一小节中,可以看到加锁的过程,再回头看 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 来判断是否重入的。