深入解析Polly v8中的重试后回调机制实现

背景介绍

在分布式系统开发中，网络请求失败是常见现象。Polly作为.NET生态中广受欢迎的弹性策略库，提供了强大的重试机制来应对这类问题。标准重试策略中的OnRetry回调会在每次重试前执行，但某些场景下我们更需要在重试失败后执行特定操作。

标准重试机制分析

Polly v8的重试策略默认行为如下：

1. 首次尝试执行并失败
2. 触发OnRetry回调（携带首次尝试结果）
3. 进行第一次重试并失败
4. 再次触发OnRetry回调（携带第一次重试结果）
5. 依此类推直到重试次数耗尽

这种设计存在两个关键特点：

• OnRetry会在原始尝试失败后立即执行
• 最后一次重试失败后不会触发OnRetry

实现重试后回调的方案

1. 扩展策略选项

首先需要创建自定义策略选项类，添加重试后回调属性：

public class CustomRetryStrategyOptions<TResult>: RetryStrategyOptions<TResult>
{
    public Func<OnRetryArguments<TResult>, ValueTask>? OnFailedRetry { get; set; }
}

2. 创建后置处理策略

实现一个专门用于重试后处理的策略：

internal sealed class LogAfterStrategy<TResult>: ResilienceStrategy<TResult>
{
    protected override async ValueTask<Outcome<TResult>> ExecuteCore<TState>(
        Func<ResilienceContext, TState, ValueTask<Outcome<TResult>>> callback,
        ResilienceContext context,
        TState state)
    {
        var outcome = await callback(context, state);
        var shouldRetryArgs = new RetryPredicateArguments<TResult>(context, outcome, options.MaxRetryAttempts);
        var handle = await options.ShouldHandle(shouldRetryArgs);
        
        if(handle)
        {
            var args = new OnRetryArguments<TResult>(context, outcome, options.MaxRetryAttempts, 
                TimeSpan.Zero, TimeSpan.Zero);
            await options.OnFailedRetry(args);
        }
        return outcome;
    }
}

3. 构建组合策略管道

将标准重试策略与后置处理策略组合：

public static ResiliencePipelineBuilder<TResult> AddCustomRetry<TResult>(
    this ResiliencePipelineBuilder<TResult> builder, 
    CustomRetryStrategyOptions<TResult> options)
{
    return builder
        .AddStrategy(context => new LogAfterStrategy<TResult>(options), options)
        .AddRetry(new() {
            // 配置标准重试参数...
            OnRetry = async args => {
                if(options.OnRetry != null)
                    await options.OnRetry(args);
                
                if(options.OnFailedRetry != null && args.AttemptNumber > 0)
                    await options.OnFailedRetry(args);
            }
        });
}

实际应用示例

var pipeline = new ResiliencePipelineBuilder<int>()
    .AddCustomRetry(new CustomRetryStrategyOptions<int>
    {
        ShouldHandle = args => PredicateResult.True(),
        Delay = TimeSpan.FromSeconds(0.1),
        MaxRetryAttempts = 3,
        OnFailedRetry = static args => {
            Console.WriteLine($"重试失败: {args.Outcome}");
            return default;
        }
    })
    .Build();

int attempt = 0;
pipeline.Execute(() => {
    Console.WriteLine($"第{attempt}次尝试");
    return attempt++; 
});

技术要点解析

1. 执行顺序控制：通过策略管道的注册顺序确保后置处理逻辑在重试策略之后执行
2. 首次尝试过滤：通过args.AttemptNumber > 0条件排除首次尝试的误触发
3. 结果处理：在最终策略中检查ShouldHandle结果决定是否执行回调
4. 上下文传递：保持ResilienceContext的延续性配置

适用场景建议

这种模式特别适合以下场景：

1. 精确的失败日志记录：在每次重试确实失败后记录详细信息
2. 资源清理：重试失败后释放临时资源
3. 状态同步：在多节点系统中同步失败状态
4. 渐进式降级：根据连续失败次数调整后续请求策略

性能考量

实现时需注意：

1. 异步回调应配置正确的ContinueOnCapturedContext
2. 复杂逻辑应考虑使用缓冲或批处理减少IO操作
3. 高频场景建议对回调进行性能测试
4. 避免在回调中执行长时间阻塞操作

通过这种扩展方式，开发者可以在不修改Polly核心代码的情况下，灵活地实现各种自定义重试后处理逻辑，大大增强了系统的可观测性和可控性。