update redis锁工具类编写

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

@@ -0,0 +1,132 @@
+package com.ruoyi.common.core.redis;
+
+import cn.hutool.core.lang.Assert;
+import cn.hutool.core.lang.Validator;
+import cn.hutool.core.util.StrUtil;
+import org.redisson.api.RCountDownLatch;
+import org.redisson.api.RLock;
+import org.redisson.api.RedissonClient;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.util.concurrent.TimeUnit;
+
+/**
+ * redis 锁管理类
+ *
+ * @author shenxinquan
+ */
+@Component
+public class RedisLockManager {
+
+	@Autowired
+	private RedissonClient redissonClient;
+
+	private final ThreadLocal<RLock> threadLocal = new ThreadLocal<>();
+
+	/**
+	 * 获取锁（不用设置超时时间，一直等待）
+	 */
+	public boolean getLock(String key) {
+		Assert.isTrue(StrUtil.isNotBlank(key), "key不能为空");
+		RLock lock = redissonClient.getLock(key);
+		threadLocal.set(lock);
+		return lock.tryLock();
+	}
+
+	/**
+	 * 设置过期时间
+	 *
+	 * @param key
+	 * @param time       过期时间
+	 * @param expireUnit 时间单位
+	 */
+	public boolean getLock(String key, long time, TimeUnit expireUnit) throws InterruptedException {
+		Assert.isTrue(StrUtil.isNotBlank(key), "key不能为空");
+		Assert.isTrue(time > 0, "过期时间必须大于0");
+		Assert.isTrue(Validator.isNotEmpty(expireUnit), "时间单位不能为空");
+		RLock lock = redissonClient.getLock(key);
+		threadLocal.set(lock);
+		return lock.tryLock(time, expireUnit);
+	}
+
+	/**
+	 * 设置过期时间
+	 *
+	 * @param key
+	 * @param waitTime   获取锁等待时间
+	 * @param leaseTime  保留锁的时间
+	 * @param expireUnit 时间单位
+	 * @throws InterruptedException
+	 */
+	public boolean getLock(String key, long waitTime, long leaseTime, TimeUnit expireUnit) throws InterruptedException {
+		Assert.isTrue(StrUtil.isNotBlank(key), "key不能为空");
+		Assert.isTrue(waitTime > 0, "获取锁等待时间必须大于0");
+		Assert.isTrue(leaseTime > 0, "保留锁的时间必须大于0");
+		Assert.isTrue(Validator.isNotEmpty(expireUnit), "时间单位不能为空");
+		RLock lock = redissonClient.getLock(key);
+		threadLocal.set(lock);
+		return lock.tryLock(waitTime, leaseTime, expireUnit);
+	}
+
+	/**
+	 * 获取计数器锁
+	 *
+	 * @param key
+	 * @param count countDownLatch 的数量
+	 */
+	public RCountDownLatch countDownLatch(String key, long count) {
+		Assert.isTrue(StrUtil.isNotBlank(key), "key不能为空");
+		Assert.isTrue(count >= 0, "count数量必须大于等于0");
+		RCountDownLatch rCountDownLatch = redissonClient.getCountDownLatch(key);
+		rCountDownLatch.trySetCount(count);
+		return rCountDownLatch;
+	}
+
+	/**
+	 * 获取公平锁
+	 *
+	 * @param key
+	 * @param waitTime   获取锁等待时间
+	 * @param leaseTime  持有锁的时间
+	 * @param expireUnit 时间单位
+	 * @return
+	 * @throws InterruptedException
+	 */
+	public boolean getFairLock(String key, long waitTime, long leaseTime, TimeUnit expireUnit) throws InterruptedException {
+		Assert.isTrue(StrUtil.isNotBlank(key), "key不能为空");
+		Assert.isTrue(waitTime > 0, "获取锁等待时间必须大于0");
+		Assert.isTrue(leaseTime > 0, "保留锁的时间必须大于0");
+		Assert.isTrue(Validator.isNotEmpty(expireUnit), "时间单位不能为空");
+		RLock lock = redissonClient.getFairLock(key);
+		threadLocal.set(lock);
+		return lock.tryLock(waitTime, leaseTime, expireUnit);
+	}
+
+	/**
+	 * 获取公平锁
+	 *
+	 * @param key
+	 * @param leaseTime  持有锁的时间
+	 * @param expireUnit 时间单位
+	 * @return
+	 * @throws InterruptedException
+	 */
+	public boolean getFairLock(String key, long leaseTime, TimeUnit expireUnit) throws InterruptedException {
+		Assert.isTrue(StrUtil.isNotBlank(key), "key不能为空");
+		Assert.isTrue(leaseTime > 0, "保留锁的时间必须大于0");
+		Assert.isTrue(Validator.isNotEmpty(expireUnit), "时间单位不能为空");
+		RLock lock = redissonClient.getFairLock(key);
+		threadLocal.set(lock);
+		return lock.tryLock(leaseTime, expireUnit);
+	}
+
+	/**
+	 * 释放锁(统一释放)
+	 */
+	public void unLock() {
+		RLock lock = threadLocal.get();
+		lock.unlock();
+		threadLocal.remove();
+	}
+}