Fix OciRegistryApi retry: only retry when closed before response

SgtSilvio · Mar 10, 2024 · a42f20e · a42f20e
1 parent b5627d2
commit a42f20e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/kotlin/io/github/sgtsilvio/gradle/oci/internal/registry/OciRegistryApi.kt b/src/main/kotlin/io/github/sgtsilvio/gradle/oci/internal/registry/OciRegistryApi.kt
@@ -633,7 +633,9 @@ internal class OciRegistryApi(httpClient: HttpClient) {
         }
 }
 
-private val RETRY_SPEC: RetrySpec = Retry.max(3).filter { error -> error is PrematureCloseException }
+private val RETRY_SPEC: RetrySpec = Retry.max(3).filter { error ->
+    (error is PrematureCloseException) && (error.message?.contains("BEFORE response") == true)
+}
 
 private class DigestVerifyingFlux(
     source: Flux<ByteBuf>,