在 Spring WebSocketClient 上检测连接错误并重新连接

Question

我正在使用 Spring Framework 的 WebSocketClient 和 WebSocketStompClient 来消费消息。设置如下所示：

    WebSocketClient webSocketClient = new StandardWebSocketClient();
    WebSocketStompClient stompClient = new WebSocketStompClient(webSocketClient);

    //messageConverter is MappingJackson2MessageConverter
    stompClient.setMessageConverter(messageConverter);
    stompClient.setTaskScheduler(new ConcurrentTaskScheduler());

    WebSocketHttpHeaders httpHeaders = new WebSocketHttpHeaders();
    StompHeaders stompHeaders = new StompHeaders();
    //... populate authentication headers ... 

    stompClient.connect(url, httpHeaders, stompHeaders, this);

类本身实现StompSessionHandlerAdapter。

然后我将重写 afterConnected() 方法以订阅所需的目的地，并且一切正常。

我也覆盖了handleException()，我期待可能会在出现连接错误时触发。我里面有一个警告日志，所以我希望它在发生连接错误时显示一些东西。

  @Override
  public void handleException(
      StompSession session,
      StompCommand command,
      StompHeaders headers,
      byte[] payload,
      Throwable exception) {
    logger.warn("Error on websocket session " + session.getSessionId(), exception);
  }

但是，当连接断开时，我看不到任何错误。 websocket 服务器只能通过专用网络访问，所以我试图模拟它的方式是断开 VPN，此时消息停止（没有通过handleException() 记录异常）。重新连接 VPN 也不会自动恢复。

如何检测到 websocket 服务器不可用并定期尝试重新连接？

Answer 1

所以事实证明还有另一种方法，handleTransportError()，它在套接字连接断开时被调用。当我覆盖它并且 websocket 服务器变得不可用时，它会被正确调用。

  @Override
  public void handleTransportError(StompSession session, Throwable exception) {
    logger.warn("Error on websocket session " + session.getSessionId(), exception);

    // todo: schedule a reconnection attempt
  }

我最终做的是保留一个标志以了解客户端是否已连接。如果这是第一次断开连接，它会尝试重新建立连接，否则它会使用 Spring 的 ThreadPoolTaskScheduler 来安排 5 秒后的重新连接尝试。