快捷導(dǎo)航

利用SpringBoot與WebSocket實(shí)現(xiàn)實(shí)時(shí)雙向通信功能

更新時(shí)間：2025年07月29日 09:15:22 作者：bug菌1

隨著互聯(lián)網(wǎng)技術(shù)的飛速發(fā)展,實(shí)時(shí)通信技術(shù)已經(jīng)逐漸成為現(xiàn)代Web應(yīng)用的核心需求,論是在即時(shí)消息系統(tǒng)還是物聯(lián)網(wǎng)(IoT)應(yīng)用中,WebSocket都提供了一種高效、持久的雙向通信方式,本文將深入解析如何利用Spring Boot實(shí)現(xiàn)WebSocket通信,需要的朋友可以參考下

前言

隨著互聯(lián)網(wǎng)技術(shù)的飛速發(fā)展，實(shí)時(shí)通信技術(shù)已經(jīng)逐漸成為現(xiàn)代Web應(yīng)用的核心需求。無(wú)論是在即時(shí)消息系統(tǒng)、在線游戲、金融交易平臺(tái)，還是物聯(lián)網(wǎng)(IoT)應(yīng)用中，WebSocket都提供了一種高效、持久的雙向通信方式。特別是在需要低延遲的場(chǎng)景下，WebSocket的優(yōu)勢(shì)尤為突出。本文將深入解析如何利用Spring Boot實(shí)現(xiàn)WebSocket通信，進(jìn)而結(jié)合STOMP協(xié)議，構(gòu)建一個(gè)功能強(qiáng)大的實(shí)時(shí)聊天室應(yīng)用。

WebSocket協(xié)議簡(jiǎn)介

WebSocket協(xié)議自HTML5引入以來(lái)，成為了現(xiàn)代Web應(yīng)用的重要組成部分。它打破了傳統(tǒng)HTTP協(xié)議只能在請(qǐng)求和響應(yīng)的周期內(nèi)進(jìn)行單向通信的限制，實(shí)現(xiàn)了雙向通信，并保持持久連接。

WebSocket協(xié)議的優(yōu)勢(shì)

低延遲：WebSocket連接是持久性的連接，避免了頻繁的握手過(guò)程。消息從客戶端到服務(wù)器，再到客戶端的延遲極低，非常適合即時(shí)通訊和實(shí)時(shí)推送場(chǎng)景。
節(jié)省帶寬：相比HTTP的請(qǐng)求-響應(yīng)模式，WebSocket連接不需要每次傳輸都重新建立連接，節(jié)省了很多帶寬和計(jì)算資源。
全雙工通信：客戶端和服務(wù)器都可以隨時(shí)發(fā)送數(shù)據(jù)，不需要等待對(duì)方回應(yīng)。這使得實(shí)時(shí)互動(dòng)應(yīng)用變得更加靈活和高效。

WebSocket的應(yīng)用場(chǎng)景

實(shí)時(shí)聊天：例如即時(shí)通訊、客服系統(tǒng)，WebSocket允許即時(shí)傳遞消息。
實(shí)時(shí)數(shù)據(jù)推送：例如股市行情、體育比賽實(shí)時(shí)得分等，WebSocket使得數(shù)據(jù)可以實(shí)時(shí)推送到客戶端。
在線游戲：WebSocket能夠?qū)崿F(xiàn)游戲玩家之間的即時(shí)互動(dòng)和實(shí)時(shí)數(shù)據(jù)同步。
物聯(lián)網(wǎng)（IoT）：WebSocket能夠幫助傳感器和設(shè)備實(shí)時(shí)通信，提供更及時(shí)的數(shù)據(jù)反饋。

Spring Boot中實(shí)現(xiàn)WebSocket

Spring Boot 是一個(gè)開(kāi)箱即用的框架，集成了多種常用功能，幫助開(kāi)發(fā)者快速實(shí)現(xiàn)應(yīng)用的功能。Spring Boot 對(duì) WebSocket 的支持非常全面，允許我們輕松地配置 WebSocket 服務(wù)。

1. 配置 WebSocket 端點(diǎn)

在 Spring Boot 中配置 WebSocket 服務(wù)非常簡(jiǎn)單，我們需要做以下幾步：

創(chuàng)建一個(gè) WebSocket 配置類(lèi)，并使用 @EnableWebSocket 注解啟用 WebSocket 功能。
在配置類(lèi)中注冊(cè) WebSocket 端點(diǎn)，配置客戶端連接路徑。

