From 900867df021ea287d78eb3893c4e0bd7304e213c Mon Sep 17 00:00:00 2001
From: 疯狂的狮子li <15040126243@163.com>
Date: 星期五, 17 九月 2021 16:19:12 +0800
Subject: [PATCH] add RedisUtils 增加限流工具
---
ruoyi-common/src/main/java/com/ruoyi/common/utils/RedisUtils.java | 14 ++++++++++++++
1 files changed, 14 insertions(+), 0 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..5ee7c1b 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
@@ -26,6 +26,20 @@
private static RedissonClient client = SpringUtils.getBean(RedissonClient.class);
/**
+ * 闄愭祦
+ *
+ * @param key 闄愭祦key
+ * @param limitType 闄愭祦绫诲瀷
+ * @param rate 閫熺巼
+ * @param rateInterval 閫熺巼闂撮殧
+ */
+ public static boolean rateLimiter(String key, RateType rateType, int rate, int rateInterval) {
+ RRateLimiter rateLimiter = client.getRateLimiter(key);
+ rateLimiter.trySetRate(rateType, rate, rateInterval, RateIntervalUnit.SECONDS);
+ return rateLimiter.tryAcquire();
+ }
+
+ /**
* 鍙戝竷閫氶亾娑堟伅
*
* @param channelKey 閫氶亾key
--
Gitblit v1.9.3