七叶笔记 » 数据库 » redis分布式锁之可重入锁的实现代码

redis分布式锁之可重入锁的实现代码

3.2、执行结果

16:35:58.051 [main] INFO com.test.ReentrantLockDemo - --------lock()执行后,getState()的值:1 lock.isLocked():true 16:35:58.055 [main] INFO com.test.ReentrantLockDemo - --------递归1次-------- 16:35:58.055 [main] INFO com.test.ReentrantLockDemo - --------lock()执行后,getState()的值:2 lock.isLocked():true 16:35:58.055 [main] INFO com.test.ReentrantLockDemo - --------递归2次-------- 16:35:58.055 [main] INFO com.test.ReentrantLockDemo - --------lock()执行后,getState()的值:3 lock.isLocked():true 16:35:58.055 [main] INFO com.test.ReentrantLockDemo - --------递归3次-------- 16:35:58.055 [main] INFO com.test.ReentrantLockDemo - --------unlock()执行后,getState()的值:2 lock.isLocked():true 16:35:58.055 [main] INFO com.test.ReentrantLockDemo - --------unlock()执行后,getState()的值:1 lock.isLocked():true 16:35:58.055 [main] INFO com.test.ReentrantLockDemo - --------unlock()执行后,getState()的值:0 lock.isLocked():false 16:35:58.055 [main] INFO com.test.ReentrantLockDemo - 执行完doSomething方法 是否还持有锁:false

3.3、 从上面栗子可以看出ReentrantLock是可重入锁,那么他是如何实现的了,我们看下源码就知道了

ReentrantLock的加锁流程是: 1,先判断是否有线程持有锁,没有加锁进行加锁 2、如果加锁成功,则设置持有锁的线程是当前线程 3、如果有线程持有了锁,则再去判断,是否是当前线程持有了锁 4、如果是当前线程持有锁,则加锁数量(state)+1

看ReentrantLock的解锁代码我们知道,每次释放锁的时候都对state减1, 当c值等于0的时候,说明锁重入次数也为0了, 最终设置当前持有锁的线程为null,state也设置为0,锁就释放了。

4、那么redis要怎么实现可重入的操作了?

看ReentrantLock的源码我们知道,它是加锁成功了,记录了当前持有锁的线程,并通过一个int类型的数字,来记录了加锁次数。

我们知道ReentrantLock的实现原理了,那么redis只要下面两个问题解决,就能实现重入锁了: 1、怎么保存当前持有的线程 2、加锁次数(重入了多少次),怎么记录维护

4.1、第一个问题:怎么保存当前持有的线程

1.上一篇文章我们用的是redis 的set命令存的是string类型,他能保存当前持有的线程吗? valus值我们可以保存当前线程的id来解决。 2. 但是集群环境下我们线程id可能是重复了那怎么解决? 项目在启动的生成一个全局进程id,使用进程id+线程id 那就是唯一的了

4.2、第二个问题:加锁次数(重入了多少次),怎么记录维护

他能记录下来加锁次数吗? 如果valus值存的格式是:系进程id+线程id+加锁次数,那可以实现

存没问题了,但是重入次数要怎么维护了, 它肯定要保证原子性的,能解决吗? 好像用java代码或者lua脚本都没法解决,因为都是实现都需要两步来维护这个重入次数的

第一步:先获取到valus值,把取到加锁次数+1 第二部:把新的值再设置进去 在执行第二步操作之前,如果这个key失效了(设置持有锁超时了),如果还能再设置进去,就会有并发问题了

5、我们已经知道SET是不支持重入锁的,但我们需要重入锁,怎么办呢?

目前对于redis的重入锁业界还是有很多解决方案的,最流行的就是采用Redisson。

6、什么是 Redisson?

