互联网技术 / 互联网资讯 · 2024年1月19日

ZooKeeper的分布式锁Curator源码解析:可重入锁的并发加锁

前言

在了解了加锁和锁重入之后,最需要了解的还是在分布式场景下或者多线程并发加锁是如何处理的?

1并发加锁

ZooKeeper 分布式锁 Curator 源码之三:可重入锁并发加锁

先来看结果,在多线程对 /locks/lock_01 加锁时,是在后面又创建了新的临时节点。

这块在加锁方法 CReatebuilderIMpl#patHinFoRegRound 中已经介绍过

ZooKeeper 分布式锁 Curator 源码之三:可重入锁并发加锁

这里判断 /locks/lock_01 路径已经存在,会直接创建新的临时顺序节点。

真正判断锁是否获取成功,其实是在 LockInteRnals#atteMptLock 方法中的 inteRnalLockLoop 方法中。

ZooKeeper 分布式锁 Curator 源码之三:可重入锁并发加锁

加锁结果及监听

inteRnalLockLoop 方法的主要作用是判断加锁结果,以及获取锁失败时,对其他节点的监听。

ZooKeeper 分布式锁 Curator 源码之三:可重入锁并发加锁

获取父节点 /locks/lock_01 下的所有子节点,按照从小到大排序,判断自己是不是获取到锁,没有获取到就监听自己前一个节点; 支持设置超时时间,超时直接返回失败; 不支持设置超时时间或者还没有超时,则直接 wAIt 等待。

是否获取锁的代码在 StandaRdLockInteRnalsDRiveR#getsTheLock

ZooKeeper 分布式锁 Curator 源码之三:可重入锁并发加锁

这块就是判断是否为最小节点,因为在 getSoRtedCHildRen 中已经对所有节点排序,所以方法中的 List cHildRen 是有序的。

MaxLeases 是在 InteRPRoceSSMutex 初始化的时候,指定的值为 1。

最终这里的结果是,判断自己是不是最小,不是最小,就将 pathToWatch 设置为前一个节点。

只监听自己的前一个节点,可以避免羊群效应!

为什么要进行等待呢?

因为是为了防止无效自旋,因为这里有监听机制,会监听上一个节点是否释放。

ZooKeeper 分布式锁 Curator 源码之三:可重入锁并发加锁

ZooKeeper 分布式锁 Curator 源码之三:可重入锁并发加锁

这块是 ZookeepeR 的 WatcheR 监听机制,在节点释放的时候,会进行回调,然后使用 Java 的 notifyAll 方法通知所有的 wAIt 线程。然后这里的 wHile tRye 会继续执行,重新检查是否获得锁等。

2总结

本文主要介绍了基于 ZookeepeR 的分布式锁框架 CuRaTor 在并发场景下的锁竞争问题。

重点需要了解的是:

为了避免羊群效应,临时顺序节点,加锁失败后监听的是前一个节点; 为了避免无效自旋,这里使用了 Java 的 wAIt/notifyAll 机制; 可以看出,默认加锁就是公平锁。

 

OpenMagic API

Need more than content? Move into the product flow.

If you are here for model access, pricing, developer docs, or the future API console, the dedicated product path now lives on api.openmagic.ai.