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;
    }
}
相关推荐
碎叶城李白15 分钟前
若依学习笔记1-validated
java·笔记·学习·validated
都叫我大帅哥41 分钟前
🌊 Redis Stream深度探险:从秒杀系统到面试通关
java·redis
都叫我大帅哥42 分钟前
Redis持久化全解析:从健忘症患者到记忆大师的逆袭
java·redis
程序猿阿越1 小时前
Kafka源码(一)Controller选举与创建Topic
java·后端·源码
程序无bug1 小时前
Spring6 当中 Bean 的生命周期的详细解析:有五步,有七步,有十步
java
二川bro1 小时前
飞算智造JavaAI:智能编程革命——AI重构Java开发新范式
java·人工智能·重构
Q_970956391 小时前
java+vue+SpringBoo校园失物招领网站(程序+数据库+报告+部署教程+答辩指导)
java·数据库·vue.js
Wyc724092 小时前
Maven
java·数据库·maven
程序猿小D2 小时前
[附源码+数据库+毕业论文]基于Spring+MyBatis+MySQL+Maven+jsp实现的电影小说网站管理系统,推荐!
java·数据库·mysql·spring·毕业设计·ssm框架·电影小说网站