SpringBoot WebSocket消息推送（群发和指定到个人或多人）

WebSocket是Web交互技术的一种，它允许服务器与客户端之间进行全双工通信，即服务器不仅可以向客户端推送信息，客户端也可以主动向服务器发送数据。在Spring Boot框架下，WebSocket的使用大大简化了开发过程，使得实时性需求的应用变得更为便捷。本文将深入探讨如何在Spring Boot项目中实现WebSocket的消息推送，包括群发和指定到个人或多人。 我们需要在Spring Boot项目中引入WebSocket的相关依赖。通常，我们会添加`spring-websocket`和`spring-messaging`两个依赖，这两个依赖包含了WebSocket所需的基本功能和消息处理支持。在`pom.xml`文件中，添加如下代码： ```xml <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-messaging</artifactId> </dependency> ``` 接下来，我们需要配置WebSocket的相关设置。在Spring Boot的配置类中，创建一个`WebSocketConfig`类，实现`WebSocketConfigurer`接口，并重写`registerStompEndpoints`和`configureMessageBroker`方法： ```java @Configuration @EnableWebSocketMessageBroker public class WebSocketConfig implements WebSocketMessageBrokerConfigurer { @Override public void registerStompEndpoints(StompEndpointRegistry registry) { // 注册WebSocket端点，设置访问路径为/ws registry.addEndpoint("/ws").withSockJS(); } @Override public void configureMessageBroker(MessageBrokerRegistry config) { // 配置消息代理，使/stomp/*前缀的消息能够被代理到/queue/和/topic/前缀的队列或主题 config.enableSimpleBroker("/topic", "/queue"); // 设置应用级的消息前缀，用于处理用户直接发送给服务器的消息 config.setApplicationDestinationPrefixes("/app"); } } ``` 在以上配置中，我们定义了一个WebSocket的访问路径`/ws`，并使用SockJS技术来提供跨域支持和多种传输协议的兼容。同时，我们设置了消息代理，允许客户端订阅`/topic`和`/queue`开头的主题。 然后，我们需要创建WebSocket的处理器类，用于处理客户端的连接、断开以及接收和发送消息。创建一个`WebSocketController`类，实现`MessageHandler`接口： ```java @Controller public class WebSocketController { @MessageMapping("/send") @SendTo("/topic/messages") public Message sendMessage(WebSocketMessage message) throws Exception { // 消息处理逻辑，这里只是一个简单的示例 return new Message("接收到的消息：" + message.getContent()); } @SubscribeMapping("/user/{username}") public void addUser(@Header("simpSessionId") String sessionId, @PathVariable String username) { // 用户登录逻辑，将用户与sessionId关联 } @MessageMapping("/user/{username}/notify") public void sendPrivateMessage(@PathVariable String username, String message) { // 发送私信逻辑，找到对应用户并发送消息 } } ``` 在`WebSocketController`中，我们定义了三个方法：`sendMessage`用于处理群发消息，`addUser`用于处理用户登录并记录其sessionId，`sendPrivateMessage`则用于指定用户发送私信。 客户端使用WebSocket时，通常会借助Stomp.js库，通过WebSocket与服务器建立连接，并订阅或发布消息。例如，客户端可以订阅`/topic/messages`主题来接收群发消息，通过`/app/user/{username}/notify`向服务器发送私信。 Spring Boot结合WebSocket可以轻松构建实时交互的应用，无论是群发消息还是指定到个人或多人的消息推送，都提供了高效且易于实现的解决方案。通过合理的配置和控制器设计，我们可以实现丰富的实时通信场景，满足各类业务需求。在实际开发中，还可以根据具体需求对连接管理、错误处理、安全性等方面进行更深入的定制。