cn.ly.base_common.dayu.custom.circuit.CircuitBreakerHandler Maven / Gradle / Ivy

Go to download
package cn.ly.base_common.dayu.custom.circuit;

import cn.ly.base_common.dayu.custom.breaker.CircuitBreaker;
import cn.ly.base_common.dayu.custom.consts.CircuitBreakerConst.Metric;
import cn.ly.base_common.dayu.custom.domain.CircuitBreakerDomain;
import cn.ly.base_common.dayu.custom.helper.CircuitBreakerRedisHelper;
import cn.ly.base_common.utils.date.LyJdk8DateUtil;
import cn.ly.base_common.utils.error.LyThrowableUtil;
import cn.ly.base_common.utils.log4j2.LyLogger;

import com.timgroup.statsd.StatsDClient;

import java.util.Optional;

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;

import lombok.AllArgsConstructor;

/**
 * Created by liaomengge on 2019/6/26.
 */
@AllArgsConstructor
public class CircuitBreakerHandler {

    private static final Logger log = LyLogger.getInstance(CircuitBreakerHandler.class);

    private StatsDClient statsDClient;
    private CircuitBreakerRedisHelper circuitBreakerRedisHelper;

    public  R doHandle(String resource, CircuitBreaker circuitBreaker) {
        return this.doHandle(CircuitBreakerDomain.builder().resource(resource).build(), circuitBreaker);
    }

    public  R doHandle(CircuitBreakerDomain circuitBreakerDomain, CircuitBreaker circuitBreaker) {
        R result;
        int failureCount = 0;
        String resource = circuitBreakerDomain.getResource();
        try {
            if (StringUtils.isBlank(resource)) {
                return circuitBreaker.execute();
            }
            failureCount = circuitBreakerRedisHelper.getFailureCount(resource);
            long latestFailureTime = circuitBreakerRedisHelper.getLatestFailureTime(resource);
            if (failureCount >= circuitBreakerRedisHelper.getCircuitBreakerConfig().getFailureThreshold()) {
                if ((LyJdk8DateUtil.getMilliSecondsTime() - latestFailureTime) <= circuitBreakerRedisHelper.getCircuitBreakerConfig().getResetMilliSeconds()) {
                    //open status
                    log.warn("Resource[{}], Custom Circuit Open...", resource);
                    Optional.ofNullable(statsDClient).ifPresent(val -> statsDClient.increment(Metric.CIRCUIT_BREAKER_PREFIX + resource));
                    return circuitBreaker.fallback();
                }
                //half open status
            }
            //close status
            result = circuitBreaker.execute();
        } catch (Throwable t) {
            log.warn("Resource[{}], request custom circuit handle failed ==> {}", resource,
                    LyThrowableUtil.getStackTrace(t));
            if (failureCount >= circuitBreakerRedisHelper.getCircuitBreakerConfig().getFailureThreshold()) {
                circuitBreakerRedisHelper.getIRedisHelper().set(circuitBreakerRedisHelper.getLatestFailureTimeStr(resource),
                        String.valueOf(LyJdk8DateUtil.getMilliSecondsTime()));
                throw t;
            }
            //todo 是否可以交换下位置
            circuitBreakerRedisHelper.incrFailureCount(resource);
            throw t;
        }
        return result;
    }
}