Add delay to moving orphaned Links

Author: freyacodes

src/main/kotlin/dev/arbjerg/lavalink/client/LavalinkClient.kt

@@ -9,13 +9,15 @@ import dev.arbjerg.lavalink.internal.ReconnectTask
 import dev.arbjerg.lavalink.protocol.v4.VoiceState
 import reactor.core.Disposable
 import reactor.core.publisher.Flux
+import reactor.core.publisher.Mono
 import reactor.core.publisher.Sinks
 import java.io.Closeable
 import java.time.Duration
 import java.util.concurrent.ConcurrentHashMap
 import java.util.concurrent.CopyOnWriteArrayList
 import java.util.concurrent.Executors
 import java.util.concurrent.TimeUnit
+import kotlin.time.toJavaDuration
 
 /**
  * @param userId ID of the bot for authenticating with Discord
@@ -169,6 +171,21 @@ class LavalinkClient(val userId: Long) : Closeable, Disposable {
             return
         }
 
+        val session = node.cachedSession
+        val canResume = session != null && session.resuming && session.timeoutSeconds > 0
+        if (canResume) {
+            node.resumeTimer = Mono.delay(Duration.ofSeconds(session!!.timeoutSeconds))
+                .subscribe() { transferNodes(node) }
+        } else {
+            transferNodes(node)
+        }
+    }
+
+    internal fun onNodeConnected(node: LavalinkNode) {
+        node.resumeTimer?.dispose()
+    }
+
+    private fun transferNodes(node: LavalinkNode) {
         linkMap.forEach { (_, link) ->
             if (link.node == node) {
                 val voiceRegion = link.cachedPlayer?.voiceRegion

src/main/kotlin/dev/arbjerg/lavalink/client/LavalinkNode.kt

@@ -54,6 +54,7 @@ class LavalinkNode(
     internal val sink: Many<ClientEvent> = Sinks.many().multicast().onBackpressureBuffer()
     val flux: Flux<ClientEvent> = sink.asFlux()
     private val reference: Disposable = flux.subscribe()
+    internal var resumeTimer: Disposable? = null
 
     internal val rest = LavalinkRestClient(this)
     val ws = LavalinkSocket(this)

src/main/kotlin/dev/arbjerg/lavalink/internal/LavalinkSocket.kt

@@ -40,6 +40,7 @@ class LavalinkSocket(private val node: LavalinkNode) : WebSocketListener(), Clos
         logger.info("${node.name} has been connected!")
         open = true
         reconnectsAttempted = 0
+        node.lavalink.onNodeConnected(node)
     }
 
     override fun onMessage(webSocket: WebSocket, text: String) {