Spring Boot实现WebSocket详解

文章目录

Spring Boot实现WebSocket详解

一、引言

在现代Web开发中,实时通信成为了提升用户体验的关键技术之一。WebSocket作为一种在单个TCP连接上进行全双工通信的协议,允许服务器主动向客户端发送消息,非常适合需要实时数据交换的应用场景,如在线聊天、实时通知等。Spring Boot通过集成Spring WebSocket模块,提供了一种简洁的方式来实现WebSocket通信。本文将详细介绍如何在Spring Boot中实现WebSocket,包括配置、编码以及实现实时消息传递和群发消息。

二、Spring Boot集成WebSocket

1、添加依赖

首先,我们需要在Spring Boot项目的pom.xml文件中添加WebSocket的依赖。

xml 复制代码
<dependencies>
    <!-- Spring Boot Web依赖 -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <!-- Spring Boot WebSocket依赖 -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-websocket</artifactId>
    </dependency>
</dependencies>

2、配置WebSocket

接下来,我们需要创建一个配置类来设置WebSocket的消息代理和端点。

java 复制代码
package com.example.demo.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.WebSocketMessageBrokerConfigurer;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/topic");
        config.setApplicationDestinationPrefixes("/app");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/websocket-endpoint").withSockJS();
    }
}

在上述配置中,我们启用了简单的消息代理,并设置了应用前缀为/app。同时,我们注册了一个STOMP端点,并启用了SockJS支持。

3、创建消息模型

定义一个简单的消息模型,用于在WebSocket通信中传递消息。

java 复制代码
package com.example.demo.model;

public class Message {
    private String content;
    private String sender;

    // 构造函数、getter和setter方法
    public Message() {}

    public Message(String content, String sender) {
        this.content = content;
        this.sender = sender;
    }

    public String getContent() {
        return content;
    }

    public void setContent(String content) {
        this.content = content;
    }

    public String getSender() {
        return sender;
    }

    public void setSender(String sender) {
        this.sender = sender;
    }
}

4、创建消息处理器

创建一个控制器来处理WebSocket消息。

java 复制代码
package com.example.demo.handler;

import com.example.demo.model.Message;
import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.stereotype.Controller;

@Controller
public class MessageHandler {

    @MessageMapping("/send")
    @SendTo("/topic/messages")
    public Message sendMessage(Message message) {
        return new Message("Hello, WebSocket!", "System");
    }
}

在这个控制器中,我们定义了一个sendMessage方法,它将处理发送到/send目的地的消息,并将响应发送到/topic/messages主题。

5、前端实现

创建一个简单的HTML页面和JavaScript代码,用于连接WebSocket服务器并发送消息。

html 复制代码
<!DOCTYPE html>
<html>
<head>
    <title>WebSocket Demo</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>
        <input type="text" id="messageInput" placeholder="Type a message...">
        <button onclick="sendMessage()">Send</button>
    </div>
    <ul id="messagesList"></ul>
    <script>
        var socket = new SockJS('/websocket-endpoint');
        var stompClient = Stomp.over(socket);
        stompClient.connect({}, function (frame) {
            stompClient.subscribe('/topic/messages', function (message) {
                var messagesList = document.getElementById('messagesList');
                var listItem = document.createElement('li');
                listItem.textContent = message.body;
                messagesList.appendChild(listItem);
            });
        });
        function sendMessage() {
            var messageInput = document.getElementById('messageInput');
            stompClient.send('/app/send', {}, messageInput.value);
            messageInput.value = '';
        }
    </script>
</body>
</html>

在上述代码中,我们创建了一个HTML页面,用户可以通过输入框发送消息,并通过JavaScript连接到WebSocket服务器。

三、实现实时消息传递和群发消息

1、实时消息传递

为了实现实时消息传递,我们可以在MessageHandler中添加一个方法来接收客户端发送的消息。

java 复制代码
@MessageMapping("/receive")
@SendTo("/topic/messages")
public Message receiveMessage(Message message) {
    return new Message("Received your message: " + message.getContent(), "System");
}

2、群发消息

WebSocketConfig中配置广播代理,以实现群发消息。

java 复制代码
@Override
public void configureMessageBroker(MessageBrokerRegistry config) {
    config.enableSimpleBroker("/topic");
    config.setApplicationDestinationPrefixes("/app");
    config.setBrokerDeliveryMode(MessageBrokerRegistry.BrokerDeliveryMode.BROADCASTING);
}

通过设置BrokerDeliveryModeBROADCASTING,所有发送到/topic/messages的消息都会被广播到所有连接的客户端。

四、总结

本文详细介绍了如何在Spring Boot中实现WebSocket,包括添加依赖、配置WebSocket、创建消息模型和处理器,以及如何实现实时消息传递和群发消息。通过这些步骤,您可以在Spring Boot应用程序中轻松集成WebSocket,实现实时通信功能。希望本文能够帮助您在开发中更加得心应手。


版权声明:本博客内容为原创,转载请保留原文链接及作者信息。

参考文章

相关推荐
paopaokaka_luck9 分钟前
智能推荐社交分享小程序(websocket即时通讯、协同过滤算法、时间衰减因子模型、热度得分算法)
数据库·vue.js·spring boot·后端·websocket·小程序
程序员NEO25 分钟前
Spring AI 对话记忆大揭秘:服务器重启,聊天记录不再丢失!
人工智能·后端
用户214118326360225 分钟前
惊爆!国内轻松白嫖 Claude Code,编程效率狂飙
后端
iccb101331 分钟前
我是如何实现在线客服系统的极致稳定性与安全性的
前端·javascript·后端
柒七爱吃麻辣烫32 分钟前
八股文系列-----SpringBoot自动配置的流程
java·spring boot·rpc
M1A137 分钟前
Java 面试系列第一弹:基础问题大盘点
java·后端·mysql
沐尘而生1 小时前
【AI智能体】智能音视频-硬件设备基于 WebSocket 实现语音交互
大数据·人工智能·websocket·机器学习·ai作画·音视频·娱乐
夕颜1111 小时前
关于 Cursor 小插曲记录
后端
考虑考虑1 小时前
go中的Map
后端·程序员·go