ZooKeeper 分布式鎖 Curator 源碼之三:可重入鎖并發(fā)加鎖
前言
在了解了加鎖和鎖重入之后,最需要了解的還是在分布式場(chǎng)景下或者多線程并發(fā)加鎖是如何處理的?
1并發(fā)加鎖
先來(lái)看結(jié)果,在多線程對(duì) /locks/lock_01 加鎖時(shí),是在后面又創(chuàng)建了新的臨時(shí)節(jié)點(diǎn)。
這塊在加鎖方法 CreateBuilderImpl#pathInForeground 中已經(jīng)介紹過(guò)
這里判斷 /locks/lock_01 路徑已經(jīng)存在,會(huì)直接創(chuàng)建新的臨時(shí)順序節(jié)點(diǎn)。
真正判斷鎖是否獲取成功,其實(shí)是在 LockInternals#attemptLock 方法中的 internalLockLoop 方法中。
加鎖結(jié)果及監(jiān)聽(tīng)
internalLockLoop 方法的主要作用是判斷加鎖結(jié)果,以及獲取鎖失敗時(shí),對(duì)其他節(jié)點(diǎn)的監(jiān)聽(tīng)。
- 獲取父節(jié)點(diǎn) /locks/lock_01 下的所有子節(jié)點(diǎn),按照從小到大排序,判斷自己是不是獲取到鎖,沒(méi)有獲取到就監(jiān)聽(tīng)自己前一個(gè)節(jié)點(diǎn);
- 支持設(shè)置超時(shí)時(shí)間,超時(shí)直接返回失敗;
- 不支持設(shè)置超時(shí)時(shí)間或者還沒(méi)有超時(shí),則直接 wait 等待。
是否獲取鎖的代碼在 StandardLockInternalsDriver#getsTheLock
這塊就是判斷是否為最小節(jié)點(diǎn),因?yàn)樵?getSortedChildren 中已經(jīng)對(duì)所有節(jié)點(diǎn)排序,所以方法中的 List
maxLeases 是在 InterProcessMutex 初始化的時(shí)候,指定的值為 1。
最終這里的結(jié)果是,判斷自己是不是最小,不是最小,就將 pathToWatch 設(shè)置為前一個(gè)節(jié)點(diǎn)。
只監(jiān)聽(tīng)自己的前一個(gè)節(jié)點(diǎn),可以避免羊群效應(yīng)!
為什么要進(jìn)行等待呢?
因?yàn)槭菫榱朔乐篃o(wú)效自旋,因?yàn)檫@里有監(jiān)聽(tīng)機(jī)制,會(huì)監(jiān)聽(tīng)上一個(gè)節(jié)點(diǎn)是否釋放。
這塊是 ZooKeeper 的 Watcher 監(jiān)聽(tīng)機(jī)制,在節(jié)點(diǎn)釋放的時(shí)候,會(huì)進(jìn)行回調(diào),然后使用 Java 的 notifyAll 方法通知所有的 wait 線程。然后這里的 while trye 會(huì)繼續(xù)執(zhí)行,重新檢查是否獲得鎖等。
2總結(jié)
本文主要介紹了基于 ZooKeeper 的分布式鎖框架 Curator 在并發(fā)場(chǎng)景下的鎖競(jìng)爭(zhēng)問(wèn)題。
重點(diǎn)需要了解的是:
- 為了避免羊群效應(yīng),臨時(shí)順序節(jié)點(diǎn),加鎖失敗后監(jiān)聽(tīng)的是前一個(gè)節(jié)點(diǎn);
- 為了避免無(wú)效自旋,這里使用了 Java 的 wait/notifyAll 機(jī)制;
- 可以看出,默認(rèn)加鎖就是公平鎖。
本文轉(zhuǎn)載自微信公眾號(hào)「程序員小航」,可以通過(guò)以下二維碼關(guān)注。轉(zhuǎn)載本文請(qǐng)聯(lián)系程序員小航公眾號(hào)。