fix: add connection timeout handling for voice server connections

Author: topi314

LavalinkServer/src/main/java/lavalink/server/player/PlayerRestHandler.kt

@@ -17,13 +17,14 @@ import org.springframework.http.HttpStatus
 import org.springframework.http.ResponseEntity
 import org.springframework.web.bind.annotation.*
 import org.springframework.web.server.ResponseStatusException
+import java.util.concurrent.TimeUnit
 
 @RestController
 class PlayerRestHandler(
     private val socketServer: SocketServer,
     private val filterExtensions: List<AudioFilterExtension>,
     private val pluginInfoModifiers: List<AudioPluginInfoModifier>,
-    serverConfig: ServerConfig,
+    private val serverConfig: ServerConfig,
     private val searchMetrics: SearchMetrics? = null
 ) {
 
@@ -127,16 +128,18 @@ class PlayerRestHandler(
                     context.koe.destroyConnection(guildId)
 
                     val conn = context.getMediaConnection(player)
+                    val timeout = serverConfig.timeouts?.connectTimeoutMs?.toLong() ?: 3000
                     conn.connect(
                         VoiceServerInfo.builder()
                             .setSessionId(it.sessionId)
                             .setEndpoint(it.endpoint)
                             .setToken(it.token)
                             .setChannelId(it.channelId!!.toLong())
                             .build()
-                    ).exceptionally {
+                    ).toCompletableFuture().orTimeout(timeout, TimeUnit.MILLISECONDS).exceptionally {
+                        context.koe.destroyConnection(guildId)
                         throw ResponseStatusException(HttpStatus.INTERNAL_SERVER_ERROR, "Failed to connect to voice server")
-                    }.toCompletableFuture().join()
+                    }.join()
                     player.provideTo(conn)
                 }
             }
@@ -240,5 +243,3 @@ class PlayerRestHandler(
         socketContext(socketServer, sessionId).destroyPlayer(guildId)
     }
 }
-
-