ApiBoot RateLimiter 集成Redis分布式限流方式

Author: hengboy

api-boot-project/api-boot-autoconfigure/src/main/java/org/minbox/framework/api/boot/autoconfigure/ratelimiter/ApiBootRateLimiterAutoConfiguration.java

@@ -17,16 +17,20 @@
 
 package org.minbox.framework.api.boot.autoconfigure.ratelimiter;
 
+import org.minbox.framework.api.boot.plugin.rate.limiter.ApiBootRateLimiter;
 import org.minbox.framework.api.boot.plugin.rate.limiter.ApiBootRateLimiterConfiguration;
 import org.minbox.framework.api.boot.plugin.rate.limiter.config.RateLimiterConfig;
 import org.minbox.framework.api.boot.plugin.rate.limiter.handler.ApiBootDefaultRateLimiterInterceptorHandler;
+import org.minbox.framework.api.boot.plugin.rate.limiter.support.GoogleGuavaRateLimiter;
 import org.springframework.boot.autoconfigure.AutoConfigureAfter;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingClass;
 import org.springframework.boot.autoconfigure.web.servlet.WebMvcAutoConfiguration;
 import org.springframework.boot.context.properties.EnableConfigurationProperties;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Import;
 
 /**
  * ApiBoot RateLimiter Auto Configuration
@@ -43,6 +47,7 @@
 @ConditionalOnClass(ApiBootRateLimiterConfiguration.class)
 @EnableConfigurationProperties(ApiBootRateLimiterProperties.class)
 @AutoConfigureAfter(WebMvcAutoConfiguration.class)
+@Import(ApiBootRateLimiterRedisAutoConfiguration.class)
 public class ApiBootRateLimiterAutoConfiguration {
     /**
      * ApiBoot Rate Limiter Properties
@@ -60,8 +65,20 @@ public ApiBootRateLimiterAutoConfiguration(ApiBootRateLimiterProperties apiBootR
      */
     @Bean
     @ConditionalOnMissingBean
-    public ApiBootDefaultRateLimiterInterceptorHandler apiBootDefaultRateLimiterInterceptorHandler() {
-        return new ApiBootDefaultRateLimiterInterceptorHandler();
+    public ApiBootDefaultRateLimiterInterceptorHandler apiBootDefaultRateLimiterInterceptorHandler(ApiBootRateLimiter apiBootRateLimiter) {
+        return new ApiBootDefaultRateLimiterInterceptorHandler(apiBootRateLimiter);
+    }
+
+    /**
+     * google guava rate limiter
+     *
+     * @return ApiBootRateLimiter
+     */
+    @Bean
+    @ConditionalOnMissingBean
+    @ConditionalOnMissingClass("org.springframework.data.redis.core.RedisTemplate")
+    public ApiBootRateLimiter googleGuavaRateLimiter() {
+        return new GoogleGuavaRateLimiter();
     }
 
     /**
@@ -71,11 +88,11 @@ public ApiBootDefaultRateLimiterInterceptorHandler apiBootDefaultRateLimiterInte
      */
     @Bean
     @ConditionalOnMissingBean
-    public ApiBootRateLimiterConfiguration apiBootRateLimiterConfiguration() {
+    public ApiBootRateLimiterConfiguration apiBootRateLimiterConfiguration(ApiBootDefaultRateLimiterInterceptorHandler apiBootDefaultRateLimiterInterceptorHandler) {
         // rate limiter config
         RateLimiterConfig rateLimiterConfig = new RateLimiterConfig();
         rateLimiterConfig.setInterceptorUrl(apiBootRateLimiterProperties.getInterceptorUrl());
 
-        return new ApiBootRateLimiterConfiguration(rateLimiterConfig, apiBootDefaultRateLimiterInterceptorHandler());
+        return new ApiBootRateLimiterConfiguration(rateLimiterConfig, apiBootDefaultRateLimiterInterceptorHandler);
     }
 }

api-boot-project/api-boot-autoconfigure/src/main/java/org/minbox/framework/api/boot/autoconfigure/ratelimiter/ApiBootRateLimiterRedisAutoConfiguration.java

@@ -0,0 +1,58 @@
+/*
+ * 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.autoconfigure.ratelimiter;
+
+import org.minbox.framework.api.boot.plugin.rate.limiter.ApiBootRateLimiter;
+import org.minbox.framework.api.boot.plugin.rate.limiter.support.RedisLuaRateLimiter;
+import org.springframework.boot.autoconfigure.AutoConfigureAfter;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
+import org.springframework.boot.autoconfigure.data.redis.RedisAutoConfiguration;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.redis.core.ReactiveRedisTemplate;
+import org.springframework.data.redis.core.RedisTemplate;
+
+/**
+ * RateLimiter Automation Configuration in redis Mode
+ *
+ * @author：恒宇少年 - 于起宇
+ * <p>
+ * DateTime：2019-05-05 17:50
+ * Blog：http://blog.yuqiyu.com
+ * WebSite：http://www.jianshu.com/u/092df3f77bca
+ * Gitee：https://gitee.com/hengboy
+ * GitHub：https://github.com/hengboy
+ */
+@Configuration
+@ConditionalOnClass(RedisTemplate.class)
+@AutoConfigureAfter(RedisAutoConfiguration.class)
+public class ApiBootRateLimiterRedisAutoConfiguration {
+
+    /**
+     * redis lua rate limiter
+     *
+     * @param redisTemplate redis template
+     * @return ApiBootRateLimiter
+     */
+    @Bean
+    @ConditionalOnMissingBean
+    public ApiBootRateLimiter redisLuaRateLimiter(RedisTemplate redisTemplate) {
+        return new RedisLuaRateLimiter(redisTemplate);
+    }
+}