SpringBootwebSocket 消息发送与接收

WebSocket是Web应用中实现双向通信的一种技术，它允许服务器主动向客户端推送数据，极大地提高了实时性。在Spring Boot框架中，我们可以轻松集成WebSocket，构建高效的消息推送系统。本篇文章将详细探讨Spring Boot如何实现WebSocket的消息发送与接收。 我们需要在Spring Boot项目中引入WebSocket的相关依赖。在`pom.xml`文件中，添加以下Maven依赖： ```xml <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency> ``` 接下来，我们需要配置WebSocket的相关设置。创建一个名为`WebSocketConfig`的配置类，继承`WebSocketConfigurer`接口，重写`registerWebSocketHandlers`方法，注册WebSocket处理器： ```java @Configuration @EnableWebSocket public class WebSocketConfig implements WebSocketConfigurer { @Override public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { registry.addHandler(new MyWebSocketHandler(), "/ws"); } } ``` 这里我们创建了一个名为`MyWebSocketHandler`的WebSocket处理器，用于处理WebSocket连接的建立、消息发送和接收等操作。这个处理器需要继承`TextWebSocketHandler`，并重写其关键方法： ```java @Service public class MyWebSocketHandler extends TextWebSocketHandler { private Set<WebSocketSession> sessions = new HashSet<>(); @Override public void afterConnectionEstablished(WebSocketSession session) throws Exception { sessions.add(session); System.out.println("新连接建立：" + session.getId()); } @Override protected void handleTextMessage(WebSocketSession session, TextMessage message) throws IOException { System.out.println("接收到消息：" + message.getPayload()); for (WebSocketSession s : sessions) { if (!s.equals(session)) { s.sendMessage(new TextMessage(message.getPayload())); } } } @Override public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception { sessions.remove(session); System.out.println("连接关闭：" + session.getId()); } } ``` 在`afterConnectionEstablished`方法中，我们将新的WebSocket会话添加到会话集合中；在`handleTextMessage`方法中，我们处理接收到的消息，并广播给其他在线用户；在`afterConnectionClosed`方法中，移除已关闭的会话。 客户端与服务器建立WebSocket连接通常通过JavaScript实现，例如： ```javascript let socket = new WebSocket('ws://localhost:8080/ws'); socket.onopen = function(event) { console.log('WebSocket连接已建立'); socket.send('你好，服务器'); }; socket.onmessage = function(event) { console.log('接收到服务器消息：' + event.data); }; socket.onclose = function(event) { console.log('WebSocket连接已关闭'); }; ``` 此外，为了支持WebSocket的鉴权和自定义协议，我们可以使用Spring的`@MessageMapping`注解来处理特定的消息。例如，定义一个接收用户信息的方法： ```java @Controller public class WebSocketController { @MessageMapping("/userInfo") @SendTo("/topic/users") public UserMessage sendUserInfo(UserInfo userInfo) throws Exception { // 处理用户信息并返回给所有用户 return userMessage; } } ``` 这里的`@MessageMapping("/userInfo")`用于处理客户端发送到`/userInfo`的请求，`@SendTo("/topic/users")`则表示将处理后的结果广播到`/topic/users`主题。 Spring Boot结合WebSocket可以方便地实现服务器主动推送消息的功能。通过配置WebSocket处理器、处理客户端连接和消息，以及在控制器中定义消息映射，我们可以构建一个高效的消息推送系统。这个`springbootwebsocket-master`项目提供了完整的示例代码，可以直接下载、导入并运行，以便于理解和实践这些概念。