package com.example.websocket.config;

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        // 注冊(cè)WebSocketHandler，"/chat"是客戶端連接WebSocket時(shí)訪問(wèn)的路徑
        registry.addHandler(new WebSocketHandlerImpl(), "/chat")
                .setAllowedOrigins("*");  // 允許跨域請(qǐng)求
    }
}

2. 創(chuàng)建 WebSocket 處理類(lèi)

WebSocket 處理類(lèi)負(fù)責(zé)接收客戶端發(fā)送的消息并做出響應(yīng)。我們通過(guò)繼承 TextWebSocketHandler 類(lèi)來(lái)實(shí)現(xiàn)消息的處理：

package com.example.websocket.handler;

import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.handler.TextWebSocketHandler;

public class WebSocketHandlerImpl extends TextWebSocketHandler {

    @Override
    public void handleTextMessage(WebSocketSession session, TextMessage message) {
        try {
            // 處理客戶端發(fā)送的消息，并向客戶端發(fā)送回應(yīng)消息
            System.out.println("Received message: " + message.getPayload());
            session.sendMessage(new TextMessage("Echo: " + message.getPayload()));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        System.out.println("WebSocket connection established");
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        System.out.println("WebSocket connection closed");
    }
}

3. 客戶端的 WebSocket 連接

在前端頁(yè)面，我們使用原生 JavaScript 創(chuàng)建 WebSocket 連接?？蛻舳送ㄟ^(guò) new WebSocket() 方法與服務(wù)器建立連接：

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>WebSocket Chat</title>
</head>
<body>
    <h1>WebSocket Chat</h1>
    <div>
        <input type="text" id="messageInput" placeholder="Enter a message">
        <button onclick="sendMessage()">Send</button>
    </div>
    <div id="messages"></div>

    <script>
        const socket = new WebSocket("ws://localhost:8080/chat");

        // 接收服務(wù)器發(fā)送的消息
        socket.onmessage = function(event) {
            const message = event.data;
            const messagesDiv = document.getElementById("messages");
            const newMessage = document.createElement("p");
            newMessage.textContent = message;
            messagesDiv.appendChild(newMessage);
        };

        // 向服務(wù)器發(fā)送消息
        function sendMessage() {
            const messageInput = document.getElementById("messageInput");
            const message = messageInput.value;
            socket.send(message);
            messageInput.value = '';  // 清空輸入框
        }
    </script>
</body>
</html>

在這個(gè)例子中，我們創(chuàng)建了一個(gè)簡(jiǎn)單的聊天室應(yīng)用，前端通過(guò) WebSocket 與后端建立了連接，客戶端輸入消息并通過(guò) WebSocket 發(fā)送給服務(wù)器，服務(wù)器將其原樣返回。

WebSocket 與 STOMP 協(xié)議集成

雖然 WebSocket 本身支持雙向通信，但它的消息處理方式較為基礎(chǔ)，適合一些簡(jiǎn)單的場(chǎng)景。然而，STOMP 協(xié)議為 WebSocket 提供了更加高級(jí)的消息處理機(jī)制，能夠?qū)崿F(xiàn)消息的廣播、訂閱、隊(duì)列處理等。

1. 引入 STOMP 依賴

首先，我們需要在 pom.xml 中引入 spring-boot-starter-websocket 和 spring-messaging 依賴：

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

<dependency>
    <groupId>org.springframework.messaging</groupId>
    <artifactId>spring-messaging</artifactId>
</dependency>

2. WebSocket 配置與 STOMP 協(xié)議啟用

通過(guò) @EnableWebSocketMessageBroker 注解啟用 STOMP 協(xié)議，配置消息代理和 STOMP 端點(diǎn)。

package com.example.websocket.config;

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) {
        // 注冊(cè)STOMP端點(diǎn)，客戶端將通過(guò)這個(gè)端點(diǎn)連接WebSocket
        registry.addEndpoint("/gs-guide-websocket").withSockJS();
    }

    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        // 配置消息代理
        registry.setApplicationDestinationPrefixes("/app");
        registry.enableSimpleBroker("/topic");
    }
}

解釋：

registerStompEndpoints 方法注冊(cè) STOMP 端點(diǎn)，客戶端將通過(guò)該端點(diǎn)連接到 WebSocket 服務(wù)。
configureMessageBroker 方法啟用消息代理，/app 是應(yīng)用消息的前綴，/topic 是消息廣播的主題。

3. WebSocket 控制器

使用 STOMP 協(xié)議時(shí)，我們可以通過(guò) @MessageMapping 注解來(lái)接收客戶端的消息，使用 @SendTo 注解廣播消息：

