Tomcat服务端如何强制关闭/清除WebSocket连接？

问题描述 #

我在Tomcat中使用jakarta.websocket，希望服务器在一段时间后主动关闭与客户端的WebSocket连接。调用session.close()后，确实触发了@OnClose注解的end()方法，但发现不遵守关闭握手协议的客户端仍能通过原会话发送二进制消息。想了解：

• 这是预期行为吗？
• 是否需要双方同意才能关闭/清除会话？
• 如何从服务端强制彻底清除会话，使客户端无法通过原会话连接服务器？

执行关闭操作的代码如下：

private class Disconnector implements Runnable {
    public void run() {
        try {
            session.close();
        } catch (IOException e) {
            // 目前未捕获到异常
        }
    }
}

解答内容 #

1. 这是预期行为吗？ #

是。session.close()只是发起WebSocket优雅关闭握手（向客户端发送Close帧），如果客户端无视该帧、不回复Close帧，Tomcat的WebSocket层默认不会立刻终止底层TCP连接，仍会接收客户端发来的消息——直到TCP层面检测到连接异常，或者服务器主动切断TCP。

2. 是否需要双方同意才能关闭/清除会话？ #

不需要。WebSocket协议的优雅关闭流程是双向握手，但服务器可以强制终止会话，无需客户端配合。默认的session.close()走的是优雅关闭逻辑，而非强制切断。

3. 强制彻底清除会话的方案 #

- 主动终止底层TCP连接

Tomcat的WebSocket会话实现类WsSession可以直接获取到底层TCP套接字，强制关闭即可：

import org.apache.tomcat.websocket.WsSession;

// 注意：需要引入Tomcat的websocket-api依赖
if (session instanceof WsSession) {
    WsSession tomcatSession = (WsSession) session;
    try {
        tomcatSession.getLocal().getSocket().close();
    } catch (IOException e) {
        // 按需处理异常，比如打印日志
    }
}

这种方式会立刻切断TCP连接，客户端无法再通过原会话发送消息。

- 配置会话超时

适合清理闲置会话，超时后Tomcat会自动关闭会话并切断TCP：

• 配置文件方式（web.xml）：

<session-config>
    <session-timeout>30</session-timeout> <!-- 单位：分钟 -->
</session-config>

• 代码方式：

// 设置会话最大闲置时间，单位毫秒，示例为30分钟
session.setMaxIdleTimeout(30 * 60 * 1000);

- 在消息处理中校验会话状态

如果客户端仍发消息过来，服务器可以在处理前校验会话状态，拒绝并强制关闭：

@OnMessage
public void onMessage(Session session, ByteBuffer message) {
    if (!session.isOpen()) {
        // 强制关闭TCP连接
        try {
            if (session instanceof WsSession) {
                ((WsSession)session).getLocal().getSocket().close();
            }
        } catch (IOException e) {}
        return;
    }
    // 正常处理消息逻辑
}

注意事项 #

• 使用WsSession会依赖Tomcat的特定实现，降低代码可移植性，若后续更换WebSocket容器（如Jetty）需调整代码。
• 强制关闭TCP连接可能导致客户端未收到服务器的最后消息，需根据业务场景权衡优雅关闭与强制关闭的使用时机。

内容的提问来源于stack exchange，提问作者Gouz