Skip to content

Commit 485c04d

Browse files
committed
添加获取项目内配置流量溢出接口实现类实例
1 parent 8ceb69c commit 485c04d

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

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

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,9 @@
2222
import org.minbox.framework.api.boot.plugin.rate.limiter.aop.interceptor.ApiBootRateLimiterMethodInterceptor;
2323
import org.minbox.framework.api.boot.plugin.rate.limiter.centre.RateLimiterConfigCentre;
2424
import org.minbox.framework.api.boot.plugin.rate.limiter.centre.support.DefaultRateLimiterConfigCentre;
25+
import org.minbox.framework.api.boot.plugin.rate.limiter.result.RateLimiterOverFlowRequest;
2526
import org.minbox.framework.api.boot.plugin.rate.limiter.support.GoogleGuavaRateLimiter;
27+
import org.springframework.beans.factory.ObjectProvider;
2628
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
2729
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
2830
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
@@ -54,9 +56,14 @@ public class ApiBootRateLimiterAutoConfiguration {
5456
* ApiBoot Rate Limiter Properties
5557
*/
5658
private ApiBootRateLimiterProperties apiBootRateLimiterProperties;
59+
/**
60+
* RateLimiter OverFlow Request
61+
*/
62+
private RateLimiterOverFlowRequest rateLimiterOverFlowRequest;
5763

58-
public ApiBootRateLimiterAutoConfiguration(ApiBootRateLimiterProperties apiBootRateLimiterProperties) {
64+
public ApiBootRateLimiterAutoConfiguration(ApiBootRateLimiterProperties apiBootRateLimiterProperties, ObjectProvider<RateLimiterOverFlowRequest> rateLimiterOverFlowRequestObjectProvider) {
5965
this.apiBootRateLimiterProperties = apiBootRateLimiterProperties;
66+
this.rateLimiterOverFlowRequest = rateLimiterOverFlowRequestObjectProvider.getIfAvailable();
6067
}
6168

6269
/**
@@ -106,6 +113,6 @@ ApiBootRateLimiterAdvisor rateLimiterAdvisor(ApiBootRateLimiterMethodInterceptor
106113
@Bean
107114
@ConditionalOnMissingBean
108115
ApiBootRateLimiterMethodInterceptor rateLimiterMethodInterceptor(ApiBootRateLimiter apiBootRateLimiter) {
109-
return new ApiBootRateLimiterMethodInterceptor(apiBootRateLimiter);
116+
return new ApiBootRateLimiterMethodInterceptor(apiBootRateLimiter, rateLimiterOverFlowRequest);
110117
}
111118
}

0 commit comments

Comments
 (0)