Spring Boot WebSocket如何实时判断与前端连接断开？

Question

在Spring Boot WebSocket应用中，如何实时捕获前端连接断开的事件是一个常见问题。当客户端与服务器的WebSocket连接中断时，我们需要及时做出响应，例如清理会话数据或更新在线状态。可以通过实现`WebSocketHandler`接口并重写`afterConnectionClosed`方法来处理连接关闭事件。此外，使用`StompSubProtocolErrorHandler`可以捕获STOMP协议下的异常断开情况。但实际开发中可能会遇到延迟检测或无法及时触发回调的问题，这通常与网络不稳定、浏览器行为或服务器配置有关。为确保准确性，可结合心跳机制（如发送定期PING消息）主动检测连接状态，从而提升实时性和可靠性。如何优雅地结合这些方法以适应复杂场景，是开发者需要重点关注的技术难点。

Answer 1

1. 基础概念：WebSocket与Spring Boot集成

在Spring Boot中，WebSocket是一种实现全双工通信的协议，允许客户端和服务器之间保持长期连接。通过这种机制，可以实时推送消息到前端，例如聊天应用、在线状态更新等。

• WebSocket的核心特性是持久连接，支持双向数据传输。
• Spring Boot提供了对WebSocket的支持，可以通过`@EnableWebSocket`注解启用相关功能。

当客户端断开连接时，服务器需要感知这一事件并做出响应。这是通过实现`WebSocketHandler`接口及其方法（如`afterConnectionClosed`）来完成的。

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new MyWebSocketHandler(), "/ws").setAllowedOrigins("*");
    }
}

2. 捕获连接关闭事件：基础实现

为了捕获WebSocket连接关闭事件，开发者通常会重写`afterConnectionClosed`方法。该方法会在客户端断开连接时被调用，从而触发清理或状态更新逻辑。

方法名称	作用
`afterConnectionClosed`	处理连接关闭后的逻辑，如清理资源或更新用户在线状态。
`handleTextMessage`	处理接收到的文本消息。

以下是一个简单的实现示例：

public class MyWebSocketHandler extends TextWebSocketHandler {
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        System.out.println("Connection closed: " + session.getId());
        // 清理会话数据或更新状态
    }
}

3. STOMP协议下的异常断开处理

如果使用STOMP协议进行消息传递，可以通过自定义`StompSubProtocolErrorHandler`来捕获异常断开情况。这有助于在复杂场景下更精确地检测连接问题。

@Bean
public StompSubProtocolErrorHandler stompSubProtocolErrorHandler() {
    return new StompSubProtocolErrorHandler() {
        @Override
        public void handleClientErrorAndClose(Session session, Throwable cause, StompHeaderAccessor headers) {
            System.out.println("STOMP error: " + cause.getMessage());
        }
    };
}

然而，在实际开发中可能会遇到延迟检测的问题，这通常与网络不稳定或浏览器行为有关。

4. 心跳机制：提升实时性与可靠性

为了解决延迟检测或无法及时触发回调的问题，可以引入心跳机制。通过定期发送PING消息并等待PONG响应，服务器能够主动检测连接状态。

心跳机制流程图

graph TD;
    A[客户端建立连接] --> B[服务器开始心跳检测];
    B --> C{是否收到PONG?};
    C --否--> D[标记连接为断开];
    C --是--> E[继续心跳检测];
    

以下是心跳机制的一个简单实现：

@Component
public class HeartbeatTask {
    @Scheduled(fixedRate = 5000)
    public void sendPing() {
        // 定期向所有活动会话发送PING消息
    }
}