From f45b50a796e1db1238b1147f11c2f96c0d405e48 Mon Sep 17 00:00:00 2001
From: sxq <812980466@qq.com>
Date: Tue, 8 Jun 2021 14:55:55 +0800
Subject: [PATCH] =?UTF-8?q?update=20redis=E9=94=81=E5=B7=A5=E5=85=B7?=
 =?UTF-8?q?=E7=B1=BB=E7=BC=96=E5=86=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../common/core/redis/RedisLockManager.java   | 132 ++++++++++++++++++
 1 file changed, 132 insertions(+)
 create mode 100644 ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockManager.java

diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockManager.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockManager.java
new file mode 100644
index 00000000..fb665a99
--- /dev/null
+++ b/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();
+	}
+}