[超詳細]SpringBoot整合WebSocket

1. 什麼是WebSocket？

WebSocket 是一種在單個 TCP 連接上進行全雙工通信的協議，它允許在瀏覽器和伺服器之間進行實時的、雙向的通信。相對於傳統的基於請求和響應的 HTTP 協議，WebSocket 提供了一種更有效、更實時的通信方式，適用於需要實時更新、實時通知和實時交互的應用。

WebSocket 的一些關鍵特點包括：

1. 全雙工通信： WebSocket 允許伺服器和客戶端在同一連接上同時進行雙向通信。這意味著伺服器可以隨時向客戶端推送數據，而不必等待客戶端發送請求。
2. 持久連接： WebSocket 連接一旦建立，會一直保持打開狀態，不會像傳統的 HTTP 連接那樣在每次請求和響應之後關閉。這減少了每次連接建立和關閉的開銷，使通信更加高效。
3. 低延遲： 由於連接保持打開狀態，WebSocket 通信具有較低的延遲，適用於實時性要求較高的應用，如實時聊天、實時數據更新等。
4. 少量的數據交換： 與 HTTP 請求和響應相比，WebSocket 數據交換的開銷較小。WebSocket 的幀頭相對較小，因此有效載荷的比例更高。
5. 相容性： 現代瀏覽器和大多數伺服器支持 WebSocket。此外，WebSocket 協議還定義了一個子協議 STOMP（Streaming Text Oriented Messaging Protocol），用於更高級的消息傳遞和訂閱功能。
6. 安全性： 與其他網路通信協議一樣，WebSocket 通信也需要一些安全性的考慮。可以使用加密協議（如 TLS）來保護數據在網路傳輸中的安全性。

2. 代碼實戰

2.1 SpringBoot導入依賴

在pom.xml中導入以下依賴，版本由SpringBoot管理

<!-- websocket -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

2.2 創建配置類

創建WebSocketConfig配置類，並將其註入到Bean容器中

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@Configuration
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }
}

2.3 創建WebSocketServer類

創建WebSocketServer類，並將其註入到Bean容器中

註意：@ServerEndpoint("/WebSocket")，該註解用於配置建立WebSocket連接的路徑，可以按需修改。

該類一般擁有以下功能：

1. WebSocket 端點註冊： WebSocket 伺服器需要註冊一個或多個 WebSocket 端點（Endpoints）。每個端點對應一種處理邏輯，可以處理客戶端發送過來的消息，以及向客戶端發送消息。這些端點通過註解或配置來定義。
2. 建立和維護連接： WebSocket 伺服器負責監聽客戶端的連接請求，一旦有客戶端連接，伺服器會創建一個 WebSocket 會話（Session）來管理這個連接。伺服器需要能夠維護這些連接，包括打開、關閉、保持心跳等操作。
3. 消息處理： 一旦客戶端連接成功，WebSocket 伺服器需要處理客戶端發送過來的消息。這可以在 WebSocket 端點中的方法上定義處理邏輯。伺服器可以根據不同的業務需求處理不同類型的消息。
4. 處理異常： 與任何網路通信一樣，WebSocket 連接可能會面臨各種異常情況，如斷開連接、網路問題等。WebSocket 伺服器需要能夠處理這些異常情況，進行適當的清理和處理。

可以將該類理解為WebSocket生命周期中會調用的方法。

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;

@Slf4j
@Component
@ServerEndpoint("/WebSocket")
public class WebSocketServer {

    private Session session;

    @OnOpen
    public void onOpen(Session session) {
        this.session = session;
        WebSocketManager.sentToUser(session, "WebSocket is connected!");
        WebSocketManager.addWebSocketServer(this);
        log.info("與SessionId：{}建立連接", session.getId());
    }

    @OnClose
    public void onClose() {
        WebSocketManager.removeWebSocketServer(this);
        log.info("WebSocket連接關閉");
    }

    @OnMessage
    public void onMessage(String message, Session session) {
        log.info("來自SessionId:{}的消息:{}", session.getId(), message);
    }

    @OnError
    public void onError(Session session, Throwable error) {
        log.error("Session:{}的WebSocket發生錯誤", session.getId(), error);
    }

    public Session getSession() {
        return session;
    }

    public String getSessionId() {
        return session.getId();
    }
}

2.4 創建WebSocketServer管理類

該類用於管理WebSocketServer（其實主要是管理Session），如果不需要發送消息給特定用戶，那麼無需創建該類，在WebSocketServer類中維護一個類變數即可。

import lombok.extern.slf4j.Slf4j;

import javax.websocket.Session;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArraySet;

@Slf4j
public class WebSocketManager {

    private final static CopyOnWriteArraySet<WebSocketServer> webSocketServerSet = new CopyOnWriteArraySet<>();

    private final static ConcurrentHashMap<String, WebSocketServer> webSocketServerMap = new ConcurrentHashMap<>();

    public static void addWebSocketServer(WebSocketServer webSocketServer){
        if (webSocketServer != null){
            webSocketServerSet.add(webSocketServer);
            webSocketServerMap.put(webSocketServer.getSessionId(), webSocketServer);
        }
    }

    public static void removeWebSocketServer(WebSocketServer webSocketServer){
        webSocketServerSet.remove(webSocketServer);
        webSocketServerMap.remove(webSocketServer.getSessionId());
    }

    /**
     * 通過SessionId發送消息給特定用戶
     * @param
     * @param msg
     */
    public static void sentToUser(String sessionId, String msg){
        Session session = webSocketServerMap.get(sessionId).getSession();
        sentToUser(session, msg);
    }

    /**
     * 通過Session發送消息給特定用戶
     * @param session
     * @param msg
     */
    public static void sentToUser(Session session, String msg){
        if (session == null){
            log.error("不存在該Session，無法發送消息");
            return;
        }
        session.getAsyncRemote().sendText(msg);
    }

    /**
     * 發送消息給所有用戶
     * @param msg
     */
    public static void sentToAllUser(String msg){
        for (WebSocketServer webSocketServer : webSocketServerSet) {
            sentToUser(webSocketServer.getSession(), msg);
        }
        log.info("向所有用戶發送WebSocket消息完畢，消息：{}", msg);
    }
}

3. 測試

使用Postman等工具進行WebSocket連接，連接路徑為WebSocket（與前文的@ServerEndpoint內容一致）。建立連接後會收到"WebSocket is connected!"

4. 後續完善

如果你的項目中有登錄功能，那麼應該在你的登錄時將用戶的Session記錄下來。你可以將Session記錄在本地緩存、Redis和Grava緩存等能夠建立關聯關係的地方。