Add retry logic to chat download

Contenda-Team · Jun 5, 2021 · 6c8cbf9 · 6c8cbf9
1 parent 8f984a5
commit 6c8cbf9
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/TwitchDownloaderCore/ChatDownloader.cs b/TwitchDownloaderCore/ChatDownloader.cs
@@ -41,6 +41,7 @@ public async Task DownloadAsync(IProgress<ProgressReport> progress, Cancellation
                 double videoStart = 0.0;
                 double videoEnd = 0.0;
                 double videoDuration = 0.0;
+                int errorCount = 0;
 
                 if (downloadType == DownloadType.Video)
                 {
@@ -76,10 +77,25 @@ public async Task DownloadAsync(IProgress<ProgressReport> progress, Cancellation
                 while (latestMessage < videoEnd)
                 {
                     string response;
-                    if (isFirst)
-                        response = await client.DownloadStringTaskAsync(String.Format("https://api.twitch.tv/v5/videos/{0}/comments?content_offset_seconds={1}", videoId, videoStart));
-                    else
-                        response = await client.DownloadStringTaskAsync(String.Format("https://api.twitch.tv/v5/videos/{0}/comments?cursor={1}", videoId, cursor));
+
+                    try
+                    {
+                        if (isFirst)
+                            response = await client.DownloadStringTaskAsync(String.Format("https://api.twitch.tv/v5/videos/{0}/comments?content_offset_seconds={1}", videoId, videoStart));
+                        else
+                            response = await client.DownloadStringTaskAsync(String.Format("https://api.twitch.tv/v5/videos/{0}/comments?cursor={1}", videoId, cursor));
+                        errorCount = 0;
+                    }
+                    catch (WebException ex)
+                    {
+                        await Task.Delay(1000 * errorCount);
+                        errorCount++;
+
+                        if (errorCount >= 10)
+                            throw ex;
+
+                        continue;
+                    }
 
                     CommentResponse commentResponse = JsonConvert.DeserializeObject<CommentResponse>(response);