You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Ktor WebSocket会话自动关闭,如何保持会话活跃?

问题

我正在学习Ktor的WebSocket功能,尝试实现多用户间的消息收发,但创建的WebSocket会话会自动关闭,且未主动触发关闭操作。会话在应用登录页面加载时初始化,已查阅相关问题但未找到适配方案(参考方案依赖HTML)。服务端日志显示会话建立后自动发送Frame CLOSE关闭会话,请问是否在服务端遗漏了保持会话运行的配置?

客户端代码

// In CommManager object
suspend fun receiveMessage(session: DefaultWebSocketSession) {
    var text = ""
    while (connected.get() == true) {
        try {
            for (othersMessages in session.incoming) {
                othersMessages as? Frame.Text ?: continue
                text = othersMessages.readText()
                println(text)
            }
        } catch (e: Exception) {
            e.printStackTrace()
        }
    }
}

// In the login page 
suspend fun createWebsocketSession() {
    val session = client.webSocketSession(method = HttpMethod.Get, host = "hosturl", port = 8080, path = "/webSocketSession")
    CommManager.connected.set(true)
    CommManager.receiveMessage(session)
}

init {
    // Initialize Websocket session here
    coroutine.launch() {
        createWebsocketSession()
    }
}

服务端代码

routing() {
    webSocket("/webSocketSession") {
        ConnectionsHandler.connectWebSocketSession(this)
    }
}

// ConnectionsHandler object
fun connectWebSocketSession(session: DefaultWebSocketSession){
    println("adding new user")
    val thisConnection = Connection(session)
    webSocketUserList += thisConnection
    println("${thisConnection.name} is connected!")
}
解答

问题核心在服务端的WebSocket处理逻辑:Ktor的webSocket块是一个协程,当该协程执行完毕时,框架会自动关闭WebSocket会话。你当前的connectWebSocketSession仅完成会话添加就返回,导致webSocket协程立刻结束,触发会话关闭。

解决步骤

  1. 服务端保持会话活跃:在webSocket块中添加挂起逻辑,维持协程运行直到会话主动关闭。可以使用awaitClose()函数,它会挂起协程直到WebSocket会话关闭,同时支持在会话关闭时执行清理操作。
  2. 清理无效连接:会话关闭时从连接列表中移除对应项,避免内存泄漏。

修改后的服务端示例代码:

routing() {
    webSocket("/webSocketSession") {
        val connection = Connection(this)
        webSocketUserList += connection
        println("${connection.name} is connected!")
        // 挂起协程直到会话关闭,同时处理连接清理
        awaitClose {
            webSocketUserList -= connection
            println("${connection.name} disconnected!")
        }
    }
}

客户端代码优化

客户端receiveMessage的外层while循环多余,因为session.incoming的迭代本身会持续接收消息直到会话关闭,可简化为:

suspend fun receiveMessage(session: DefaultWebSocketSession) {
    try {
        session.incoming.filterIsInstance<Frame.Text>().forEach { frame ->
            val text = frame.readText()
            println(text)
        }
    } catch (e: Exception) {
        e.printStackTrace()
        connected.set(false)
    }
}

内容的提问来源于stack exchange,提问作者ktorlearning

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.06.27 12:12:55