Polly库中实现基于断路器状态的自定义回退策略

背景介绍

在现代分布式系统中，断路器模式(Circuit Breaker Pattern)是一种重要的弹性策略，用于防止系统在依赖服务不可用时持续发送请求导致雪崩效应。Polly作为.NET生态中最流行的弹性库之一，提供了强大的断路器实现。然而，在实际应用中，我们有时需要更精细地控制断路器的行为，比如在断路器多次打开后才触发回退策略。

问题分析

在Polly v8中，开发者尝试结合断路器策略和回退策略，但希望只在断路器被打开特定次数(如3次)后才触发回退。初始实现尝试使用ResilienceContext来跟踪断路器打开次数，但发现状态无法在多次执行间保持。

核心问题在于对ResilienceContext生命周期的理解不足。ResilienceContext是为单次执行设计的，每次调用ExecuteAsync都会创建一个新的上下文实例，导致状态无法跨调用保持。

解决方案

1. 共享状态模式

正确的实现方式是创建一个共享状态对象，在每次执行前将其附加到ResilienceContext中。以下是关键实现步骤：

// 定义状态记录类
record State()
{
    public int CircuitOpenedCount { get; set; }
}

// 创建共享状态
var state = new State();

// 每次执行前设置状态到上下文
var context = ResilienceContextPool.Shared.Get("test-resilience-context");
context.Properties.Set(circuitOpenedContextKey, state);

2. 断路器策略配置

在断路器策略中，通过事件钩子更新状态：

var circuitBreakerOptions = new CircuitBreakerStrategyOptions<HttpResponseMessage>
{
    // ...其他配置...
    OnOpened = static args =>
    {
        var state = args.Context.Properties.TryGetValue(circuitOpenedContextKey, out var state2) 
            ? state2 : new();
        state.CircuitOpenedCount++;
        return default;
    },
    OnClosed = static args =>
    {
        args.Context.Properties.Set(circuitOpenedContextKey, new());
        return default;
    }
};

3. 回退策略条件判断

在回退策略中检查断路器打开次数：

var fallbackOptions = new FallbackStrategyOptions<HttpResponseMessage>()
{
    ShouldHandle = static args =>
    {
        if (args.Outcome.Exception is BrokenCircuitException)
        {
            if (args.Context.Properties.TryGetValue(circuitOpenedContextKey, out var state) 
                && state.CircuitOpenedCount >= circuitOpenLimit)
            {
                return PredicateResult.True();
            }
        }
        return PredicateResult.False();
    }
};

实现原理

这种实现方式利用了Polly的上下文属性系统，但通过外部共享状态对象解决了跨调用状态保持的问题。关键在于：

1. 状态对象独立于ResilienceContext存在
2. 每次执行前将状态对象附加到上下文
3. 策略通过上下文访问和修改共享状态

最佳实践

1. 状态对象设计：建议将状态对象设计为不可变(immutable)或线程安全类型，避免并发问题
2. 资源清理：确保在不再需要时清理状态对象，防止内存泄漏
3. 日志记录：在状态变更时添加适当日志，便于调试和监控
4. 性能考量：对于高频调用场景，考虑状态访问的性能影响

扩展思考

这种模式不仅适用于断路器打开次数统计，还可以应用于：

• 基于历史失败率的动态策略调整
• 渐进式回退策略
• 系统健康度评估
• 自适应超时设置

Polly的弹性策略组合提供了极大的灵活性，理解其执行上下文和状态管理机制是构建复杂弹性策略的基础。通过合理设计共享状态，可以实现各种高级弹性模式，满足不同业务场景的需求。