Redisson 分布式锁源码 10:读写锁

903 阅读4分钟

前言

Redisson 还支持可重入读写锁,允许在分布式场景下,同时有多个读锁和一个写锁处于加锁状态。

使用读写锁

Redisson 读写锁实现了 JUC 下的 ReadWriteLock,使用方式基本相同。

源码

加锁源码基本和之前的可重入锁加锁无区别,唯一的差异就是在 Lua 脚本这里。

所以下面着重分析 Lua 脚本。

读锁源码

源码地址:org.redisson.RedissonReadLock#tryLockInnerAsync

参数列表:

  1. KEYS[1]:锁名字 anyRWLock
  2. KEYS[2]:锁超时 key {锁名字}:UUID:ThreadId:rwlock_timeout 组成的字符串,{anyRWLock}:e70b1307-9ddd-43de-ac9d-9c42b5c99a0d:1:rwlock_timeout
  3. ARGV[1]:锁时间,默认 30s
  4. ARGV[2]:当前线程,UUID:ThreadId 组成的字符串,e70b1307-9ddd-43de-ac9d-9c42b5c99a0d:1
  5. ARGV[3]:写锁名字,getWriteLockName(threadId) 写锁名字,UUID:ThreadId:write 组成的字符串, e70b1307-9ddd-43de-ac9d-9c42b5c99a0d:1:write

首次加读锁

  1. 锁不存在,直接走第一部分
  2. 设置锁 anyRWLock 的 mode 是 read,表示这是个读锁
  3. 设置锁 anyRWLock 的 e70b1307-9ddd-43de-ac9d-9c42b5c99a0d:1(当前线程)值为 1
  4. 设置锁 {anyRWLock}:e70b1307-9ddd-43de-ac9d-9c42b5c99a0d:1:rwlock_timeout:1 的值是 1,表示当前线程,当前重入的超时时间
  5. 设置两个 RedisKey 的过期时间
读锁重入

如果是重入的情况下:

  1. 锁存在,且是读锁,直接进入第二部分
  2. 对锁 anyRWLock 的 e70b1307-9ddd-43de-ac9d-9c42b5c99a0d:1(当前线程)值自增 1 表是重入
  3. 再创建 {anyRWLock}:e70b1307-9ddd-43de-ac9d-9c42b5c99a0d:1:rwlock_timeout:2 表示第二次加锁的超时时间
读读支持

  1. 锁存在,进入第二部分
  2. 对当前线程的值自增 1,这里已经是第二个线程了
  3. 设置第二个线程 {anyRWLock}:7c390320-78e3-497f-a3d8-ac34a44d0464:48:rwlock_timeout:1 的超时时间
写读互斥

已经加了读锁了,此时写锁进来,不满足第一部分,也不满足第二部分,所以直接返回当前锁的剩余时间。

然后再 Java 代码中进行 while (true) 自旋等待。

通过上面可以看出,在读锁的时候:

  1. 锁 anyRWLock 是哈希表结构的
  2. 加锁时,会对哈希表设置 mode 字段来表示这个锁是读锁还是写锁,mode = read 表示读锁
  3. 加锁时,会对哈希表设置当前线程 anyRWLock 的 UUID:ThreadId 字段,值表示重入次数
  4. 每次加锁,会额外维护一个 key 表示这次锁的超时时间,这个 key 的结构是 {锁名字}:UUID:ThreadId:rwlock_timeout:重入次数

写锁源码

源码地址:org.redisson.RedissonWriteLock#tryLockInnerAsync

参数列表:

  1. KEYS[1]:当前锁 anyRWLock
  2. ARGV[1]:锁时间,默认 30s
  3. ARGV[2]:写锁名字,UUID:ThreadId:write 组成的字符串,c69a9ed4-5c30-4952-814e-c0b94ad03a7f:1:write

写锁源码相对比较好理解:

  1. 判断锁的模式,是写锁
  2. 锁不存在直接创建
  3. 锁存在,再判断是不是自己,是自己则重入

这么下来,可以看出直接满足,写写互斥,读写互斥,当前线程又可以重入。

总结

到这里基本上读写锁就看完了,读锁实现的稍微复杂一些,写锁简单明了。

在读锁的时候:

  1. 锁 anyRWLock 是哈希表结构
  2. 加锁时,会对哈希表设置 mode 字段来表示这个锁是读锁还是写锁,mode = read 表示读锁
  3. 加锁时,会对哈希表设置当前线程 anyRWLock 的 UUID:ThreadId 字段,值表示重入次数
  4. 每次加锁,会额外维护一个 key 表示这次锁的超时时间,这个 key 的结构是 {锁名字}:UUID:ThreadId:rwlock_timeout:重入次数

在写锁的时候:

  1. 锁 anyRWLock 是哈希表结构
  2. 加锁时,会对哈希表设置 mode 字段来表示这个锁是读锁还是写锁,mode = write 表示写锁
  3. 在 anyRWLock 中再额外维护一个字段 UUID:ThreadId:write 表示重入次数

至于看门狗,这些都和之前的一样,就不额外介绍了。

相关推荐