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

fix: handle case which resource may not decr in some error

Browse Source
This commit is contained in:
hongqiaowei
2021-01-09 20:41:44 +08:00
parent 2ecc63642a
commit c27235e9ef
1 changed files with 12 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -77,10 +77,8 @@ 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);
}
@@ -114,8 +112,6 @@ try {
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();
@@ -130,10 +126,10 @@ incr = true;
} else {
long start = System.currentTimeMillis();
return chain.filter(exchange)
.doAfterTerminate(
() -> {
.doOnSuccess(
r -> {
inTheEnd(exchange, start, currentTimeSlot, true);
}
)
@@ -141,14 +137,16 @@ incr = true;
t -> {
inTheEnd(exchange, start, currentTimeSlot, false);
}
);
}
}
catch (Throwable t) {
if (incr) {
).doOnCancel(
() -> {
inTheEnd(exchange, start, currentTimeSlot, false);
}
}
)
.doOnDiscard(
Object.class,
o -> { inTheEnd(exchange, start, currentTimeSlot, false); }
);
}
}
return chain.filter(exchange);