使用Polly实现基于HTTP响应内容的自动重试策略

背景介绍

在分布式系统开发中，我们经常需要处理各种不稳定的服务调用。Polly是一个流行的.NET弹性与瞬态故障处理库，它提供了多种策略如重试、断路器、超时等来增强应用程序的健壮性。在实际应用中，我们有时会遇到一种特殊场景：服务返回HTTP 200状态码，但响应内容中包含特定标记(如"pending")表示操作尚未完成，需要客户端稍后重试。

问题分析

传统上，Polly的重试策略通常基于HTTP状态码或异常类型来触发。但当服务返回200 OK但内容不满足要求时，我们需要自定义重试逻辑。这种情况常见于异步操作API，服务端接受请求后立即返回，但实际处理需要时间，客户端需要通过轮询获取最终结果。

解决方案

方案一：使用Polly v8 API

Polly v8引入了新的构建器API，可以更灵活地定义重试条件：

new ResiliencePipelineBuilder<HttpResponseMessage>()
    .AddRetry(new()
    {
        MaxRetryAttempts = 3,
        ShouldHandle = async args => 
        {
            if(args.Outcome.Result is not null)
            {    
                HttpResponseMessage message = args.Outcome.Result;
                if(message.StatusCode == HttpStatusCode.OK) 
                {
                    string responseMessage = await message.Content.ReadAsStringAsync(); 
                    if(responseMessage.Contains("pending"))
                    {
                        return true;
                    }
                }
            }
            return false;
        }
    }).Build();

这种方式的优势在于：

1. 使用异步方式读取响应内容
2. 清晰的条件判断逻辑
3. 与现代.NET编程风格更契合

方案二：使用Polly v7 API

对于仍在使用Polly v7风格API的项目，有以下几种实现方式：

1. 同步读取响应内容

Policy<HttpResponseMessage>
    .HandleResult(response =>
        response.StatusCode == HttpStatusCode.OK
        && response.Content.ReadAsStringAsync().GetAwaiter().GetResult().Contains("pending"))
    .WaitAndRetry(...)

注意：这种方式会阻塞线程，仅适用于小响应体。

2. 结合Fallback策略

var fallback = Policy<HttpResponseMessage>
    .HandleResult(res => res.StatusCode == HttpStatusCode.OK)
    .FallbackAsync(async (dr, ctx, ct) =>
    {
        var responseBody = await dr.Result.Content.ReadAsStringAsync(ct);
        if(responseBody.Contains("pending"))
        {
            throw new ShouldRetryException();
        }
        return dr.Result;
    });

var retry = Policy<HttpResponseMessage>
    .Handle<ShouldRetryException>()
    .WaitAndRetryAsync(...);

var policy = Policy.WrapAsync(retry, fallback);

3. 内联重试逻辑

var res = await retry.ExecuteAsync(async ct => 
{
    var response = await client.GetAsync("...", ct);  
    
    if(response.StatusCode != HttpStatusCode.OK)
        return response;

    var responseBody = await response.Content.ReadAsStringAsync(ct);
    if(responseBody.Contains("pending"))
    {
        throw new ShouldRetryException();
    }

    return response;
}, CancellationToken.None);

最佳实践建议

1. 重试次数与间隔：根据业务需求合理设置重试次数和间隔时间，避免过度重试
2. 响应体大小：对于大响应体，优先使用异步读取方式
3. 日志记录：记录重试事件，便于问题排查
4. 取消支持：确保重试逻辑支持取消操作
5. 资源清理：正确处理HTTP响应资源

总结

Polly提供了灵活的方式来实现基于响应内容的自定义重试策略。无论是使用v7还是v8 API，开发者都可以根据项目需求和团队偏好选择合适的方式。对于新项目，推荐使用v8 API以获得更好的可读性和异步支持；对于已有项目，可以通过组合策略或内联逻辑实现相同功能。理解这些模式后，开发者可以更有效地处理各种复杂的服务调用场景。