Retry connecting to the API upon Socket exceptions

Fixes springdoc#154
hosamaly · Sep 18, 2024 · cda433e · cda433e
1 parent 0b20fe3
commit cda433e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/kotlin/org/springdoc/openapi/gradle/plugin/OpenApiGeneratorTask.kt b/src/main/kotlin/org/springdoc/openapi/gradle/plugin/OpenApiGeneratorTask.kt
@@ -21,8 +21,8 @@ import org.gradle.api.tasks.Optional
 import org.gradle.api.tasks.OutputDirectory
 import org.gradle.api.tasks.TaskAction
 import java.io.FileInputStream
-import java.net.ConnectException
 import java.net.HttpURLConnection
+import java.net.SocketException
 import java.net.URL
 import java.security.KeyStore
 import java.security.SecureRandom
@@ -95,7 +95,7 @@ open class OpenApiGeneratorTask : DefaultTask() {
 		try {
 			val isYaml = url.lowercase(Locale.getDefault()).matches(Regex(".+[./]yaml(/.+)*"))
 			val sslContext = getCustomSslContext()
-			await ignoreException ConnectException::class withPollInterval Durations.ONE_SECOND atMost Duration.of(
+			await ignoreException SocketException::class withPollInterval Durations.ONE_SECOND atMost Duration.of(
 				waitTimeInSeconds.get().toLong(),
 				SECONDS
 			) until {