流量溢出后添加自定义返回格式

Author: hengboy

api-boot-project/api-boot-plugins/api-boot-plugin-rate-limiter/src/main/java/org/minbox/framework/api/boot/plugin/rate/limiter/aop/interceptor/ApiBootRateLimiterMethodInterceptor.java

@@ -21,11 +21,13 @@
 import org.aopalliance.intercept.MethodInvocation;
 import org.minbox.framework.api.boot.plugin.rate.limiter.ApiBootRateLimiter;
 import org.minbox.framework.api.boot.plugin.rate.limiter.annotation.RateLimiter;
+import org.minbox.framework.api.boot.plugin.rate.limiter.result.RateLimiterOverFlowRequest;
 import org.minbox.framework.api.boot.plugin.tools.AopTools;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.aop.support.AopUtils;
 import org.springframework.util.Assert;
+import org.springframework.util.ObjectUtils;
 
 import java.lang.reflect.Method;
 
@@ -50,9 +52,14 @@ public class ApiBootRateLimiterMethodInterceptor implements MethodInterceptor {
      * ApiBoot RateLimiter
      */
     private ApiBootRateLimiter apiBootRateLimiter;
+    /**
+     * Response results after flow exceeding
+     */
+    private RateLimiterOverFlowRequest overFlowRequest;
 
-    public ApiBootRateLimiterMethodInterceptor(ApiBootRateLimiter apiBootRateLimiter) {
+    public ApiBootRateLimiterMethodInterceptor(ApiBootRateLimiter apiBootRateLimiter, RateLimiterOverFlowRequest overFlowRequest) {
         this.apiBootRateLimiter = apiBootRateLimiter;
+        this.overFlowRequest = overFlowRequest;
         Assert.notNull(apiBootRateLimiter, "No ApiBootRateLimiter implementation class instance.");
         logger.info("ApiBootDefaultRateLimiterInterceptorHandler load complete.");
     }
@@ -82,6 +89,10 @@ public Object invoke(MethodInvocation invocation) throws Throwable {
         } catch (Exception e) {
             logger.error("Current Limiting Request Encountered Exception.", e);
         }
+        // If an instance is created
+        if (!ObjectUtils.isEmpty(overFlowRequest)) {
+            return overFlowRequest.overflow(invocation.getArguments());
+        }
         return null;
     }
 

api-boot-project/api-boot-plugins/api-boot-plugin-rate-limiter/src/main/java/org/minbox/framework/api/boot/plugin/rate/limiter/result/RateLimiterOverFlowRequest.java

@@ -0,0 +1,40 @@
+/*
+ * Copyright [2019] [恒宇少年 - 于起宇]
+ *
+ *      Licensed under the Apache License, Version 2.0 (the "License");
+ *      you may not use this file except in compliance with the License.
+ *      You may obtain a copy of the License at
+ *
+ *          http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *      Unless required by applicable law or agreed to in writing, software
+ *      distributed under the License is distributed on an "AS IS" BASIS,
+ *      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *      See the License for the specific language governing permissions and
+ *      limitations under the License.
+ *
+ */
+
+package org.minbox.framework.api.boot.plugin.rate.limiter.result;
+
+/**
+ * Response results after flow exceeding
+ *
+ * @author：恒宇少年 - 于起宇
+ * <p>
+ * DateTime：2019-05-24 17:53
+ * Blog：http://blog.yuqiyu.com
+ * WebSite：http://www.jianshu.com/u/092df3f77bca
+ * Gitee：https://gitee.com/hengboy
+ * GitHub：https://github.com/hengboy
+ */
+@FunctionalInterface
+public interface RateLimiterOverFlowRequest {
+    /**
+     * Response results after flow exceeding
+     *
+     * @param methodArgs request method args
+     * @return
+     */
+    Object overflow(Object[] methodArgs);
+}