Provide better error when dd times out

code-asher · code-asher · commit 7685b37a77fb · 2023-04-27T17:54:18.000-08:00
diff --git a/src/main/kotlin/com/coder/gateway/sdk/Retry.kt b/src/main/kotlin/com/coder/gateway/sdk/Retry.kt
@@ -3,6 +3,7 @@ package com.coder.gateway.sdk
 import com.intellij.openapi.diagnostic.Logger
 import com.intellij.openapi.progress.ProcessCanceledException
 import com.intellij.ssh.SshException
+import com.jetbrains.gateway.ssh.deploy.DeployException
 import kotlinx.coroutines.delay
 import java.util.Random
 import java.util.concurrent.TimeUnit
@@ -67,7 +68,15 @@ suspend fun <T> suspendingRetryWithExponentialBackOff(
           while (remainingMs > 0) {
               val remainingS = TimeUnit.MILLISECONDS.toSeconds(remainingMs)
               val remaining = if (remainingS < 1) "now" else "in $remainingS second${if (remainingS > 1) "s" else ""}"
-              update(attempt, unwrappedEx, remaining)
+              // When the worker upload times out Gateway just says it failed.
+              // Even the root cause (IllegalStateException) is useless. The
+              // error also includes a very long useless tmp path.  With all
+              // that in mind, provide a better error.
+              val mungedEx =
+                  if (unwrappedEx is DeployException && unwrappedEx.message.contains("Worker binary deploy failed"))
+                      DeployException("Failed to upload worker binary...it may have timed out", unwrappedEx)
+                  else unwrappedEx
+              update(attempt, mungedEx, remaining)
               val next = min(remainingMs, TimeUnit.SECONDS.toMillis(1))
               remainingMs -= next
               delay(next)
