From e252a5386c19df2df3a17e444eeb135cb490ff2e Mon Sep 17 00:00:00 2001 From: Cedric Beust Date: Wed, 28 Oct 2015 21:23:51 -0700 Subject: [PATCH] More robust server loop. --- .../kotlin/com/beust/kobalt/internal/remote/KobaltServer.kt | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/main/kotlin/com/beust/kobalt/internal/remote/KobaltServer.kt b/src/main/kotlin/com/beust/kobalt/internal/remote/KobaltServer.kt index 23cdcdc7..93f8cc8a 100644 --- a/src/main/kotlin/com/beust/kobalt/internal/remote/KobaltServer.kt +++ b/src/main/kotlin/com/beust/kobalt/internal/remote/KobaltServer.kt @@ -69,8 +69,8 @@ public class KobaltServer @Inject constructor(val args: Args) : Runnable, IComma log(1, "Listening to port $portNumber") var quit = false val serverSocket = ServerSocket(portNumber) - while (! quit) { - val clientSocket = serverSocket.accept() + val clientSocket = serverSocket.accept() + while (!quit) { outgoing = PrintWriter(clientSocket.outputStream, true) if (pending.size > 0) { log(1, "Emptying the queue, size $pending.size()") @@ -99,6 +99,8 @@ public class KobaltServer @Inject constructor(val args: Args) : Runnable, IComma } } catch(ex: SocketException) { log(1, "Client disconnected, resetting") + } catch(ex: Exception) { + log(1, "Command failed: ${ex.message}") } } }