spring-websocket --WebSocketStompClient端实现自动重连

转自https://blog.youkuaiyun.com/u013174217/article/details/53376793

websocketstompclient端实现和服务端建立连接和通信之后，如果服务端断开或重启，client端需要实现自动定时重连机制。

查阅了官网和一些资料没有找到相关的源码，自己写了个：

在client端重写StompSessionHandlerAdapter的类中重写handleTransportError方法：

[java]  view plain  copy

1. @Override  
2. public void handleTransportError(StompSession session, Throwable exception) {  
3.   
4.     if (exception instanceof ConnectionLostException || !isConnected) {  
5.         // if connection lost, call this  
6.         isConnected = false;  
7.         while (!isConnected) try {  
8.             Thread.sleep(5000);  
9.             //new connect--start  
10.             WebSocketClient webSocketClient = new SockJsClient(transports);  
11.             WebSocketStompClient stompClient = new WebSocketStompClient(webSocketClient);  
12.             stompClient.setInboundMessageSizeLimit(size);  
13.             stompClient.setMessageConverter(new MappingJackson2MessageConverter());  
14.             stompClient.setTaskScheduler(new DefaultManagedTaskScheduler()); // for heartbeats  
15.             stompClient.connect(url, wsHeaders, sessionHandler);  
16.             //new connect--end  
17.             isConnected = true;  
18.         } catch (Exception e) {  
19.             logger.error("error {}", e.getMessage());  
20.         }  
21.     }  
22. }  

代码解释：WebSocketStompClient的connect失败后会被这个方法捕获并处理，此处捕获ConnectionLostException错误，则每隔5秒重新建立连接。

                  创建connect即new connect的代码区域为创建client连接的地方，需要根据代码情况自己修改。

运行之后服务端断开，client端会每隔5s发起一次连接，连接失败则报错。如果服务端启动后，client端则会正常连接成功。

以上这种方法，并不是一种很好的方法，

更好的是

ListenableFuture<StompSession> future = stompClient.connect(url, logHandler);
try {
    session = future.get();
    session.setAutoReceipt(true);
}
catch (Exception e) {
    e.printStackTrace();
}

每隔2秒钟，boolean connected = session.isConnected();

如果为false就，重新调用一次connect