websocket ping pong机制Java实战

WebSocket 是一种支持全双工通信的协议，在使用 WebSocket 进行通信时，服务器和客户端之间会保持一个长连接，双方可以随时发送和接收消息。为了保持这个长连接，WebSocket 会定期发送一个 ping 消息，如果客户端在规定时间内没有回复一个 pong 消息，服务器就会认为客户端已经断开连接。这就是所谓的 WebSocket ping pong 机制。

下面是 Java 的 WebSocket ping pong 机制示例：

import java.net.URI;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import org.java_websocket.client.WebSocketClient;
import org.java_websocket.handshake.ServerHandshake;

public class WebSocketPingPongDemo {
    private static final String WS_URL = "ws://localhost:8080/ws"; // WebSocket 服务端地址
    private static final int PING_INTERVAL = 5; // ping 间隔时间，单位秒
    private static final int PING_TIMEOUT = 3; // ping 超时时间，单位秒

    public static void main(String[] args) throws Exception {
        WebSocketClient client = new WebSocketClient(new URI(WS_URL)) {
            private CountDownLatch pingLatch = new CountDownLatch(1);

            @Override
            public void onOpen(ServerHandshake handshake) {
                System.out.println("WebSocket 连接已建立");
                startPingThread();
            }

            @Override
            public void onMessage(String message) {
                System.out.println("收到服务器消息：" + message);
            }

            @Override
            public void onClose(int code, String reason, boolean remote) {
                System.out.println("WebSocket 连接已关闭，code=" + code + ", reason=" + reason + ", remote=" + remote);
                stopPingThread();
            }

            @Override
            public void onError(Exception ex) {
                System.out.println("WebSocket 连接出错：" + ex.getMessage());
            }

            private void startPingThread() {
                new Thread(() -> {
                    while (!isClosed()) {
                        try {
                            Thread.sleep(PING_INTERVAL * 1000);
                            sendPing();
                            if (!pingLatch.await(PING_TIMEOUT, TimeUnit.SECONDS)) {
                                System.out.println("WebSocket ping 超时，断开连接");
                                close();
                                break;
                            }
                            pingLatch = new CountDownLatch(1);
                        } catch (InterruptedException e) {
                            break;
                        }
                    }
                }).start();
            }

            private void stopPingThread() {
                interruptPingThread();
            }

            private void interruptPingThread() {
                try {
                    pingLatch.countDown();
                } catch (Exception e) {
                }
            }

            private void sendPing() {
                try {
                    System.out.println("发送 ping 消息");
                    pingLatch = new CountDownLatch(1);
                    send("ping");
                } catch (Exception e) {
                }
            }
        };
        client.connect();
    }
}

在这个示例中，我们使用了 Java-WebSocket 库来实现 WebSocket 客户端。在客户端建立连接后，我们启动了一个线程来定时发送 ping 消息，如果在规定时间内没有收到服务器的 pong 消息，就认为连接已经断开。在发送 ping 消息时，我们使用了一个 CountDownLatch 来等待 pong 消息的到来，如果在规定时间内没有得到 pong 消息，就认为连接已经断开。在断开连接时，我们中断了 ping 线程，以免它一直等待 pong 消息。