Java WebSocketClient 自动重连机制实现

org.java_websocket.client.WebSocketClient 类本身没有提供内置的自动重连机制。

本文将介绍如何通过手动方式实现 WebSocketClient 的自动重连，并提供完整的代码示例。

代码示例

import org.java_websocket.client.WebSocketClient;
import org.java_websocket.handshake.ServerHandshake;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

public class MyWebSocketClient extends WebSocketClient {

    private final static int RECONNECT_INTERVAL = 5; // 重连间隔时间，单位为秒
    private ScheduledExecutorService executorService;

    public MyWebSocketClient(URI serverUri) {
        super(serverUri);
    }

    @Override
    public void onOpen(ServerHandshake handshakedata) {
        System.out.println('WebSocket 连接已打开');
    }

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

    @Override
    public void onClose(int code, String reason, boolean remote) {
        System.out.println('WebSocket 连接已关闭，原因：' + reason);
        // 连接关闭时启动重连定时器
        executorService.schedule(() -> {
            try {
                if (!this.isOpen()) {
                    this.reconnect();
                }
            } catch (URISyntaxException e) {
                e.printStackTrace();
            }
        }, RECONNECT_INTERVAL, TimeUnit.SECONDS);
    }

    @Override
    public void onError(Exception ex) {
        System.out.println('WebSocket 出现错误：' + ex.getMessage());
    }

    // 连接方法
    public void connect() throws URISyntaxException {
        super.connect();
        executorService = Executors.newSingleThreadScheduledExecutor();
    }

    // 重连方法
    public void reconnect() throws URISyntaxException {
        System.out.println('正在尝试重连...');
        this.close();
        this.connect();
    }
}

代码说明

定时器: 使用 ScheduledExecutorService 创建一个定时器，用于在连接断开后定时尝试重连。
onClose() 方法: 当 WebSocket 连接关闭时，该方法会被调用。在该方法中，我们启动定时器，尝试每隔 RECONNECT_INTERVAL 秒重连一次。
reconnect() 方法: 该方法负责关闭现有连接并创建新的连接。

注意事项

可以根据实际需求调整重连间隔时间 RECONNECT_INTERVAL。
为了避免无限重连，可以设置最大重连次数，并在超过最大次数后停止重试。
在实际应用中，还需要考虑网络状态变化等因素，以实现更加健壮的重连机制。