在Spring Boot中,可以使用WebSocket实现点对点发送消息。WebSocket是一种在单个TCP连接上进行全双工通信的协议,它可以在Web浏览器和服务器之间建立一个持久连接,使得服务器可以主动向客户端推送数据。
以下是使用Spring Boot实现点对点发送消息的步骤:
pom.xml
文件中添加以下依赖:<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(new WebSocketHandler(), "/ws").setAllowedOrigins("*");
}
}
上述示例代码中,WebSocketHandler()
是自定义的WebSocket处理器,/ws
是WebSocket的端点路径。
public class WebSocketHandler extends TextWebSocketHandler {
private static final List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();
@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
sessions.add(session);
}
@Override
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
// 处理收到的消息
for (WebSocketSession s : sessions) {
s.sendMessage(message);
}
}
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
sessions.remove(session);
}
}
上述示例代码中,afterConnectionEstablished()
方法在WebSocket连接建立时调用,handleTextMessage()
方法用于处理收到的文本消息,afterConnectionClosed()
方法在WebSocket连接关闭时调用。
@RestController
@RequestMapping("/api")
public class MessageController {
@Autowired
private SimpMessagingTemplate messagingTemplate;
@PostMapping("/send/{userId}")
public void sendMessage(@PathVariable("userId") String userId, @RequestBody String message) {
messagingTemplate.convertAndSendToUser(userId, "/ws", message);
}
}
上述示例代码中,sendMessage()
方法用于发送消息到指定的WebSocket连接。
至此,点对点发送消息的实现就完成了。你可以通过向/api/send/{userId}
发送POST请求,传递userId
和message
参数来发送消息。
请注意,上述示例只是演示了基本的点对点发送消息的实现方式。在实际应用中,你可能需要根据具体的需求进行进一步的开发和优化。
辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
推荐阅读: springboot怎么覆盖jar中的配置