Retry failed NexusAPI calls.

wabbajack-tools · Jan 3, 2020 · 412d854 · 412d854
1 parent d20de58
commit 412d854
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 6 deletions.
diff --git a/Wabbajack.Common/Consts.cs b/Wabbajack.Common/Consts.cs
@@ -95,5 +95,6 @@ public static string UserAgent
 
         public static string WabbajackCacheHostname = "build.wabbajack.org";
         public static int WabbajackCachePort = 80;
+        public static int MaxHTTPRetries = 4;
     }
 }
diff --git a/Wabbajack.Lib/NexusApi/NexusApi.cs b/Wabbajack.Lib/NexusApi/NexusApi.cs
@@ -222,14 +222,28 @@ public static async Task<NexusApiClient> Get(string apiKey = null)
 
         public async Task<T> Get<T>(string url)
         {
-            var response = await HttpClient.GetAsync(url, HttpCompletionOption.ResponseHeadersRead);
-            UpdateRemaining(response);
-            if (!response.IsSuccessStatusCode)
-                throw new HttpRequestException($"{response.StatusCode} - {response.ReasonPhrase}");
+            int retries = 0;
+            TOP:
+            try
+            {
+                var response = await HttpClient.GetAsync(url, HttpCompletionOption.ResponseHeadersRead);
+                UpdateRemaining(response);
+                if (!response.IsSuccessStatusCode)
+                    throw new HttpRequestException($"{response.StatusCode} - {response.ReasonPhrase}");
+
 
-            using (var stream = await response.Content.ReadAsStreamAsync())
+                using (var stream = await response.Content.ReadAsStreamAsync())
+                {
+                    return stream.FromJSON<T>();
+                }
+            }
+            catch (TimeoutException)
             {
-                return stream.FromJSON<T>();
+                if (retries == Consts.MaxHTTPRetries)
+                    throw;
+                Utils.Log($"Nexus call to {url} failed, retrying {retries} of {Consts.MaxHTTPRetries}");
+                retries++;
+                goto TOP;
             }
         }