From f2d587f7e8bfaf86991642281860e9da6dd1f343 Mon Sep 17 00:00:00 2001
From: zhouyl <583641232@qq.com>
Date: Mon, 2 Sep 2024 15:36:19 +0800
Subject: [PATCH] =?UTF-8?q?:sparkles:=20=20=20=E9=99=90=E6=B5=81=E9=98=B2?=
 =?UTF-8?q?=E6=AD=A2DDoS=E6=94=BB=E5=87=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../demo/RedisRateLimiterController.java      | 63 +++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 tp-functional/src/main/java/com/inscloudtech/functional/controller/demo/RedisRateLimiterController.java

diff --git a/tp-functional/src/main/java/com/inscloudtech/functional/controller/demo/RedisRateLimiterController.java b/tp-functional/src/main/java/com/inscloudtech/functional/controller/demo/RedisRateLimiterController.java
new file mode 100644
index 0000000..3da2582
--- /dev/null
+++ b/tp-functional/src/main/java/com/inscloudtech/functional/controller/demo/RedisRateLimiterController.java
@@ -0,0 +1,63 @@
+package com.inscloudtech.functional.controller.demo;
+
+import com.inscloudtech.common.annotation.RateLimiter;
+import com.inscloudtech.common.core.domain.R;
+import com.inscloudtech.common.enums.LimitType;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+
+/**
+ * 限流样例
+ *
+ */
+@Slf4j
+@RestController
+@RequestMapping("/demo/rateLimiter")
+public class RedisRateLimiterController {
+
+    /**
+     * 测试全局限流
+     * 全局影响
+     */
+    @RateLimiter(count = 2, time = 10)
+    @GetMapping("/test")
+    public R<String> test(String value) {
+        return R.ok("操作成功", value);
+    }
+
+    /**
+     * 测试请求IP限流
+     * 同一IP请求受影响
+     */
+    @RateLimiter(count = 2, time = 10, limitType = LimitType.IP)
+    @GetMapping("/testip")
+    public R<String> testip(String value) {
+        return R.ok("操作成功", value);
+    }
+
+    /**
+     * 测试集群实例限流
+     * 启动两个后端服务互不影响
+     */
+    @RateLimiter(count = 2, time = 10, limitType = LimitType.CLUSTER)
+    @GetMapping("/testcluster")
+    public R<String> testcluster(String value) {
+        return R.ok("操作成功", value);
+    }
+
+    /**
+     * 测试请求IP限流(key基于参数获取)
+     * 同一IP请求受影响
+     *
+     * 简单变量获取 #变量 复杂表达式 #{#变量 != 1 ? 1 : 0}
+     */
+    @RateLimiter(count = 2, time = 10, limitType = LimitType.IP, key = "#value")
+    @GetMapping("/testObj")
+    public R<String> testObj(String value) {
+        return R.ok("操作成功", value);
+    }
+
+}