HttpClientFactory 結(jié)合 Polly 輕松實現(xiàn)重試機制

Intro

我們的服務(wù)里有一個 API 會去調(diào)用第三方的接口，設(shè)置了超時時間，最近偶爾會發(fā)生超時的情況，微軟在提供?HttpClientFactory?的同時，也提供了一個基于?Polly?的一個擴(kuò)展，我們可以借助它輕松地實現(xiàn)重試，熔斷等行為。

Sample

使用 Polly 擴(kuò)展需要引用 nuget 包：Microsoft.Extensions.Http.Polly

使用示例：

services.AddHttpClient(ServiceConst.IterableHttpClientName,?x?=>
{
????x.Timeout?=?new?TimeSpan(0,?0,?3);
})
????.AddTransientHttpErrorPolicy(builder?=>
????{
????????return?builder.Or()
????????????.Or()
????????????.Or()
????????????.OrResult(res?=>?res.StatusCode?==?HttpStatusCode.TooManyRequests?||?(int)res.StatusCode?>=?500)
????????????.RetryAsync(5)
????????????;
????})

通過?AddTransientHttpErrorPolicy?擴(kuò)展方法來注冊一個 Polly 的 policy，具體可以通過?policyBuilder?委托來定制自己要處理的情況和 policy 行為，支持方式有很多可以簡單的指定重試，也可以指定?WaitANdRetryAsync?等待一段時間后重試，可以重試一次也可以一直重試下去，非常的靈活，可以根據(jù)自己的業(yè)務(wù)場景進(jìn)行定制化配置，這里的示例直接是用了簡單的重試機制

單元測試

下面提供了一個測試重試的單元測試，也可以作為使用示例的一個參考：


[Fact]
public?async?Task?TaskCanceledException()
{
????var?ticks?=?new?ConcurrentBag<long>();
????var?retryLimit?=?5;

????var?services?=?new?ServiceCollection();
????services.AddHttpClient("test",?x?=>
????????{
????????????x.Timeout?=?TimeSpan.FromSeconds(1);
????????})
????????.AddTransientHttpErrorPolicy(builder?=>
????????{
????????????return?builder.Or()
????????????????????.Or()
????????????????????.Or()
????????????????????.OrResult(res?=>
????????????????????????res.StatusCode?==?HttpStatusCode.TooManyRequests?||?(int)res.StatusCode?>=?500)
????????????????????.RetryAsync(retryLimit)
????????????????;
????????})
????????.AddHttpMessageHandler(()?=>?new?MockHttpHandler(request?=>
????????{
????????????ticks.Add(DateTime.UtcNow.Ticks);
????????????throw?new?TaskCanceledException();
????????}))

????????;
????await?using?var?provider?=?services.BuildServiceProvider();
????try
????{
????????using?var?response?=?await?provider.GetRequiredService()
????????????.CreateClient("test")
????????????.GetAsync("api/test");
????}
????catch?(Exception?e)
????{
????????Assert.True(e?is?OperationCanceledException);
????}
????Assert.Equal(retryLimit?+?1,?ticks.Count);
}

private?class?MockHttpHandler?:?DelegatingHandler
{
????private?readonly?Func?_getResponseFunc;

????public?MockHttpHandler(Func?getResponseFunc)
????{
????????_getResponseFunc?=?getResponseFunc;
????}

????protected?override?Task?SendAsync(HttpRequestMessage?request,?CancellationToken?cancellationToken)
????{
????????return?Task.FromResult(_getResponseFunc(request));
????}
}