【Spring源码】WebSocket做推送动作的底层实例

一、前瞻

Ok，开始我们今天的对Spring的【模块阅读】。

在这里插入图片描述

那就挑Web里的WebSocket模块，先思考下本次阅读的阅读线索：

WebSocket在Spring里起到什么作用
这个模块采用了什么设计模式
我们都知道WebSocket可以主动推送消息给用户，那做推送动作的底层实例究竟是谁

二、探索

我们先整体看下整个模块的组织结构。

在这里插入图片描述

要解决阅读线索3，看起来应该从上图里的server模块开始。

找了一圈发现并没有什么线索，那我们就从阅读官方文档这个方向入手。

在这里插入图片描述

官方文档表面心跳采用了SckJS协议，所以我们应该从图2里的sockjs模块开始探索。

既然是发送消息给用户，那必定有send的动作，我们在sockjs模块全局搜索下。

在这里插入图片描述

我们根据搜索定位到了发送消息的WebSocketSession接口。

	/*** Send a WebSocket message: either {@link TextMessage} or {@link BinaryMessage}.* <p><strong>Note:</strong> The underlying standard WebSocket session (JSR-356) does* not allow concurrent sending. Therefore, sending must be synchronized. To ensure* that, one option is to wrap the {@code WebSocketSession} with the* {@link org.springframework.web.socket.handler.ConcurrentWebSocketSessionDecorator* ConcurrentWebSocketSessionDecorator}.* @see org.springframework.web.socket.handler.ConcurrentWebSocketSessionDecorator*/void sendMessage(WebSocketMessage<?> message) throws IOException;

整体看下该接口的类图，看下是哪些具体的子类对象实现了sendMessage接口。

在这里插入图片描述

我们找到了以上链条的sendMessage具体实现：

	@Overridepublic final void sendMessage(WebSocketMessage<?> message) throws IOException {Assert.state(!isClosed(), "Cannot send a message when session is closed");Assert.isInstanceOf(TextMessage.class, message, "SockJS supports text messages only");sendMessageInternal(((TextMessage) message).getPayload());}

发现sockjs确实是像官方文档所说的，用户客户端发送心跳给服务器，服务器再使用以上的sendMessageInternal保存心跳。但并不是我们要找的阅读线索3。

我们找下WebSocketSession接口的其他子类链条实现，发现真正发送消息给用户的应该是AbstractWebSocketSession。

在这里插入图片描述

可以在代码里找到发送各种消息格式的实现。

到这我们就解决了阅读线索3：

推送动作的底层实例究竟是谁

而关于阅读线索2：WebSocket模块采用了什么设计模式，我们在AbstractWebSocketHandler发现了以下代码块。

public abstract class AbstractWebSocketHandler implements WebSocketHandler {@Overridepublic void afterConnectionEstablished(WebSocketSession session) throws Exception {}@Overridepublic void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {if (message instanceof TextMessage textMessage) {handleTextMessage(session, textMessage);}else if (message instanceof BinaryMessage binaryMessage) {handleBinaryMessage(session, binaryMessage);}else if (message instanceof PongMessage pongMessage) {handlePongMessage(session, pongMessage);}else {throw new IllegalStateException("Unexpected WebSocket message type: " + message);}}protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {}protected void handleBinaryMessage(WebSocketSession session, BinaryMessage message) throws Exception {}protected void handlePongMessage(WebSocketSession session, PongMessage message) throws Exception {}@Overridepublic void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {}@Overridepublic void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {}@Overridepublic boolean supportsPartialMessages() {return false;}}