Java连接websocket优雅断线、重连功能

为了实现优雅重连和重试,您需要在代码中添加一些逻辑来处理连接失败或断开连接的情况。

实现代码如下:

java 复制代码
import javax.websocket.*;
import java.io.IOException;

@ClientEndpoint
public class WebSocketClientEndpoint {

    private Session userSession = null;
    private MessageHandler messageHandler;
    private int retryCount = 0;

    public WebSocketClientEndpoint(String endpointURI) {
        try {
            WebSocketContainer container = ContainerProvider.getWebSocketContainer();
            container.connectToServer(this, URI.create(endpointURI));
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    @OnOpen
    public void onOpen(Session userSession) {
        System.out.println("opening websocket");
        this.userSession = userSession;
    }

    @OnClose
    public void onClose(Session userSession, CloseReason reason) {
        System.out.println("closing websocket");
        this.userSession = null;
        retryConnect();
    }

    @OnMessage
    public void onMessage(String message) {
        if (this.messageHandler != null) {
            this.messageHandler.handleMessage(message);
        }
    }

    @OnOpenException
    public void onOpenException(Exception ex) {
        System.out.println("Error opening websocket: " + ex.getMessage());
        retryConnect();
    }

    private void retryConnect() {
        if (retryCount < 5) {
            try {
                Thread.sleep(1000); // 等待1秒后重试连接
                WebSocketContainer container = ContainerProvider.getWebSocketContainer();
                container.connectToServer(this, URI.create("ws://localhost:8080/my-websocket-endpoint"));
                retryCount++;
            } catch (Exception e) {
                System.out.println("Error reconnecting: " + e.getMessage());
                retryConnect(); // 如果连接失败,继续重试连接
            }
        } else {
            System.out.println("Max retries reached, giving up.");
        }
    }

    public void sendMessage(String message) throws IOException {
        this.userSession.getBasicRemote().sendText(message);
    }

    public void addMessageHandler(MessageHandler handler) {
        this.messageHandler = handler;
    }
}
相关推荐
我是华为OD~HR~栗栗呀2 小时前
华为OD-23届-测试面经
java·前端·c++·python·华为od·华为·面试
yy7634966682 小时前
WPF 之 简单高效的Revit多语言支持方案
java·大数据·linux·服务器·wpf
我是华为OD~HR~栗栗呀2 小时前
华为od面经-23届-Java面经
java·c语言·c++·python·华为od·华为·面试
青云交6 小时前
Java 大视界 -- 基于 Java 的大数据机器学习模型在图像识别中的迁移学习与模型优化
java·大数据·迁移学习·图像识别·模型优化·deeplearning4j·机器学习模型
2501_909800817 小时前
Java 集合框架之 Set 接口
java·set接口
断剑zou天涯7 小时前
【算法笔记】暴力递归尝试
java·笔记·算法
Nobody_Cares8 小时前
JWT令牌
java
沐浴露z8 小时前
Kafka入门:基础架构讲解,安装与使用
java·分布式·kafka
神秘的土鸡8 小时前
从数据仓库到数据中台再到数据飞轮:我的数据技术成长之路
java·服务器·aigc·数据库架构·1024程序员节
vir028 小时前
P1928 外星密码(dfs)
java·数据结构·算法·深度优先·1024程序员节