ImprovedWebSocket.kt

fun Route.gameSocketRoute() {
    route("/v1/game") {
        standardWebSocket { socket, clientId, _, payload ->
            when (payload) {
                is GameMove -> {
                    // Do something here
                }
                is DisconnectRequest -> {
                    // Do something here
                }
            }
        }
    }
}

fun Route.standardWebSocket(
    handleFrame: suspend (
        socket: DefaultWebSocketServerSession,
        clientId: String,
        frameTextReceived: String,
        payload: BaseModel
    ) -> Unit
) {
    webSocket {
        try {
            incoming.consumeEach { frame ->
                if (frame is Frame.Text) {
                    val frameTextReceived = frame.readText()
                    val jsonObject = JsonParser.parseString(frameTextReceived).asJsonObject
                    val type = when (jsonObject.get("type").asString) {
                        TYPE_GAME_MOVE -> GameMove::class.java
                        TYPE_DISCONNECT_REQUEST -> DisconnectRequest::class.java
                        else -> BaseModel::class.java
                    }
                    val payload = gson.fromJson(frameTextReceived, type)
                    handleFrame(this, clientId, frameTextReceived, payload)
                }
            }
        } catch (e: Exception) {
            e.printStackTrace()
        } finally {
            // Handle Socket Closed
        }
    }
}