From dab50b4e1489c4b5f2d6f41aa2e768189f74658c Mon Sep 17 00:00:00 2001
From: "Yangkai.Shen" <237497819@qq.com>
Date: Mon, 30 Sep 2019 15:22:12 +0800
Subject: [PATCH] =?UTF-8?q?:tada:=20=E6=B7=BB=E5=8A=A0=E9=99=90=E6=B5=81?=
 =?UTF-8?q?=E6=B3=A8=E8=A7=A3=EF=BC=8C=E5=BC=82=E5=B8=B8=E6=8D=95=E8=8E=B7?=
 =?UTF-8?q?=EF=BC=8C=E6=B5=8B=E8=AF=95=E7=B1=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../redis/annotation/RateLimiter.java         | 48 +++++++++++++++++++
 .../redis/controller/TestController.java      | 40 ++++++++++++++++
 .../redis/handler/GlobalExceptionHandler.java | 24 ++++++++++
 .../src/main/resources/application.yml        |  4 ++
 4 files changed, 116 insertions(+)
 create mode 100644 spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/annotation/RateLimiter.java
 create mode 100644 spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/controller/TestController.java
 create mode 100644 spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/handler/GlobalExceptionHandler.java
 create mode 100644 spring-boot-demo-ratelimit-redis/src/main/resources/application.yml

diff --git a/spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/annotation/RateLimiter.java b/spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/annotation/RateLimiter.java
new file mode 100644
index 0000000..dda0954
--- /dev/null
+++ b/spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/annotation/RateLimiter.java
@@ -0,0 +1,48 @@
+package com.xkcoding.ratelimit.redis.annotation;
+
+import org.springframework.core.annotation.AliasFor;
+import org.springframework.core.annotation.AnnotationUtils;
+
+import java.lang.annotation.*;
+import java.util.concurrent.TimeUnit;
+
+/**
+ * <p>
+ * 限流注解，添加了 {@link AliasFor} 必须通过 {@link AnnotationUtils} 获取，才会生效
+ * </p>
+ *
+ * @author yangkai.shen
+ * @date Created in 2019/9/30 10:31
+ * @see AnnotationUtils
+ */
+@Target(ElementType.METHOD)
+@Retention(RetentionPolicy.RUNTIME)
+@Documented
+public @interface RateLimiter {
+    long DEFAULT_REQUEST = 10;
+
+    /**
+     * max 最大请求数
+     */
+    @AliasFor("max") long value() default DEFAULT_REQUEST;
+
+    /**
+     * max 最大请求数
+     */
+    @AliasFor("value") long max() default DEFAULT_REQUEST;
+
+    /**
+     * 限流key
+     */
+    String key() default "";
+
+    /**
+     * 超时时长，默认1分钟
+     */
+    long timeout() default 1;
+
+    /**
+     * 超时时间单位，默认 分钟
+     */
+    TimeUnit timeUnit() default TimeUnit.MINUTES;
+}
diff --git a/spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/controller/TestController.java b/spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/controller/TestController.java
new file mode 100644
index 0000000..48d57fa
--- /dev/null
+++ b/spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/controller/TestController.java
@@ -0,0 +1,40 @@
+package com.xkcoding.ratelimit.redis.controller;
+
+import cn.hutool.core.lang.Dict;
+import com.xkcoding.ratelimit.redis.annotation.RateLimiter;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * <p>
+ * 测试
+ * </p>
+ *
+ * @author yangkai.shen
+ * @date Created in 2019/9/30 10:30
+ */
+@Slf4j
+@RestController
+public class TestController {
+
+    @RateLimiter(value = 5)
+    @GetMapping("/test1")
+    public Dict test1() {
+        log.info("【test1】被执行了。。。。。");
+        return Dict.create().set("msg", "hello,world!").set("description", "别想一直看到我，不信你快速刷新看看~");
+    }
+
+    @GetMapping("/test2")
+    public Dict test2() {
+        log.info("【test2】被执行了。。。。。");
+        return Dict.create().set("msg", "hello,world!").set("description", "我一直都在，卟离卟弃");
+    }
+
+    @RateLimiter(value = 2)
+    @GetMapping("/test3")
+    public Dict test3() {
+        log.info("【test3】被执行了。。。。。");
+        return Dict.create().set("msg", "hello,world!").set("description", "别想一直看到我，不信你快速刷新看看~");
+    }
+}
diff --git a/spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/handler/GlobalExceptionHandler.java b/spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/handler/GlobalExceptionHandler.java
new file mode 100644
index 0000000..30bdb31
--- /dev/null
+++ b/spring-boot-demo-ratelimit-redis/src/main/java/com/xkcoding/ratelimit/redis/handler/GlobalExceptionHandler.java
@@ -0,0 +1,24 @@
+package com.xkcoding.ratelimit.redis.handler;
+
+import cn.hutool.core.lang.Dict;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+import org.springframework.web.bind.annotation.RestControllerAdvice;
+
+/**
+ * <p>
+ * 全局异常拦截
+ * </p>
+ *
+ * @author yangkai.shen
+ * @date Created in 2019/9/30 10:30
+ */
+@Slf4j
+@RestControllerAdvice
+public class GlobalExceptionHandler {
+
+    @ExceptionHandler(RuntimeException.class)
+    public Dict handler(RuntimeException ex) {
+        return Dict.create().set("msg", ex.getMessage());
+    }
+}
diff --git a/spring-boot-demo-ratelimit-redis/src/main/resources/application.yml b/spring-boot-demo-ratelimit-redis/src/main/resources/application.yml
new file mode 100644
index 0000000..af5002c
--- /dev/null
+++ b/spring-boot-demo-ratelimit-redis/src/main/resources/application.yml
@@ -0,0 +1,4 @@
+server:
+  port: 8080
+  servlet:
+    context-path: /demo