add auto-retry on steamcmd errors

patrix87 · Jun 26, 2023 · 1ca32de · 1ca32de
1 parent d7d520b
commit 1ca32de
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/functions/server/Update-Server.psm1 b/functions/server/Update-Server.psm1
@@ -56,7 +56,22 @@ function Update-Server {
   #Run the update String
   Write-ServerMsg "$UpdateType $ValidatingString $VersionString Build."
   try {
-    $Task = Start-Process $Global.SteamCMD -ArgumentList "+runscript `"$ScriptPath`"" -Wait -PassThru -NoNewWindow
+    $ExitCode = -1
+    $Retries = 0
+    while ($ExitCode -ne 0 -and $Retries -lt $Global.MaxDownloadRetries) {
+      $Task = Start-Process $Global.SteamCMD -ArgumentList "+runscript `"$ScriptPath`"" -Wait -PassThru -NoNewWindow
+      $ExitCode = $Task.ExitCode
+      if ($ExitCode -ne 0) {
+        Write-ServerMsg "SteamCMD failed to complete. Retrying..."
+        $Retries++
+      }
+      else {
+        Write-ServerMsg "SteamCMD completed successfully."
+      }
+    }
+    if ($Retries -eq $Global.MaxDownloadRetries) {
+      Exit-WithError -ErrorMsg "SteamCMD failed to complete after $Global.MaxDownloadRetries retries."
+    }
   }
   catch {
     Exit-WithError -ErrorMsg "SteamCMD failed to complete."

diff --git a/global.psm1 b/global.psm1
@@ -46,6 +46,9 @@ $GlobalDetails = @{
 
   #Lock Timeout in miuntes
   LockTimeout          = 10
+
+  #Max download retries
+  MaxDownloadRetries   = 5
 }
 
 #Create the object