redis 限流

CPyeah · CPyeah · commit 46e875627a1b · 2022-05-23T11:00:24.000+08:00
diff --git a/java-redis/src/test/java/org/cp/javaredis/SimpleLimit.java b/java-redis/src/test/java/org/cp/javaredis/SimpleLimit.java
@@ -0,0 +1,47 @@
+package org.cp.javaredis;
+
+import java.time.LocalDateTime;
+import java.util.concurrent.TimeUnit;
+import org.junit.jupiter.api.Test;
+import org.redisson.api.RRateLimiter;
+import org.redisson.api.RateIntervalUnit;
+import org.redisson.api.RateType;
+import org.redisson.api.RedissonClient;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+
+@SpringBootTest
+public class SimpleLimit {
+
+	@Autowired
+	RedissonClient redissonClient;
+
+	@Test
+	public void test() throws InterruptedException {
+		// redisson&#29992;&#20102;zset&#26469;&#35760;&#24405;&#35831;&#27714;&#30340;&#20449;&#24687;&#65292;&#36825;&#26679;&#21487;&#20197;&#38750;&#24120;&#24039;&#22937;&#30340;&#36890;&#36807;&#27604;&#36739;score&#65292;&#20063;&#23601;&#26159;&#35831;&#27714;&#30340;&#26102;&#38388;&#25139;&#65292;&#26469;&#21028;&#26029;&#24403;&#21069;&#35831;&#27714;&#36317;&#31163;&#19978;&#19968;&#20010;&#35831;&#27714;&#26377;&#27809;&#26377;&#36229;&#36807;&#19968;&#20010;&#20196;&#29260;&#29983;&#20135;&#21608;&#26399;&#12290;&#22914;&#26524;&#36229;&#36807;&#20102;&#65292;&#21017;&#35828;&#26126;&#20196;&#29260;&#26742;&#20013;&#30340;&#20196;&#29260;&#38656;&#35201;&#29983;&#20135;&#65292;&#20043;&#21069;&#29992;&#25481;&#20102;&#22810;&#23569;&#20010;&#23601;&#29983;&#20135;&#22810;&#23569;&#20010;&#65292;&#32780;&#20043;&#21069;&#29992;&#25481;&#20102;&#22810;&#23569;&#20010;&#20196;&#29260;&#30340;&#20449;&#24687;&#20063;&#22312;zset&#20013;&#20445;&#23384;&#20102;&#12290;
+		RRateLimiter limiter = redissonClient.getRateLimiter("limiter_1");
+		// 1&#21495;&#38480;&#27969;&#22120;&#65292; &#27599;&#19968;&#31186; &#20801;&#35768;&#19968;&#20010;&#35831;&#27714;
+		limiter.trySetRate(RateType.PER_CLIENT, 1, 1, RateIntervalUnit.SECONDS);
+
+		for (int i = 0; i < 10; i++) {
+			new Thread(() -> {
+
+				// &#38459;&#22622;
+//				limiter.acquire(1);
+
+				boolean result = limiter.tryAcquire(1, 100, TimeUnit.MILLISECONDS);
+				if (!result) {
+					// &#24555;&#36895;&#22833;&#36133;
+					System.out.println(Thread.currentThread().getName() + " failed !");
+					return;
+				}
+
+				System.out.println(Thread.currentThread().getName() + " -> " + LocalDateTime.now());
+				// do something
+			}).start();
+		}
+
+		Thread.sleep(12000);
+	}
+
+}
