你還在使用 WebSocket 實現實時消息推送嗎？
不過，咱們今天不聊那些新興的玩意兒，還是回歸初心，好好聊聊 WebSocket，畢竟它可是實時通信領域的“老將”了。我們就用 Java 來實現一個簡單的 WebSocket 消息推送示例，讓大家看看 WebSocket 的魅力所在。
WebSocket 是什么？
WebSocket 是一種網絡通信協議，它提供了一種在單個 TCP 連接上進行全雙工通訊的渠道。簡單來說，就是客戶端和服務器之間只需要建立一個連接，就可以通過這個連接雙向傳輸數據，而且數據可以隨時發送，不需要像 HTTP 那樣每次請求都要建立連接，大大提高了通信效率。
為什么選擇 WebSocket？
- 實時性：WebSocket 最大的特點就是實時性，數據可以隨時從服務器發送到客戶端，非常適合需要實時通信的應用場景，比如在線聊天、實時通知等。
- 減少資源消耗：由于 WebSocket 只需要建立一次連接，就可以一直保持通信狀態，所以相比 HTTP 的多次請求/響應模式，WebSocket 大大減少了資源的消耗。
- 更好的二進制支持：WebSocket 提供了對二進制數據的支持，這使得它在處理一些復雜的數據類型時更加高效。
Java 實現 WebSocket 消息推送
要用 Java 實現 WebSocket 消息推送，我們可以使用 Spring Boot 框架，它提供了對 WebSocket 的良好支持。下面是一個簡單的示例：
添加依賴
首先，在你的 Spring Boot 項目的 pom.xml 文件中添加 WebSocket 的依賴：
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>配置 WebSocket
然后，創建一個配置類來啟用和配置 WebSocket：
import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
@Override
public void registerStompEndpoints(StompEndpointRegistry registry) {
registry.addEndpoint("/ws").withSockJS();
}
@Override
public void configureMessageBroker(MessageBrokerRegistry registry) {
registry.enableSimpleBroker("/topic");
registry.setApplicationDestinationPrefixes("/app");
}
}這里我們注冊了一個 /ws 的 WebSocket 端點，并啟用了 SockJS 作為備選方案，以支持那些不支持 WebSocket 的瀏覽器。同時，我們還配置了消息代理，用于將消息從服務器推送到客戶端。
創建消息處理器
接下來，創建一個消息處理器來處理 WebSocket 消息：
import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.stereotype.Controller;
@Controller
public class WebSocketController {
@MessageMapping("/hello")
@SendTo("/topic/greetings")
public String greeting(String message) throws Exception {
return "Hello, " + message + "!";
}
}在這個處理器中，我們定義了一個 /hello 的消息映射，當接收到這個消息時，處理器會將其處理并發送到 /topic/greetings 主題下。
客戶端連接 WebSocket
最后，我們需要在客戶端連接到 WebSocket 服務器，并訂閱相應的主題來接收消息。這里以瀏覽器為例，使用 SockJS 和 Stomp.js 庫：
<!DOCTYPE html>
<html>
<head>
<title>WebSocket Test</title>
<script src="https://cdn.jsdelivr.net/npm/sockjs-client/dist/sockjs.min.js"></script>
<script src="https://cdn.jsdelivr.net/npm/stompjs/lib/stomp.min.js"></script>
</head>
<body>
<div id="greetings"></div>
<script type="text/javascript">
var socket = new SockJS('/ws');
var stompClient = Stomp.over(socket);
stompClient.connect({}, function (frame) {
stompClient.subscribe('/topic/greetings', function (greeting) {
var message = JSON.parse(greeting.body).content;
document.getElementById("greetings").innerHTML += "<p>" + message + "</p>";
});
stompClient.send("/app/hello", {}, "World");
});
</script>
</body>
</html>在這個 HTML 頁面中，我們首先通過 SockJS 連接到 /ws 端點，然后通過 Stomp.js 發送和接收消息。當頁面加載時，它會向服務器發送一個 /hello 消息，并訂閱 /topic/greetings 主題來接收服務器的回復。
測試 WebSocket 消息推送
現在，一切都已經準備好了，我們可以啟動 Spring Boot 應用，并打開瀏覽器訪問上面創建的 HTML 頁面。當你打開頁面時，你應該能看到頁面上顯示 “Hello, World!” 的消息，這就是從服務器通過 WebSocket 實時推送到客戶端的消息。
結語
雖然 WebSocket 已經是個“老將”了，但它在實時通信領域的地位依然不可撼動。通過上面的示例，我們可以看到使用 Java 和 Spring Boot 實現 WebSocket 消息推送是多么簡單和高效。當然，隨著技術的發展，我們也需要關注那些新興的實時通信技術和工具，但 WebSocket 作為一個成熟、穩定的解決方案，依然值得我們深入學習和使用。
