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

作者：程序员小航 2021-07-06 08:37:29

开发

前端

分布式

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

[[409489]]

前言

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

1、使用读写锁

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

2、源码

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

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

读锁源码

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

参数列表：

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

首次加读锁

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

读锁重入

如果是重入的情况下：

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

读读支持

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

写读互斥

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

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

通过上面可以看出，在读锁的时候：

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

写锁源码

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

参数列表：

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

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

• 判断锁的模式，是写锁
• 锁不存在直接创建
• 锁存在，再判断是不是自己，是自己则重入

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

3、总结

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

在读锁的时候：

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

在写锁的时候：

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

至于看门狗，这些都和之前的一样，就不额外介绍了。
本文转载自微信公众号「程序员小航」，可以通过以下二维码关注。转载本文请联系程序员小航公众号。