close receiving socket

DatL4g · Jul 20, 2024 · a2f3b00 · a2f3b00
1 parent 9101c28
commit a2f3b00
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/commonMain/kotlin/dev/datlag/k2k/connect/ConnectionServer.kt b/src/commonMain/kotlin/dev/datlag/k2k/connect/ConnectionServer.kt
@@ -40,6 +40,7 @@ internal class ConnectionServer(
 
         receiveJob = scope.launch(Dispatcher.IO) {
             while (currentCoroutineContext().isActive) {
+                closeSocket()
                 val socketAddress = InetSocketAddress(NetInterface.getLocalAddress(), port)
 
                 connectedSocket = socket.bind(socketAddress) {
@@ -66,13 +67,10 @@ internal class ConnectionServer(
         }
     }
 
-    override fun close() {
+    private fun closeSocket() {
         connectedSocket?.close()
         connectedSocket = null
 
-        receiveJob?.cancel()
-        receiveJob = null
-
         socket = aSocket(SelectorManager(Dispatcher.IO)).let {
             if (immediate) {
                 it.tcpNoDelay().tcp()
@@ -81,4 +79,11 @@ internal class ConnectionServer(
             }
         }
     }
+
+    override fun close() {
+        receiveJob?.cancel()
+        receiveJob = null
+
+        closeSocket()
+    }
 }