SpringCloud/spring-cloud-gateway
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Adds burst default.

Browse Source
This commit is contained in:
Spencer Gibb
2017-10-06 12:20:03 -04:00
parent a13939bfa2
commit 072b2ad7de
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
spring-cloud-gateway-core/src/main/java/org/springframework/cloud/gateway/filter/factory/RequestRateLimiterWebFilterFactory.java
View File

@@ -53,7 +53,6 @@ public class RequestRateLimiterWebFilterFactory implements WebFilterFactory, App
this.context = context;
}
@Override
public List<String> argNames() {
return Arrays.asList(REPLENISH_RATE_KEY, BURST_CAPACITY_KEY, KEY_RESOLVER_NAME_KEY);
@@ -66,7 +65,12 @@ public class RequestRateLimiterWebFilterFactory implements WebFilterFactory, App
int replenishRate = args.getInt(REPLENISH_RATE_KEY);
// How much bursting do you want to allow?
int capacity = args.getInt(BURST_CAPACITY_KEY);
int burstCapacity;
if (args.hasFieldName(BURST_CAPACITY_KEY)) {
burstCapacity = args.getInt(BURST_CAPACITY_KEY);
} else {
burstCapacity = 0;
}
String beanName;
if (args.hasFieldName(KEY_RESOLVER_NAME_KEY)) {
@@ -78,7 +82,8 @@ public class RequestRateLimiterWebFilterFactory implements WebFilterFactory, App
return (exchange, chain) ->
keyResolver.resolve(exchange).flatMap(key ->
rateLimiter.isAllowed(key, replenishRate, capacity).flatMap(response -> {
//TODO: if key is empty?
rateLimiter.isAllowed(key, replenishRate, burstCapacity).flatMap(response -> {
//TODO: set some headers for rate, tokens left
if (response.isAllowed()) {