你還在使用 WebSocket 實(shí)現(xiàn)實(shí)時(shí)消息推送嗎？
不過，咱們今天不聊那些新興的玩意兒，還是回歸初心，好好聊聊 WebSocket，畢竟它可是實(shí)時(shí)通信領(lǐng)域的“老將”了。我們就用 Java 來實(shí)現(xiàn)一個(gè)簡單的 WebSocket 消息推送示例，讓大家看看 WebSocket 的魅力所在。
WebSocket 是什么？
WebSocket 是一種網(wǎng)絡(luò)通信協(xié)議，它提供了一種在單個(gè) TCP 連接上進(jìn)行全雙工通訊的渠道。簡單來說，就是客戶端和服務(wù)器之間只需要建立一個(gè)連接，就可以通過這個(gè)連接雙向傳輸數(shù)據(jù)，而且數(shù)據(jù)可以隨時(shí)發(fā)送，不需要像 HTTP 那樣每次請(qǐng)求都要建立連接，大大提高了通信效率。
為什么選擇 WebSocket？
- 實(shí)時(shí)性：WebSocket 最大的特點(diǎn)就是實(shí)時(shí)性，數(shù)據(jù)可以隨時(shí)從服務(wù)器發(fā)送到客戶端，非常適合需要實(shí)時(shí)通信的應(yīng)用場景，比如在線聊天、實(shí)時(shí)通知等。
- 減少資源消耗：由于 WebSocket 只需要建立一次連接，就可以一直保持通信狀態(tài)，所以相比 HTTP 的多次請(qǐng)求/響應(yīng)模式，WebSocket 大大減少了資源的消耗。
- 更好的二進(jìn)制支持：WebSocket 提供了對(duì)二進(jìn)制數(shù)據(jù)的支持，這使得它在處理一些復(fù)雜的數(shù)據(jù)類型時(shí)更加高效。
Java 實(shí)現(xiàn) WebSocket 消息推送
要用 Java 實(shí)現(xiàn) WebSocket 消息推送，我們可以使用 Spring Boot 框架，它提供了對(duì) WebSocket 的良好支持。下面是一個(gè)簡單的示例：
添加依賴
首先，在你的 Spring Boot 項(xiàng)目的 pom.xml 文件中添加 WebSocket 的依賴：
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>配置 WebSocket
然后，創(chuàng)建一個(gè)配置類來啟用和配置 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");
}
}這里我們注冊(cè)了一個(gè) /ws 的 WebSocket 端點(diǎn)，并啟用了 SockJS 作為備選方案，以支持那些不支持 WebSocket 的瀏覽器。同時(shí)，我們還配置了消息代理，用于將消息從服務(wù)器推送到客戶端。
創(chuàng)建消息處理器
接下來，創(chuàng)建一個(gè)消息處理器來處理 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 + "!";
}
}在這個(gè)處理器中，我們定義了一個(gè) /hello 的消息映射，當(dāng)接收到這個(gè)消息時(shí)，處理器會(huì)將其處理并發(fā)送到 /topic/greetings 主題下。
客戶端連接 WebSocket
最后，我們需要在客戶端連接到 WebSocket 服務(wù)器，并訂閱相應(yīng)的主題來接收消息。這里以瀏覽器為例，使用 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>在這個(gè) HTML 頁面中，我們首先通過 SockJS 連接到 /ws 端點(diǎn)，然后通過 Stomp.js 發(fā)送和接收消息。當(dāng)頁面加載時(shí)，它會(huì)向服務(wù)器發(fā)送一個(gè) /hello 消息，并訂閱 /topic/greetings 主題來接收服務(wù)器的回復(fù)。
測試 WebSocket 消息推送
現(xiàn)在，一切都已經(jīng)準(zhǔn)備好了，我們可以啟動(dòng) Spring Boot 應(yīng)用，并打開瀏覽器訪問上面創(chuàng)建的 HTML 頁面。當(dāng)你打開頁面時(shí)，你應(yīng)該能看到頁面上顯示 “Hello, World!” 的消息，這就是從服務(wù)器通過 WebSocket 實(shí)時(shí)推送到客戶端的消息。
結(jié)語
雖然 WebSocket 已經(jīng)是個(gè)“老將”了，但它在實(shí)時(shí)通信領(lǐng)域的地位依然不可撼動(dòng)。通過上面的示例，我們可以看到使用 Java 和 Spring Boot 實(shí)現(xiàn) WebSocket 消息推送是多么簡單和高效。當(dāng)然，隨著技術(shù)的發(fā)展，我們也需要關(guān)注那些新興的實(shí)時(shí)通信技術(shù)和工具，但 WebSocket 作為一個(gè)成熟、穩(wěn)定的解決方案，依然值得我們深入學(xué)習(xí)和使用。
