From cf9f9fe4bb2efeea1ed878b993b74a5e37729dca Mon Sep 17 00:00:00 2001
From: Alexander Shabalin <alexander.shabalin@jetbrains.com>
Date: Thu, 22 Oct 2020 13:28:43 +0300
Subject: [PATCH] Retry server errors in dependency downloader (#4458)

---
 .../kotlin/konan/util/DependencyDownloader.kt | 27 ++++++++++++-------
 1 file changed, 18 insertions(+), 9 deletions(-)

diff --git a/shared/src/main/kotlin/org/jetbrains/kotlin/konan/util/DependencyDownloader.kt b/shared/src/main/kotlin/org/jetbrains/kotlin/konan/util/DependencyDownloader.kt
index b00dc06856a..7b4dfd3d4d1 100644
--- a/shared/src/main/kotlin/org/jetbrains/kotlin/konan/util/DependencyDownloader.kt
+++ b/shared/src/main/kotlin/org/jetbrains/kotlin/konan/util/DependencyDownloader.kt
@@ -175,22 +175,31 @@ class DependencyDownloader(
 
         var attempt = 1
         var waitTime = 0L
+        val handleException = { e: Exception ->
+            if (attempt >= maxAttempts) {
+                throw e
+            }
+            attempt++
+            waitTime += attemptIntervalMs
+            println("Cannot download a dependency: $e\n" +
+                    "Waiting ${waitTime.toDouble() / 1000} sec and trying again (attempt: $attempt/$maxAttempts).")
+            // TODO: Wait better
+            Thread.sleep(waitTime)
+        }
         while (true) {
             try {
                 tryDownload(source, tmpFile)
                 break
             } catch (e: HTTPResponseException) {
-                throw e
-            } catch (e: IOException) {
-                if (attempt >= maxAttempts) {
+                if (e.responseCode >= 500) {
+                    // Retry server errors.
+                    handleException(e)
+                } else {
+                    // Do not retry client errors.
                     throw e
                 }
-                attempt++
-                waitTime += attemptIntervalMs
-                println("Cannot download a dependency: $e\n" +
-                        "Waiting ${waitTime.toDouble() / 1000} sec and trying again (attempt: $attempt/$maxAttempts).")
-                // TODO: Wait better
-                Thread.sleep(waitTime)
+            } catch (e: IOException) {
+                handleException(e)
             }
         }