fizzgate/fizz-gateway-node
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: 210109 test

Browse Source
This commit is contained in:
hongqiaowei
2021-01-09 19:54:38 +08:00
parent 9f957ff95e
commit 2ecc63642a
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/we/filter/FlowControlFilter.java
View File

@@ -77,8 +77,10 @@ public class FlowControlFilter extends ProxyAggrFilter {
long currentTimeSlot = flowStat.currentTimeSlotId();
ResourceRateLimitConfig rlc = resourceRateLimitConfigService.getResourceRateLimitConfig(ResourceRateLimitConfig.GLOBAL);
ResourceRateLimitConfig globalConfig = rlc;
long start = System.currentTimeMillis();
boolean incr = false;
boolean concurrentOrRpsExceed = false;
try {
if (rlc != null && rlc.isEnable()) {
concurrentOrRpsExceed = !flowStat.incrRequest(rlc.resource, currentTimeSlot, rlc.concurrents, rlc.qps);
}
@@ -112,6 +114,8 @@ public class FlowControlFilter extends ProxyAggrFilter {
log.debug(WebUtils.getClientReqPath(exchange) + " already apply rate limit rule: " + rlc, LogService.BIZ_ID, exchange.getRequest().getId());
}
incr = true;
if (concurrentOrRpsExceed) {
StringBuilder b = ThreadContext.getStringBuilder();
@@ -126,7 +130,7 @@ public class FlowControlFilter extends ProxyAggrFilter {
} else {
long start = System.currentTimeMillis();
return chain.filter(exchange)
.doAfterTerminate(
() -> {
@@ -140,6 +144,12 @@ public class FlowControlFilter extends ProxyAggrFilter {
);
}
}
catch (Throwable t) {
if (incr) {
inTheEnd(exchange, start, currentTimeSlot, false);
}
}
}
return chain.filter(exchange);
}