Redisson是Redis官方推荐的Java版的Redis客户端。 它基于Java实用工具包中常用接口,为使用者提供了一系列具有分布式特性的常用工具类。 它在网络通信上是基于NIO的Netty框架,保证网络通信的高性能。 在分布式锁的功能上,它提供了一系列的分布式锁;如:可重入锁(Reentrant Lock)、公平锁(Fair Lock、联锁(MultiLock)、 红锁(RedLock)、 读写锁(ReadWriteLock)等等。

Redisson github地址

7、Redisson的分布锁如何使用

引入依赖包

代码

执行结果

2021-05-23 22:49:01.322 INFO 69041 --- [nio-9090-exec-1] org.redisson.Version : Redisson 3.15.5 2021-05-23 22:49:01.363 INFO 69041 --- [sson-netty-5-22] o.r.c.pool.MasterConnectionPool : 10 connections initialized for /127.0.0.1:6379 2021-05-23 22:49:01.363 INFO 69041 --- [sson-netty-5-23] o.r.c.pool.MasterPubSubConnectionPool : 1 connections initialized for /127.0.0.1:6379 2021-05-23 22:49:01.367 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------------start--------------- 2021-05-23 22:49:01.435 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------lock()执行后,getState()的值:1 lock.isLocked():true 2021-05-23 22:49:01.436 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------递归1次-------- 2021-05-23 22:49:01.442 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------lock()执行后,getState()的值:2 lock.isLocked():true 2021-05-23 22:49:01.442 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------递归2次-------- 2021-05-23 22:49:01.448 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------lock()执行后,getState()的值:3 lock.isLocked():true 2021-05-23 22:49:01.448 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------递归3次-------- 2021-05-23 22:49:01.456 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------unlock()执行后,getState()的值:2 lock.isLocked():true 2021-05-23 22:49:01.461 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------unlock()执行后,getState()的值:1 lock.isLocked():true 2021-05-23 22:49:01.465 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------unlock()执行后,getState()的值:0 lock.isLocked():false 2021-05-23 22:49:01.467 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : 执行完doSomething方法 是否还持有锁:false 2021-05-23 22:49:01.467 INFO 69041 --- [nio-9090-exec-1] com.test.ReentrantLockDemo1 : --------------end---------------

看控制台打印能清楚知道Redisson是支持可重入锁了。

8、那么Redisson是如何实现的了?

我们跟一下lock.lock()的代码,发现它最终调用的是org.redisson.RedissonLock#tryLockInnerAsync的方法,具体如下:

8.1、上面的代码,用到的redis命令先梳理一下

exists 查询一个key是否存在

EXISTS key [key ...] 返回值 如下的整数结果 1 如果key存在 0 如果key不存在

hincrby :将hash中指定域的值增加给定的数字

pexpire:设置key的有效时间以毫秒为单位

hexists:判断field是否存在于hash中

pttl:获取key的有效毫秒数

8.2、看lua脚本传入的参数我们知道:

KEYS[1] = key的值 ARGV[1]) = 持有锁的时间 ARGV[2] = getLockName(threadId) 下面id就算系统在启动的时候会全局生成的uuid 来作为当前进程的id,加上线程id就是getLockName(threadId)了,可以理解为:进程ID+系统ID = ARGV[2]

8.3、代码截图

从截图上可以看到,它是使用lua脚本来保证多个命令执行的原子性,使用了hash来实现了分布式锁 现在我们来看下lua脚本的加锁流程

8.4、第一个if判断

204行:它是先判断了当前key是否存在,从EXISTS命令我们知道返回值是0说明key不存在,说明没有加锁 205行:hincrby命令是对 ARGV[2] = 进程ID+系统ID 进行原子自增加1 206行:是对整个hash设置过期期间

8.5、下面来看第二个if判断

209行:判断field是否存在于hash中,如果存在返回1,返回1说明是当前进程+当前线程ID 之前已经获得到锁了 210行:hincrby命令是对 ARGV[2] = 进程ID+系统ID 进行原子自增加1,说明重入次数加1了 211行:再对整个hash设置过期期间

8.6、下图是redis可视化工具看到是如何在hash存储的结构

Redisson的整个加锁流程跟ReentrantLock的加锁逻辑基本相同

8.7、解锁代码位于 org.redisson.RedissonLock#unlockInnerAsync,如下:

看这个解锁的Lua脚本,流程跟Reentrantlock的解锁逻辑也基本相同没啥好说的了。

以上就是redis分布式锁-可重入锁的详细内容,更多关于redis分布式锁的资料请关注七叶笔记其它相关文章!

相关文章