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协程立刻结束,触发会话关闭。
解决步骤
- 服务端保持会话活跃:在
webSocket块中添加挂起逻辑,维持协程运行直到会话主动关闭。可以使用awaitClose()函数,它会挂起协程直到WebSocket会话关闭,同时支持在会话关闭时执行清理操作。 - 清理无效连接:会话关闭时从连接列表中移除对应项,避免内存泄漏。
修改后的服务端示例代码:
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
相关产品推荐
相关产品推荐

