Redisson 分布式锁源码——公平锁加锁

 前言

默认的加锁逻辑是非公平的。程序员

在加锁失败时,线程会进入 while 循环,一直尝试得到锁,这时候是多线程进行竞争。就是说谁抢到就是谁的。面试

Redisson 提供了 公平锁 机制,使用方式以下:redis

RLock fairLock = redisson.getFairLock("anyLock");
// 最多见的使用方法
fairLock.lock();

下面一块儿看下公平锁是如何实现的?数据结构

公平锁

直接定位到源码方法:RedissonFairLock#tryLockInnerAsync多线程

好家伙,这一大块代码,我截图也截不完,我们直接分析 lua 脚本。ide

PS:虽然咱不懂 lua,可是这一堆堆的 if else 我们大概仍是能看懂的。lua

由于 debug 发现 command == RedisCommands.EVAL_LONG,因此直接看下面一部分。spa

6ba8af6524a793111523c08d34b4c8c6.jpeg

这么长,连呼好几声好家伙!线程

先来看看参数都有啥?debug

  1. KEYS[1]:加锁的名字,anyLock
  2. KEYS[2]:加锁等待队列,redisson_lock_queue:{anyLock}
  3. KEYS[3]:等待队列中线程锁时间的 set 集合,redisson_lock_timeout:{anyLock},是按照锁的时间戳存放到集合中的;
  4. ARGV[1]:锁超时时间 30000;
  5. ARGV[2]:UUID:ThreadId 组合 a3da2c83-b084-425c-a70f-5d9a08b37f31:1
  6. ARGV[3]:threadWaitTime 默认 300000;
  7. ARGV[4]:currentTime 当前时间戳。

3a4066fbc137cfc804d17359b2476a95.png

加锁队列和集合是含有大括号的字符串。{XXXX} 是指这个 key 仅使用 XXXX 用来计算 slot 的位置。

Lua 脚本分析

上面的 lua 脚本是分为几块的,我们分别从不一样的角度看下上面代码的执行。

首次加锁(Thread1)

fa2475fde103aca06b77168347b5e9fb.jpeg

第一部分,由于是首次加锁,因此等待队列为空,直接 跳出循环。这一部分执行结束。

af7667e701ab29c04ef1da6f55c086ab.jpeg

第二部分:

  1. 当锁不存在,等待队列为空或队首是当前线程,两个条件都知足时,进入内部逻辑;
  2. 从等待队列和超时集合中删除当前线程,这时候等待队列和超时集合都是空的,不须要任何操做;
  3. 减小队列中全部等待线程的超时时间,也不须要任何操做;
  4. 加锁并设置超时时间。

执行完这里就 return 了。因此后面几部分就暂时不看了。

至关于下面两个命令(整个 lua 脚本都是原子的!):

> hset anyLock a3da2c83-b084-425c-a70f-5d9a08b37f31:1 1
> pexpire anyLock 30000

021bc4869ca202129ca2e5edce6469f2.png

Thread2 加锁

当 Thread1 加锁完成以后,此时 Thread2 来加锁。

Thread2 能够是本实例其余线程,也能够是其余实例的线程。

7c16ee6f99723db2edc20615fdd29cae.jpeg

第一部分,虽然锁被 Thread1 占用了,可是等待队列是空的,直接跳出循环。

739607cca316e7ab557f726f0e0b2b57.jpeg

第二部分,锁存在,直接跳过。

cadb63572cf819a024879cea4938a649.jpeg

第三部分,线程是否持锁,没有持锁,直接跳过。

第四部分,线程是否在等待队列中,Thread2 才来加锁,不在里面,直接跳过。

f872a97ff6da1ad242b8c60f2a3bacb8.jpeg

Thread2 最后会来到这里:

  1. 从线程等待队列 redisson_lock_queue:{anyLock} 中获取最后一个线程;
  2. 由于等待队列是空的,因此直接获取当前锁的剩余时间 ttl anyLock
  3. 组装超时时间 timeout = ttl + 300000 + 当前时间戳,这个 300000 是默认 60000*5
  4. 使用 zadd 将 Thread2 放到等待线程有序集合,而后使用 rpush 将 Thread2 再放到等待队列中。

zadd KEYS[3] timeout ARGV[2]

这里使用 zadd 命令分别放置的是,redisson_lock_timeout:{anyLock},超时时间戳(1624612689520),线程(UUID2:Thread2)。

其中超时时间戳当分数,用来在有序集合中排序,表示加锁的顺序。

dca68c31dae1ff3f76d69495abcc2d1a.png

Thread3 加锁

a5237a14fd30907924a9e457b8b4b40f.png

Thread1 占有了锁,Thread2 在等待,此时线程 3 来了。

c78d07e84051502061ce97bc41f85a05.jpeg

获取 firstThreadId2 此时队列是有线程的是 UUID2:Thread2。

判断 firstThreadId2 的分数(超时时间戳)是否是小于当前时间戳:

  1. 小于等于则说明超时了,移除 firstThreadId2;
  2. 大于,则会进入后续判断。

第2、3、四部分都不知足条件。

0b114442b97f7a23be4f285363081e0a.jpeg

Thread3 最后也会来到这里:

  1. 从线程等待队列 redisson_lock_queue:{anyLock} 中获取最后一个线程;
  2. 最后一个线程存在,且不是本身,则 ttl = lastThreadId 超时时间戳 - 当前时间戳,就是看最后一个线程还有多久超时;
  3. 组装超时时间 timeout = ttl + 300000 + 当前时间戳,这个 300000 是默认 60000*5,在最后一个线程的超时时间上加上 300000 以及当前时间戳,就是 Thread3 的超时时间戳。
  4. 使用 zadd 将 Thread3 放到等待线程有序集合,而后使用 rpush 将 Thread3 再放到等待队列中。

928ae2637bcfa1f38b553ffa6b1fed97.png

总结

本文主要总结了公平锁的加锁逻辑,这涉及到比较多的 Redis 操做,作一下简要总结:

  1. Redis Hash 数据结构:存放当前锁,Redis Key 就是锁,Hash 的 field 是加锁线程,Hash 的 value 是 重入次数;
  2. Redis List 数据结构:充当线程等待队列,新的等待线程会使用 rpush 命令放在队列右边;
  3. Redis sorted set 有序集合数据结构:存放等待线程的顺序,分数 score 用来是等待线程的超时时间戳。

deae57129c833505afebc73f598e3fde.png

须要理解的就是这里会额外添加一个等待队列,以及有序集合。

对照着 Java 公平锁源码阅读,理解起来效果更好。

最后

最近我整理了整套《JAVA核心知识点总结》,说实话 ,做为一名Java程序员,不论你需不须要面试都应该好好看下这份资料。拿到手老是不亏的~个人很多粉丝也所以拿到腾讯字节快手等公司的Offer

Java进阶之路群,找管理员获取哦-!

99b8a0b9c6f3f0b4312c6b27e245f3b0.png

相关文章
相关标签/搜索