Spring Boot WebSocket Ping/Pong即时响应及自定义处理咨询

问题场景 #

当前基于Spring Boot的WebSocket实现中，服务器在处理文本或二进制消息时，无法立即响应Ping请求，必须等待当前消息处理完成（比如示例中30秒的模拟耗时）才会回复Pong，这不符合WebSocket RFC中Ping需立即响应的规范。原本预期Tomcat/Jetty等容器自动处理Ping/Pong，但现在需要改为让自定义的WebSocket Handler接收并处理Ping消息，而非容器自动处理。

现有代码实现 #

服务器端（Spring Boot WebSocket Handler） #

public class HandlePingRequest implements WebSocketHandler {

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        // TODO Auto-generated method stub
        
    }

    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
        // TODO Auto-generated method stub

       log.info("received websocket message {}",message);
       Thread.sleep(30000);
       session.sendMessage(new TextMessage("Response message"));
        
    }

    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        // TODO Auto-generated method stub
        
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        // TODO Auto-generated method stub
        
    }

    @Override
    public boolean supportsPartialMessages() {
        // TODO Auto-generated method stub
        return false;
    }
    
}

客户端（Python） #

import websocket
import time
websocket.enableTrace(True)
ws = websocket.WebSocket()

ws.connect("ws://localhost:8080/v0")
ws.ping()
ws.send('Hello')
ws.ping()
time.sleep(800)
ws.close()

解决方案 #

要实现自定义Handler处理Ping消息，需完成两步：禁用容器自动Ping/Pong处理，修改Handler逻辑处理Ping帧。

1. 禁用容器自动Ping处理 #

以Tomcat为例，通过自定义容器配置禁用自动Ping处理：

import org.springframework.boot.web.embedded.tomcat.TomcatWebSocketServletWebServerFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

@Configuration
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new HandlePingRequest(), "/v0")
                .setAllowedOrigins("*");
    }

    @Bean
    public TomcatWebSocketServletWebServerFactory tomcatWebSocketServletWebServerFactory() {
        TomcatWebSocketServletWebServerFactory factory = new TomcatWebSocketServletWebServerFactory();
        factory.addWebSocketContainerCustomizers(container -> {
            // 禁用Tomcat自动处理Ping/Pong，让自定义Handler接收Ping消息
            container.setAutoPingEnabled(false);
        });
        return factory;
    }
}

如果使用Jetty容器，可通过类似方式自定义Jetty的WebSocket容器配置，禁用自动Ping处理。

2. 修改WebSocketHandler处理Ping消息 #

在handleMessage方法中判断消息类型，对Ping消息立即回复Pong：

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.socket.*;

public class HandlePingRequest implements WebSocketHandler {
    private static final Logger log = LoggerFactory.getLogger(HandlePingRequest.class);

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        // 连接建立后的逻辑
    }

    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
        // 优先处理Ping消息，立即回复Pong
        if (message instanceof PingMessage) {
            session.sendMessage(new PongMessage(((PingMessage) message).getPayload()));
            log.info("Received Ping, sent Pong immediately");
            return;
        }

        // 处理文本消息逻辑
        if (message instanceof TextMessage) {
            log.info("Received websocket message: {}", message);
            Thread.sleep(30000); // 模拟耗时业务处理
            session.sendMessage(new TextMessage("Response message"));
        }
    }

    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        // 传输错误处理逻辑
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        // 连接关闭后的逻辑
    }

    @Override
    public boolean supportsPartialMessages() {
        return false;
    }
}

修改后，Ping消息会被Handler优先处理，立即回复Pong，不会被耗时的文本消息处理阻塞，符合RFC规范要求。

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