背景介绍
前段时间,在写公司的一个项目的时候,用到了分布式锁,一个同事告诉我说,分布式锁解锁在高并发的时候会报错。
下面看下模拟代码:
这里锁的时间是 5 秒,而业务执行的时间是 20 秒。这里模拟的是锁的时间少于业务执行的时间。
第二次执行的时候,就会报错,如下:
java.lang.IllegalMonitorStateException: attempt to unlock lock, not locked by current thread by node id: 660a38bb-c50b-4117-8ee2-67da7b4303c6 thread-id: 62
错误分析
通过这个错误信息,也知道该如何解决这个问题,我们只需要判断是当前线程再去解锁,就不会报错的。
可是为什么会报错呢?
所以,我们需要先去搞清楚具体的执行流程。
最开始,我以为,如果被锁住,运行到 ① 就会被返回,后面经过测试,实际上是会走到第 ② 步,尝试获取不到锁,就会返回,在返回之前呢,会执行 finally
的代码,因为 redisson 对锁有续租的功能,所以,这时候锁还是锁住的,解锁就会报错,也就是第 ③ 步。
实际上,我们的想法的,如果业务执行出错,我们在 finally 进行解锁,以防止程序死锁。
显然这样写代码,不是我们所期望的,并且代码也有问题。
优化代码
代码语言:java复制@RequestMapping("/try-lock")
public String tryLock() {
RLock rLock = redissonClient.getLock("demo-spring-boot-redisson:try-lock");
if (Objects.isNull(rLock)) {
return "lock exception";
}
boolean tryLock;
try {
tryLock = rLock.tryLock(3, 60, TimeUnit.SECONDS);
} catch (InterruptedException e) {
return "get lock exception";
}
if (!tryLock) {
return "get lock failed";
}
try {
TimeUnit.SECONDS.sleep(20);
return "success";
} catch (Exception e) {
return "business exception";
} finally {
if (rLock.isLocked() && rLock.isHeldByCurrentThread()) {
rLock.unlock();
}
}
}
以上,便是优化后的代码,我们来一起分析一下。
分布式加锁主要分为三步。
第一步,主要是获取 RLock
对象,并且我们对它做了判空。
RLock rLock = redissonClient.getLock("demo-spring-boot-redisson:try-lock");
if (Objects.isNull(rLock)) {
return "lock exception";
}
第二步,尝试加锁,加锁失败,返回加锁失败。
代码语言:java复制boolean tryLock;
try {
tryLock = rLock.tryLock(3, 60, TimeUnit.SECONDS);
} catch (InterruptedException e) {
return "get lock exception";
}
if (!tryLock) {
return "get lock failed";
}
这里我们用的是 tryLock
,第一个参数 waitTime
,意思是等待 5 秒,如果还没获取到,就不再等待。第二个参数是 leaseTime
,意思是锁的释放时间。
第三步,就是我们业务代码。
代码语言:java复制try {
TimeUnit.SECONDS.sleep(20);
return "success";
} catch (Exception e) {
return "business exception";
} finally {
if (rLock.isLocked() && rLock.isHeldByCurrentThread()) {
rLock.unlock();
}
}
在 finally 里,我们做锁的释放操作,在释放之前,我们对锁的状态和是否是当前线程做了判断。
OK,如果你在实际的业务中如果遇到什么问题,欢迎留言探讨。
引申分析
- lock 和 tryLock 区别?
简单来说,lock 会一直阻塞,而 tryLock 加锁失败,会返回 false。
- 如果锁的时间少于业务的时间,会怎么样?
通过上面的分析,我们知道 tryLock 会加锁失败,而 lock,在锁到释放时间后,即便业务没有执行完,也会继续执行,并且不会报错。
代码语言:java复制@RequestMapping("/lock")
public String lock() {
RLock rLock = redissonClient.getLock("demo-spring-boot-redisson:lock");
if (Objects.isNull(rLock)) {
return "exception";
}
try {
rLock.lock(5, TimeUnit.SECONDS);
System.out.println("execute business");
TimeUnit.SECONDS.sleep(20);
return "success";
} catch (Exception e) {
return "lock exception";
} finally {
if (rLock.isLocked()) {
rLock.unlock();
}
}
}
- 在解锁的时候,不判断锁的状态,会报错吗,反正都会解锁?
tryLock 不会。
而 lock 会报错,报错信息如下:
java.lang.IllegalMonitorStateException: attempt to unlock lock, not locked by current thread by node id: 8b7b0374-506b-442f-9bc4-9e1c1cbf4d46 thread-id: 61