From a39a98cda9d758948b97d5d25e31a9428bad21ef Mon Sep 17 00:00:00 2001
From: 疯狂的狮子li <15040126243@163.com>
Date: 星期一, 06 十二月 2021 14:35:55 +0800
Subject: [PATCH] update 移除 包错误引用
---
ruoyi-common/src/main/java/com/ruoyi/common/utils/RedisUtils.java | 118 +++++++++++++++++++++++++++++++++++++++++++++++-----------
1 files changed, 95 insertions(+), 23 deletions(-)
diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/RedisUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/RedisUtils.java
index 5fcfb9d..3ae0fd4 100644
--- a/ruoyi-common/src/main/java/com/ruoyi/common/utils/RedisUtils.java
+++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/RedisUtils.java
@@ -1,6 +1,6 @@
package com.ruoyi.common.utils;
-import com.google.common.collect.Lists;
+import cn.hutool.core.collection.IterUtil;
import com.ruoyi.common.utils.spring.SpringUtils;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;
@@ -23,7 +23,33 @@
@SuppressWarnings(value = {"unchecked", "rawtypes"})
public class RedisUtils {
- private static RedissonClient client = SpringUtils.getBean(RedissonClient.class);
+ private static final RedissonClient CLIENT = SpringUtils.getBean(RedissonClient.class);
+
+ /**
+ * 闄愭祦
+ *
+ * @param key 闄愭祦key
+ * @param rateType 闄愭祦绫诲瀷
+ * @param rate 閫熺巼
+ * @param rateInterval 閫熺巼闂撮殧
+ * @return -1 琛ㄧず澶辫触
+ */
+ public static long rateLimiter(String key, RateType rateType, int rate, int rateInterval) {
+ RRateLimiter rateLimiter = CLIENT.getRateLimiter(key);
+ rateLimiter.trySetRate(rateType, rate, rateInterval, RateIntervalUnit.SECONDS);
+ if (rateLimiter.tryAcquire()) {
+ return rateLimiter.availablePermits();
+ } else {
+ return -1L;
+ }
+ }
+
+ /**
+ * 鑾峰彇瀹炰緥id
+ */
+ public static String getClientId() {
+ return CLIENT.getId();
+ }
/**
* 鍙戝竷閫氶亾娑堟伅
@@ -33,13 +59,13 @@
* @param consumer 鑷畾涔夊鐞�
*/
public static <T> void publish(String channelKey, T msg, Consumer<T> consumer) {
- RTopic topic = client.getTopic(channelKey);
+ RTopic topic = CLIENT.getTopic(channelKey);
topic.publish(msg);
consumer.accept(msg);
}
public static <T> void publish(String channelKey, T msg) {
- RTopic topic = client.getTopic(channelKey);
+ RTopic topic = CLIENT.getTopic(channelKey);
topic.publish(msg);
}
@@ -51,7 +77,7 @@
* @param consumer 鑷畾涔夊鐞�
*/
public static <T> void subscribe(String channelKey, Class<T> clazz, Consumer<T> consumer) {
- RTopic topic = client.getTopic(channelKey);
+ RTopic topic = CLIENT.getTopic(channelKey);
topic.addListener(clazz, (channel, msg) -> consumer.accept(msg));
}
@@ -62,7 +88,30 @@
* @param value 缂撳瓨鐨勫��
*/
public static <T> void setCacheObject(final String key, final T value) {
- client.getBucket(key).set(value);
+ setCacheObject(key, value, false);
+ }
+
+ /**
+ * 缂撳瓨鍩烘湰鐨勫璞★紝淇濈暀褰撳墠瀵硅薄 TTL 鏈夋晥鏈�
+ *
+ * @param key 缂撳瓨鐨勯敭鍊�
+ * @param value 缂撳瓨鐨勫��
+ * @param isSaveTtl 鏄惁淇濈暀TTL鏈夋晥鏈�(渚嬪: set涔嬪墠ttl鍓╀綑90 set涔嬪悗杩樻槸涓�90)
+ * @since Redis 6.X 浠ヤ笂浣跨敤 setAndKeepTTL 鍏煎 5.X 鏂规
+ */
+ public static <T> void setCacheObject(final String key, final T value, final boolean isSaveTtl) {
+ RBucket<Object> bucket = CLIENT.getBucket(key);
+ if (isSaveTtl) {
+ try {
+ bucket.setAndKeepTTL(value);
+ } catch (Exception e) {
+ long timeToLive = bucket.remainTimeToLive();
+ bucket.set(value);
+ bucket.expire(timeToLive, TimeUnit.MILLISECONDS);
+ }
+ } else {
+ bucket.set(value);
+ }
}
/**
@@ -73,8 +122,8 @@
* @param timeout 鏃堕棿
* @param timeUnit 鏃堕棿棰楃矑搴�
*/
- public static <T> void setCacheObject(final String key, final T value, final Integer timeout, final TimeUnit timeUnit) {
- RBucket<T> result = client.getBucket(key);
+ public static <T> void setCacheObject(final String key, final T value, final long timeout, final TimeUnit timeUnit) {
+ RBucket<T> result = CLIENT.getBucket(key);
result.set(value);
result.expire(timeout, timeUnit);
}
@@ -99,7 +148,7 @@
* @return true=璁剧疆鎴愬姛锛沠alse=璁剧疆澶辫触
*/
public static boolean expire(final String key, final long timeout, final TimeUnit unit) {
- RBucket rBucket = client.getBucket(key);
+ RBucket rBucket = CLIENT.getBucket(key);
return rBucket.expire(timeout, unit);
}
@@ -110,8 +159,19 @@
* @return 缂撳瓨閿�煎搴旂殑鏁版嵁
*/
public static <T> T getCacheObject(final String key) {
- RBucket<T> rBucket = client.getBucket(key);
+ RBucket<T> rBucket = CLIENT.getBucket(key);
return rBucket.get();
+ }
+
+ /**
+ * 鑾峰緱key鍓╀綑瀛樻椿鏃堕棿
+ *
+ * @param key 缂撳瓨閿��
+ * @return 鍓╀綑瀛樻椿鏃堕棿
+ */
+ public static <T> long getTimeToLive(final String key) {
+ RBucket<T> rBucket = CLIENT.getBucket(key);
+ return rBucket.remainTimeToLive();
}
/**
@@ -120,7 +180,7 @@
* @param key
*/
public static boolean deleteObject(final String key) {
- return client.getBucket(key).delete();
+ return CLIENT.getBucket(key).delete();
}
/* */
@@ -132,7 +192,7 @@
* @return
*/
public static void deleteObject(final Collection collection) {
- RBatch batch = client.createBatch();
+ RBatch batch = CLIENT.createBatch();
collection.forEach(t -> {
batch.getBucket(t.toString()).deleteAsync();
});
@@ -147,7 +207,7 @@
* @return 缂撳瓨鐨勫璞�
*/
public static <T> boolean setCacheList(final String key, final List<T> dataList) {
- RList<T> rList = client.getList(key);
+ RList<T> rList = CLIENT.getList(key);
return rList.addAll(dataList);
}
@@ -158,7 +218,7 @@
* @return 缂撳瓨閿�煎搴旂殑鏁版嵁
*/
public static <T> List<T> getCacheList(final String key) {
- RList<T> rList = client.getList(key);
+ RList<T> rList = CLIENT.getList(key);
return rList.readAll();
}
@@ -170,7 +230,7 @@
* @return 缂撳瓨鏁版嵁鐨勫璞�
*/
public static <T> boolean setCacheSet(final String key, final Set<T> dataSet) {
- RSet<T> rSet = client.getSet(key);
+ RSet<T> rSet = CLIENT.getSet(key);
return rSet.addAll(dataSet);
}
@@ -181,7 +241,7 @@
* @return
*/
public static <T> Set<T> getCacheSet(final String key) {
- RSet<T> rSet = client.getSet(key);
+ RSet<T> rSet = CLIENT.getSet(key);
return rSet.readAll();
}
@@ -193,7 +253,7 @@
*/
public static <T> void setCacheMap(final String key, final Map<String, T> dataMap) {
if (dataMap != null) {
- RMap<String, T> rMap = client.getMap(key);
+ RMap<String, T> rMap = CLIENT.getMap(key);
rMap.putAll(dataMap);
}
}
@@ -205,7 +265,7 @@
* @return
*/
public static <T> Map<String, T> getCacheMap(final String key) {
- RMap<String, T> rMap = client.getMap(key);
+ RMap<String, T> rMap = CLIENT.getMap(key);
return rMap.getAll(rMap.keySet());
}
@@ -217,7 +277,7 @@
* @param value 鍊�
*/
public static <T> void setCacheMapValue(final String key, final String hKey, final T value) {
- RMap<String, T> rMap = client.getMap(key);
+ RMap<String, T> rMap = CLIENT.getMap(key);
rMap.put(hKey, value);
}
@@ -229,8 +289,20 @@
* @return Hash涓殑瀵硅薄
*/
public static <T> T getCacheMapValue(final String key, final String hKey) {
- RMap<String, T> rMap = client.getMap(key);
+ RMap<String, T> rMap = CLIENT.getMap(key);
return rMap.get(hKey);
+ }
+
+ /**
+ * 鍒犻櫎Hash涓殑鏁版嵁
+ *
+ * @param key Redis閿�
+ * @param hKey Hash閿�
+ * @return Hash涓殑瀵硅薄
+ */
+ public static <T> T delCacheMapValue(final String key, final String hKey) {
+ RMap<String, T> rMap = CLIENT.getMap(key);
+ return rMap.remove(hKey);
}
/**
@@ -241,7 +313,7 @@
* @return Hash瀵硅薄闆嗗悎
*/
public static <K, V> Map<K, V> getMultiCacheMapValue(final String key, final Set<K> hKeys) {
- RMap<K, V> rMap = client.getMap(key);
+ RMap<K, V> rMap = CLIENT.getMap(key);
return rMap.getAll(hKeys);
}
@@ -252,7 +324,7 @@
* @return 瀵硅薄鍒楄〃
*/
public static Collection<String> keys(final String pattern) {
- Iterable<String> iterable = client.getKeys().getKeysByPattern(pattern);
- return Lists.newArrayList(iterable);
+ Iterable<String> iterable = CLIENT.getKeys().getKeysByPattern(pattern);
+ return IterUtil.toList(iterable);
}
}
--
Gitblit v1.9.3