open connection socket just once

DatL4g · Jul 20, 2024 · 8181c6c · 8181c6c
1 parent a2f3b00
commit 8181c6c
Showing 1 changed file with 16 additions and 19 deletions.
diff --git a/src/commonMain/kotlin/dev/datlag/k2k/connect/ConnectionServer.kt b/src/commonMain/kotlin/dev/datlag/k2k/connect/ConnectionServer.kt
@@ -39,28 +39,25 @@ internal class ConnectionServer(
         close()
 
         receiveJob = scope.launch(Dispatcher.IO) {
-            while (currentCoroutineContext().isActive) {
-                closeSocket()
-                val socketAddress = InetSocketAddress(NetInterface.getLocalAddress(), port)
+            val socketAddress = InetSocketAddress(NetInterface.getLocalAddress(), port)
 
-                connectedSocket = socket.bind(socketAddress) {
-                    reuseAddress = true
-                }.accept().also {
-                    it.use { boundSocket ->
-                        suspendCatching {
-                            val readChannel = boundSocket.openReadChannel()
-                            val buffer = ByteArray(readChannel.availableForRead)
-                            while (true) {
-                                val bytesRead = readChannel.readAvailable(buffer)
-                                if (bytesRead <= 0) {
-                                    break
-                                }
-
-                                listener(buffer)
+            connectedSocket = socket.bind(socketAddress) {
+                reuseAddress = true
+            }.accept().also {
+                it.use { boundSocket ->
+                    suspendCatching {
+                        val readChannel = boundSocket.openReadChannel()
+                        val buffer = ByteArray(readChannel.availableForRead)
+                        while (true) {
+                            val bytesRead = readChannel.readAvailable(buffer)
+                            if (bytesRead <= 0) {
+                                break
                             }
-                        }.onFailure {
-                            boundSocket.close()
+
+                            listener(buffer)
                         }
+                    }.onFailure {
+                        boundSocket.close()
                     }
                 }
             }