Spring AOP实现断路器模式（Circuit Breaker）

无界猴

6个月前更新

037653

断路器模式是微服务架构中提高系统弹性的重要设计模式，用于防止级联故障。Spring AOP（面向切面编程）可以优雅地实现断路器逻辑。以下是详细实现方案：

1. 断路器模式核心概念

断路器有三种状态：

CLOSED：正常请求，监控错误率
OPEN：快速失败，直接拒绝请求
HALF-OPEN：尝试部分请求探测恢复情况

2. 基于Spring AOP的实现步骤

（1）定义断路器注解

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface CircuitBreaker {
    int failureThreshold() default 3; // 触发OPEN状态的失败次数
    long timeout() default 5000;     // OPEN状态持续时间(ms)
    Class<? extends Throwable>[] triggerExceptions() default {Exception.class};
}

（2）实现断路器状态机

public class CircuitBreakerState {
    private volatile State state = State.CLOSED;
    private AtomicInteger failureCount = new AtomicInteger(0);
    private long lastFailureTime;

    enum State { CLOSED, OPEN, HALF_OPEN }

    public boolean allowRequest() {
        if (state == State.OPEN) {
            return System.currentTimeMillis() - lastFailureTime > timeout;
        }
        return true;
    }

    public void recordFailure(int timeout) {
        failureCount.incrementAndGet();
        lastFailureTime = System.currentTimeMillis();
        if (failureCount.get() >= threshold) {
            state = State.OPEN;
        }
    }

    public void reset() {
        state = State.CLOSED;
        failureCount.set(0);
    }
}

（3）AOP切面实现

@Aspect
@Component
public class CircuitBreakerAspect {

    private final Map<String, CircuitBreakerState> breakers = new ConcurrentHashMap<>();

    @Around("@annotation(circuitBreaker)")
    public Object aroundAdvice(ProceedingJoinPoint pjp, CircuitBreaker circuitBreaker) throws Throwable {
        String methodKey = pjp.getSignature().toLongString();
        CircuitBreakerState breaker = breakers.computeIfAbsent(methodKey, 
            k -> new CircuitBreakerState(circuitBreaker.failureThreshold(), circuitBreaker.timeout()));

        if (!breaker.allowRequest()) {
            throw new CircuitBreakerOpenException("Service unavailable");
        }

        try {
            Object result = pjp.proceed();
            breaker.reset(); // 成功则重置状态
            return result;
        } catch (Throwable ex) {
            if (Arrays.stream(circuitBreaker.triggerExceptions())
                     .anyMatch(e -> e.isAssignableFrom(ex.getClass()))) {
                breaker.recordFailure();
            }
            throw ex;
        }
    }
}

（4）使用示例

@Service
public class PaymentService {

    @CircuitBreaker(failureThreshold = 5, timeout = 10000)
    public String processPayment(String orderId) {
        // 调用第三方支付接口
        return thirdPartyPaymentClient.charge(orderId);
    }
}

3. 高级优化方案

（1）结合Hystrix（可选）

@CircuitBreaker(fallbackMethod = "fallbackProcess")
public String processPayment(String orderId) {
    // ...
}

public String fallbackProcess(String orderId) {
    return "Fallback payment processed";
}

（2）状态监控暴露端点

@RestController
@RequestMapping("/circuit-breakers")
public class CircuitBreakerMonitor {

    @Autowired
    private CircuitBreakerAspect aspect;

    @GetMapping
    public Map<String, String> getBreakerStates() {
        return aspect.getBreakers().entrySet().stream()
            .collect(Collectors.toMap(
                Map.Entry::getKey,
                e -> e.getValue().getState().name()
            ));
    }
}

（3）分布式场景改造

@Bean
public CircuitBreakerStateStore breakerStore() {
    return new RedisCircuitBreakerStateStore(redisTemplate); 
}

// 在AOP切面中替换本地Map为分布式存储

4. 对比Spring Cloud CircuitBreaker

特性	本方案	Spring Cloud CircuitBreaker
依赖项	仅需Spring AOP	需要Spring Cloud依赖
配置灵活性	完全自定义	预定义配置
分布式支持	需自行实现	原生支持
监控集成	需手动暴露	集成Actuator