利用断路器状态统计sentinel熔断次数

最近项目需要sentinel熔断时记录熔断的次数,在经过一阵搜索后决定利用断路器的状态变化来实现此功能
然而,遇到了这样的一个情况,断路器的状态在第一次熔断时正常从close–>open,但在后续(熔断时间内blocked或者熔断结束后重新开启断路器)断路器的状态变化均为half open–>open,导致无法区分究竟是熔断时间内还是再次被熔断了,因此使用了一个flag来做标记实现了功能。

本方法利用了断路器的状态统计熔断次数,代码修改自chatgpt

代码如下:

/**
 * @ClassName: SentinelBlockHandler
 * @Description: sentinel统一异常处理
 * @author: liuzijian @ haiyisoft
 * @Date: 2023/11/7 11:54
 * @version: 1.0
 **/
@Component
@Slf4j
public class SentinelBlockHandler implements BlockExceptionHandler {

    private boolean flag = false;
    @Resource
    private ServerDegradeLogsRepository serverDegradeLogsRepository;
    @Value("${server.port}")
    private Integer serverPort;

    private SentinelBlockHandler() {
        // 私有构造方法,防止外部实例化
    }

    public static SentinelBlockHandler getInstance() {
        return SingletonHolder.INSTANCE;
    }

    private static class SingletonHolder {
        private static final SentinelBlockHandler INSTANCE = new SentinelBlockHandler();
    }

    static {
        registerStateChangeObserver();
    }

    private static void registerStateChangeObserver() {
        EventObserverRegistry.getInstance().addStateChangeObserver("logging",
                (prevState, newState, rule, snapshotValue) -> {
                    List<CircuitBreakerStateChangeObserver> observers = EventObserverRegistry.getInstance().getStateChangeObservers();
                    if (newState == CircuitBreaker.State.OPEN) {
                        getInstance().flag = true;
                        System.err.println(String.format("===> resource: %s, %s -> OPEN at %d, snapshotValue=%.2f",
                                rule.getResource(), prevState.name(), TimeUtil.currentTimeMillis(), snapshotValue));
                    } else {
                        System.err.println(String.format("===> resource: %s, %s -> %s at %d",
                                rule.getResource(), prevState.name(), newState.name(), TimeUtil.currentTimeMillis()));
                    }
                });
    }

    @Override
    public void handle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, BlockException e) throws Exception {
        /**
         * 请求被限流了
         */
        if (e instanceof FlowException){
            httpServletResponse.setStatus(429);
            ResponseService.httpResponse(httpServletResponse, ResponseStatus.SENTINEL_FLOW_ERROR);
            log.error("请求被限流了:{}",e.getMessage());
        }
        /**
         * 请求被熔断降级了
         */
        else if (e instanceof DegradeException) {
            System.out.println("进入熔断异常");
            if (getInstance().flag){
                InetAddress addr = InetAddress.getLocalHost();
                String degradeText = "请求被熔断降级了:{" + e.getMessage() + "}";
                ServerDegradeLogs serverDegradeLogs = new ServerDegradeLogs(addr.getHostAddress(), this.serverPort, degradeText);
                serverDegradeLogsRepository.save(serverDegradeLogs);
                getInstance().flag = false;
            }
            httpServletResponse.setStatus(429);
            ResponseService.httpResponse(httpServletResponse, ResponseStatus.SENTINEL_DEGRADE_ERROR);
            log.error("请求被熔断降级了:{}",e.getMessage());
        }
        /**
         * 热点被限流了
         */
        else if (e instanceof ParamFlowException) {
            httpServletResponse.setStatus(429);
            ResponseService.httpResponse(httpServletResponse, ResponseStatus.SENTINEL_PARAM_FLOW_ERROR);
            log.error("热点被限流了:{}",e.getMessage());
        }
        /**
         * 请求没有权限
         */
        else if (e instanceof AuthorityException) {
            httpServletResponse.setStatus(401);
            ResponseService.httpResponse(httpServletResponse, ResponseStatus.UNAUTHORIZED);
            log.error("请求没有权限:{}",e.getMessage());
        }
    }
}

参考链接:
断路器的使用:https://github.com/alibaba/Sentinel/issues/1939

你可能感兴趣的:(项目,java后端,sentinel,开发语言)