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 test(String value) { + return R.ok("操作成功", value); + } + + /** + * 测试请求IP限流 + * 同一IP请求受影响 + */ + @RateLimiter(count = 2, time = 10, limitType = LimitType.IP) + @GetMapping("/testip") + public R testip(String value) { + return R.ok("操作成功", value); + } + + /** + * 测试集群实例限流 + * 启动两个后端服务互不影响 + */ + @RateLimiter(count = 2, time = 10, limitType = LimitType.CLUSTER) + @GetMapping("/testcluster") + public R 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 testObj(String value) { + return R.ok("操作成功", value); + } + +}