fix: Handle transient error for HTTP request

Closes #276
TheCloudTheory · Sep 13, 2024 · 5c27650 · 5c27650
1 parent 20258b7
commit 5c27650
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/ace/WhatIf/AzureWhatIfHandler.cs b/ace/WhatIf/AzureWhatIfHandler.cs
@@ -128,7 +128,7 @@ public AzureWhatIfHandler(string scopeId,
         return data;
     }
 
-    private async Task<HttpResponseMessage> SendInitialRequest(CancellationToken cancellationToken)
+    private async Task<HttpResponseMessage> SendInitialRequest(CancellationToken cancellationToken, int retryCount = 0)
     {
         var token = GetToken(cancellationToken);
         var request = new HttpRequestMessage(HttpMethod.Post, CreateUrlBasedOnScope());
@@ -152,8 +152,23 @@ private async Task<HttpResponseMessage> SendInitialRequest(CancellationToken can
         request.Content = new StringContent(templateContent, Encoding.UTF8, "application/json");
         request.Headers.Authorization = new AuthenticationHeaderValue("Bearer", token);
 
-        var response = await httpClient.Value.SendAsync(request, cancellationToken);
-        return response;
+        try
+        {
+            var response = await httpClient.Value.SendAsync(request, cancellationToken);
+            return response;
+        }
+        catch(IOException ex)
+        {
+            if(retryCount < 3)
+            {
+                this.logger.LogError(ex, "Error while sending request to What If API. Retrying...");
+                await Task.Delay(5000, cancellationToken);
+                return await SendInitialRequest(cancellationToken, retryCount + 1);
+            }
+
+            this.logger.LogError(ex, "Error while sending request to What If API.");
+            throw;
+        }
     }
 
     private string CreateUrlBasedOnScope()