package com.example.websocket.controller;

import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.stereotype.Controller;

@Controller
public class ChatController {

    @MessageMapping("/chat.sendMessage")
    @SendTo("/topic/public")
    public String sendMessage(String message) throws Exception {
        return message;
    }
}

解釋：

@MessageMapping("/chat.sendMessage") 表示當(dāng)客戶端發(fā)送消息到 /app/chat.sendMessage 時(shí)，方法將被觸發(fā)。
@SendTo("/topic/public") 表示將結(jié)果發(fā)送到 /topic/public 主題，并廣播給所有訂閱該主題的客戶端。

4. 前端頁(yè)面集成 STOMP 協(xié)議

客戶端通過(guò) SockJS 和 Stomp.js 庫(kù)與服務(wù)器建立 WebSocket 連接，并進(jìn)行消息交換：

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>STOMP WebSocket Chat</title>
    <script src="https://cdn.jsdelivr.net/npm/sockjs-client/dist/sockjs.min.js"></script>
    <script src="https://cdn.jsdelivr.net/npm/stompjs"></script>
</head>
<body>
    <h1>STOMP WebSocket Chat</h1>
    <div>
        <input type="text" id="messageInput" placeholder="Enter a message">
        <button onclick="sendMessage()">Send</button>
    </div>
    <div id="messages"></div>

    <script>
        const socket = new SockJS("/gs-guide-websocket");
        const stompClient = Stomp.over(socket);

        stompClient.connect({}, function (frame) {
            console.log('Connected: ' + frame);
            stompClient.subscribe('/topic/public', function (messageOutput) {
                const message = messageOutput.body;
                const messagesDiv = document.getElementById("messages");
                const newMessage = document.createElement("p");
                newMessage.textContent = message;
                messagesDiv.appendChild(newMessage);
            });
        });

        function sendMessage() {
            const messageInput = document.getElementById("messageInput");
            const message = messageInput.value;
            stompClient.send("/app/chat.sendMessage", {}, message);
            messageInput.value = '';
        }
    </script>
</body>
</html>

解釋：

SockJS 提供 WebSocket 降級(jí)支持，確?？蛻舳嗽?WebSocket 不可用時(shí)也能連接。
stompClient.send("/app/chat.sendMessage", {}, message) 發(fā)送消息到服務(wù)器，服務(wù)器將消息廣播到所有訂閱的客戶端。

小結(jié)

WebSocket 是一種持久連接協(xié)議，提供了低延遲的雙向通信，適用于實(shí)時(shí)數(shù)據(jù)傳輸場(chǎng)景。
STOMP 協(xié)議 使 WebSocket 消息的處理更加高效，支持發(fā)布/訂閱模型，適合于構(gòu)建復(fù)雜的實(shí)時(shí)應(yīng)用。
使用 Spring Boot，我們能夠輕松配置 WebSocket 和 STOMP，快速實(shí)現(xiàn)實(shí)時(shí)通信功能。

總結(jié)

通過(guò)本文的學(xué)習(xí)，我們了解了如何在 Spring Boot 中實(shí)現(xiàn) WebSocket 雙向通信，并結(jié)合 STOMP 協(xié)議構(gòu)建一個(gè)簡(jiǎn)單的實(shí)時(shí)聊天應(yīng)用。WebSocket 提供了高效的實(shí)時(shí)通信機(jī)制，而 STOMP 協(xié)議則為 WebSocket 提供了更加靈活的消息推送方式，能夠滿足復(fù)雜應(yīng)用場(chǎng)景的需求。

實(shí)時(shí)通信是現(xiàn)代Web應(yīng)用的重要組成部分，WebSocket 和 STOMP 協(xié)議為我們提供了低延遲、可靠的解決方案。通過(guò)結(jié)合 Spring Boot 的 WebSocket 支持，開(kāi)發(fā)者可以非常方便地構(gòu)建實(shí)時(shí)通信應(yīng)用。掌握這些技術(shù)，可以為你在開(kāi)發(fā)即時(shí)消息系統(tǒng)、在線游戲、物聯(lián)網(wǎng)等應(yīng)用中提供強(qiáng)有力的支持。

希望本文的深入解析能幫助你更好地理解 WebSocket 和 STOMP 協(xié)議的原理，并能夠在實(shí)際項(xiàng)目中應(yīng)用這些技術(shù)！

以上就是利用SpringBoot與WebSocket實(shí)現(xiàn)實(shí)時(shí)雙向通信功能的詳細(xì)內(nèi)容，更多關(guān)于SpringBoot WebSocket雙向通信的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: