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

Fix circuit breaker config update bug

Browse Source
This commit is contained in:
hongqiaowei
2022-01-27 18:10:24 +08:00
parent b548cd1e47
commit 05f696e187
3 changed files with 42 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fizz-core/src/main/java/we/controller/CacheCheckController.java
View File

@@ -99,6 +99,6 @@ public class CacheCheckController {
@GetMapping("/circuitBreakers")
public Mono<String> circuitBreakers(ServerWebExchange exchange) {
return Mono.just(JacksonUtils.writeValueAsString(circuitBreakManager.getCircuitBreakerMap()));
return Mono.just(JacksonUtils.writeValueAsString(circuitBreakManager.getResource2circuitBreakerMap()));
}
}

44
fizz-core/src/main/java/we/stats/circuitbreaker/CircuitBreakManager.java
View File

@@ -47,7 +47,9 @@ public class CircuitBreakManager {
// private Map<String/*child resource*/, String/*parent resource*/> parentResourceMap = new HashMap<>(128);
private final Map<String/*resource*/, CircuitBreaker> circuitBreakerMap = new HashMap<>(64);
private final Map<Long, CircuitBreaker> circuitBreakerMap = new HashMap<>(64);
private final Map<String, CircuitBreaker> resource2circuitBreakerMap = new HashMap<>(64);
private final Set<String> circuitBreakersFromServiceDefault = new HashSet<>(64);
@@ -81,7 +83,8 @@ public class CircuitBreakManager {
for (Map.Entry<Object, Object> e : es) {
json = (String) e.getValue();
CircuitBreaker cb = JacksonUtils.readValue(json, CircuitBreaker.class);
circuitBreakerMap.put(cb.resource, cb);
circuitBreakerMap.put(cb.id, cb);
resource2circuitBreakerMap.put(cb.resource, cb);
// updateParentResourceMap(cb);
LOGGER.info("init circuit breaker {}", cb);
}
@@ -132,10 +135,17 @@ public class CircuitBreakManager {
try {
CircuitBreaker cb = JacksonUtils.readValue(message, CircuitBreaker.class);
if (cb.isDeleted) {
circuitBreakerMap.remove(cb.resource);
circuitBreakerMap.remove(cb.id);
resource2circuitBreakerMap.remove(cb.resource);
LOGGER.info("remove circuit breaker: {}", cb);
} else {
circuitBreakerMap.put(cb.resource, cb);
CircuitBreaker prev = circuitBreakerMap.get(cb.id);
if (prev != null) {
resource2circuitBreakerMap.remove(prev.resource);
}
circuitBreakerMap.put(cb.id, cb);
resource2circuitBreakerMap.put(cb.resource, cb);
circuitBreakersFromServiceDefault.remove(cb.resource);
LOGGER.info("update circuit breaker: {}", cb);
}
// updateParentResourceMap(cb);
@@ -143,7 +153,7 @@ public class CircuitBreakManager {
if (cb.type == CircuitBreaker.Type.SERVICE_DEFAULT) {
if (cb.isDeleted || !cb.serviceDefaultEnable) {
for (String resource : circuitBreakersFromServiceDefault) {
circuitBreakerMap.remove(resource);
resource2circuitBreakerMap.remove(resource);
}
circuitBreakersFromServiceDefault.clear();
}
@@ -185,12 +195,12 @@ public class CircuitBreakManager {
public boolean permit(ServerWebExchange exchange, long currentTimeWindow, FlowStat flowStat, String service, String path) {
String resource = ResourceIdUtils.buildResourceId(null, null, null, service, path);
// return permit(exchange, currentTimeWindow, flowStat, resource);
CircuitBreaker cb = circuitBreakerMap.get(resource);
CircuitBreaker cb = resource2circuitBreakerMap.get(resource);
if (cb == null) {
resource = ResourceIdUtils.buildResourceId(null, null, null, service, null);
cb = circuitBreakerMap.get(resource);
cb = resource2circuitBreakerMap.get(resource);
if (cb == null) {
cb = circuitBreakerMap.get(ResourceIdUtils.SERVICE_DEFAULT_RESOURCE);
cb = resource2circuitBreakerMap.get(ResourceIdUtils.SERVICE_DEFAULT_RESOURCE);
if (cb != null && cb.serviceDefaultEnable) {
cb = buildCircuitBreakerFromServiceDefault(service, resource);
} else {
@@ -211,7 +221,7 @@ public class CircuitBreakManager {
}
private CircuitBreaker buildCircuitBreakerFromServiceDefault(String service, String resource) {
CircuitBreaker serviceDefaultCircuitBreaker = circuitBreakerMap.get(ResourceIdUtils.SERVICE_DEFAULT_RESOURCE);
CircuitBreaker serviceDefaultCircuitBreaker = resource2circuitBreakerMap.get(ResourceIdUtils.SERVICE_DEFAULT_RESOURCE);
CircuitBreaker cb = new CircuitBreaker();
cb.type = CircuitBreaker.Type.SERVICE;
@@ -229,9 +239,11 @@ public class CircuitBreakManager {
cb.resumeDuration = serviceDefaultCircuitBreaker.resumeDuration;
cb.initGradualResumeTimeWindowContext();
}
cb.responseContentType = serviceDefaultCircuitBreaker.responseContentType;
cb.responseContent = serviceDefaultCircuitBreaker.responseContent;
cb.stateStartTime = serviceDefaultCircuitBreaker.stateStartTime;
circuitBreakerMap.put(resource, cb);
resource2circuitBreakerMap.put(resource, cb);
circuitBreakersFromServiceDefault.add(resource);
return cb;
@@ -255,12 +267,12 @@ public class CircuitBreakManager {
public void correctCircuitBreakerStateAsError(ServerWebExchange exchange, long currentTimeWindow, FlowStat flowStat, String service, String path) {
String resource = ResourceIdUtils.buildResourceId(null, null, null, service, path);
// correctCircuitBreakerState4error(exchange, currentTimeWindow, flowStat, resource);
CircuitBreaker cb = circuitBreakerMap.get(resource);
CircuitBreaker cb = resource2circuitBreakerMap.get(resource);
if (cb == null) {
resource = ResourceIdUtils.buildResourceId(null, null, null, service, null);
cb = circuitBreakerMap.get(resource);
cb = resource2circuitBreakerMap.get(resource);
if (cb == null) {
cb = circuitBreakerMap.get(ResourceIdUtils.SERVICE_DEFAULT_RESOURCE);
cb = resource2circuitBreakerMap.get(ResourceIdUtils.SERVICE_DEFAULT_RESOURCE);
if (cb != null && cb.serviceDefaultEnable) {
cb = buildCircuitBreakerFromServiceDefault(service, resource);
} else {
@@ -297,11 +309,11 @@ public class CircuitBreakManager {
}*/
public CircuitBreaker getCircuitBreaker(String resource) {
return circuitBreakerMap.get(resource);
return resource2circuitBreakerMap.get(resource);
}
public Map<String, CircuitBreaker> getCircuitBreakerMap() {
return circuitBreakerMap;
public Map<String, CircuitBreaker> getResource2circuitBreakerMap() {
return resource2circuitBreakerMap;
}
/*public Map<String, String> getParentResourceMap() {

2
fizz-core/src/test/java/we/stats/circuitbreaker/CircuitBreakManagerTests.java
View File

@@ -87,7 +87,7 @@ public class CircuitBreakManagerTests {
cb.breakDuration = 5 * 1000;
cb.resumeStrategy = CircuitBreaker.ResumeStrategy.IMMEDIATE;
cb.stateStartTime = currentTimeWindow;
Map<String, CircuitBreaker> circuitBreakerMap = circuitBreakManager.getCircuitBreakerMap();
Map<String, CircuitBreaker> circuitBreakerMap = circuitBreakManager.getResource2circuitBreakerMap();
circuitBreakerMap.put(cb.resource, cb);
ResourceStat resourceStat = flowStat.getResourceStat(cb.resource);