Merge pull request #68 from stebet/host-fix-retries

Fixing retries
HaveIBeenPwned · May 29, 2024 · d64ffb2 · d64ffb2
2 parents b3d7e61 + 472d561
commit d64ffb2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/HaveIBeenPwned.PwnedPasswords.Downloader/Program.cs b/src/HaveIBeenPwned.PwnedPasswords.Downloader/Program.cs
@@ -49,7 +49,7 @@ static IHostBuilder CreateHostBuilder(string[] args) =>
         IHttpClientBuilder clientBuilder = services.AddHttpClient("PwnedPasswords");
         clientBuilder.AddResilienceHandler("retry", b =>
         {
-            b.AddRetry(new RetryStrategyOptions<HttpResponseMessage> { MaxRetryAttempts = 10, OnRetry = OnRequestErrorAsync });
+            b.AddRetry(new RetryStrategyOptions<HttpResponseMessage> { MaxRetryAttempts = 10, OnRetry = OnRequestErrorAsync, ShouldHandle = ShouldHandle });
         });
         clientBuilder.ConfigurePrimaryHttpMessageHandler(() =>
         {
@@ -78,6 +78,7 @@ static IHostBuilder CreateHostBuilder(string[] args) =>
 #endif
         });
     });
+static ValueTask<bool> ShouldHandle(RetryPredicateArguments<HttpResponseMessage> predicate) => ValueTask.FromResult((predicate.Outcome.Result?.IsSuccessStatusCode ?? false) != true);
 
 static ValueTask OnRequestErrorAsync(OnRetryArguments<HttpResponseMessage> args)
 {