wezone
/
Ry-master
10
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update 独立 CountDownLatch 方法

This commit is contained in:
疯狂的狮子li 2021-06-10 14:04:41 +08:00
parent 1bf9dd1217
commit f6ccfe6ed3
1 changed files with 4 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockManager.java
View File

@ -32,11 +32,6 @@ public class RedisLockManager {
*/ */
private final static Integer FAIR_LOCK = 2; private final static Integer FAIR_LOCK = 2;
/**
* 计数锁
*/
private final static Integer COUNT_LOCK = 3;
/** /**
* 存放当前线程获取锁的类型 * 存放当前线程获取锁的类型
*/ */
@ -45,20 +40,18 @@ public class RedisLockManager {
/** /**
* 获取锁 * 获取锁
*/ */
private <T> T getLock(String key, Integer lockType) { private RLock getLock(String key, Integer lockType) {
Assert.isTrue(StrUtil.isNotBlank(key), "key不能为空"); Assert.isTrue(StrUtil.isNotBlank(key), "key不能为空");
threadLocal.set(lockType); threadLocal.set(lockType);
Object lock; RLock lock;
if (BASE_LOCK.equals(lockType)) { if (BASE_LOCK.equals(lockType)) {
lock = redissonClient.getLock(key); lock = redissonClient.getLock(key);
} else if (FAIR_LOCK.equals(lockType)) { } else if (FAIR_LOCK.equals(lockType)) {
lock = redissonClient.getFairLock(key); lock = redissonClient.getFairLock(key);
} else if (COUNT_LOCK.equals(lockType)) {
lock = redissonClient.getCountDownLatch(key);
} else { } else {
throw new RuntimeException("锁不存在!"); throw new RuntimeException("锁不存在!");
} }
return (T)lock; return lock;
} }
/** /**
@ -118,7 +111,7 @@ public class RedisLockManager {
*/ */
public RCountDownLatch getCountDownLatch(String key, long count) { public RCountDownLatch getCountDownLatch(String key, long count) {
Assert.isTrue(count >= 0, "count数量必须大于等于0"); Assert.isTrue(count >= 0, "count数量必须大于等于0");
RCountDownLatch rCountDownLatch = getLock(key, COUNT_LOCK); RCountDownLatch rCountDownLatch = redissonClient.getCountDownLatch(key);
rCountDownLatch.trySetCount(count); rCountDownLatch.trySetCount(count);
return rCountDownLatch; return rCountDownLatch;
} }