完美解决 Spring WebSocket 自动断开连接再创建引发的问题
问题描述
在使用 Spring WebSocket 开发时，经常会遇到一个棘手的问题：WebSocket 连接自动断开后，重连时服务端新建了连接，但浏览器依旧保持上一次的连接，导致双方无法通信。
分析原因
我们知道，WebSocket 是基于 TCP 协议实现的，而 TCP 协议是面向连接的，所以每次连接都需要进行握手等建立过程。由于 WebSocket 连接是长连接，当连接断开后，Spring WebSocket 会尝试自动重新建立连接。但是，如果客户端与服务端之间的连接没有完全释放，就会出现上述情况。
解决方案
针对这个问题，我们可以采用以下方案：
- 在服务端配置心跳检测，若客户端长时间没有响应，则关闭连接。
- 在客户端监听
onclose事件，当连接断开时发起重连请求。 - 在服务端添加拦截器，在本次请求中传递 WebSocketSession 等信息，并在拦截器中进行相关处理。
下面我们分别介绍具体实现方法。
方案一：心跳检测
有些情况下，我们可能需要实现心跳检测，以防止客户端假死或网络波动造成连接断开。在 Spring WebSocket 中，可以通过 WebSocketHandlerDecoratorFactory 实现心跳检测。具体实现方式如下：
------ ----- ----------------------------------------- ---------- -------------------------------- -
------- ----- --- ------------------
------- ----- ------ -----------------
------ --------------------------------------------- ------------------ ------ ----------------- -
---------------------- - ------------------
--------------------- - -----------------
-
---------
------ ---------------- -------------- ---------------- -------- -
------ --- ---------------------------------- -
------- ------------------ --------------
---------
------ ---- ------------------------------------------- -------- ------ --------- -
-- -----------
------------------ - -------
----------- -- ----------------------- --------------------------------
------------------------------------------
-
---------
------ ---- ------------------------------ -------- ------------------- -------- ------ --------- -
-- -----------
-- ------------------- -- ----- -
---------------------------------
-
------------------ - -------
----------- -- ----------------------- --------------------------------
---------------------------- ---------
-
---------
------ ---- -------------------------------------- -------- ----------- ------------ ------ --------- -
-- -----------
-- ------------------- -- ----- -
---------------------------------
-
------------------------------------ -------------
-
--
-
-在上述代码中，我们定义了一个 HeartbeatWebSocketHandlerDecoratorFactory 类，用于创建心跳检测的 WebSocket 处理器。其中，heartbeatInterval 参数表示心跳间隔，heartbeatMessage 参数表示心跳消息。在 decorate 方法中，我们对 WebSocket 处理器进行了装饰，添加了心跳任务。具体实现方式如下：
- 在连接建立时启动心跳任务，并且使用
execute方法将任务提交到任务队列。 - 在收到消息后重置任务，即取消之前的任务并重新启动新的任务。
- 在连接关闭时取消心跳任务。
方案二：重连机制
如果客户端与服务端之间的连接已经断开，我们需要重新建立连接。可以通过监听 onclose 事件来实现这个功能。具体实现方式如下：
-------- --------- -
--- ------ - --- ----------------------------------------
------------- - --------------- -
- ---------------------------------------------------------- --------
-------------------------------------------------------------------------------------