第3篇：Spring Boot + WebSocket + 消息队列STOMP协议发布订阅模式 实现多频道实时消息广播

基于发布订阅主题 模式实现原理图前提环境已经搭建好具体看第二篇1 maven依赖springboot 版本3.3.1!-- 核心包 -- dependency groupIdorg.springframework.boot/groupId artifactIdspring-boot-starter-websocket/artifactId /dependency !-- 必须添加用于 STOMP Relay 底层的 TCP 通信 -- dependency groupIdorg.springframework.boot/groupId artifactIdspring-boot-starter-reactor-netty/artifactId /dependency dependency groupIdorg.projectlombok/groupId artifactIdlombok/artifactId scopeprovided/scope /dependency2 核心类 WebSocketConfig 代码这里设置了2个模式 内存 和 外部 Rabbitmq本地开发调试可以用内存模式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; /** * WebSocket 配置类用于配置基于 STOMP 协议的 WebSocket 消息代理。 * 本配置支持前端通过 SockJS 连接到后端并使用简单内存消息代理Simple Broker进行消息路由。 */ Configuration EnableWebSocketMessageBroker // 启用 WebSocket 消息代理功能支持 STOMP 协议 public class WebSocketConfig implements WebSocketMessageBrokerConfigurer { /** * 配置消息代理Message Broker用于处理客户端订阅和消息转发。 * * param registry 消息代理注册器 */ Override public void configureMessageBroker(MessageBrokerRegistry registry) { // 【方式一使用内置的简单内存消息代理Simple Broker】 // 适用于开发或轻量级场景不依赖外部消息中间件如 RabbitMQ、ActiveMQ。 // 客户端可订阅以 /topic 或 /queue 开头的目的地服务端可通过 SimpMessagingTemplate 发送消息。 //registry.enableSimpleBroker(/topic, /queue); // 【方式二使用外部 STOMP 消息代理如 RabbitMQ、ActiveMQ——已注释】 // 若需在生产环境中使用高可用、持久化、集群等能力应启用 STOMP Relay 并连接到外部消息中间件。 // 下面是启用 STOMP 中继代理Relay的示例配置当前被注释 registry.enableStompBrokerRelay(/topic, /queue) .setRelayHost(localhost) // 外部消息代理如 ActiveMQ/RabbitMQ的主机地址 .setRelayPort(61613) // STOMP 协议端口ActiveMQ 默认为 61613RabbitMQ 需启用插件 .setClientLogin(admin) // 客户端连接消息代理时使用的用户名 .setClientPasscode(admin) // 客户端连接消息代理时使用的密码 .setSystemLogin(admin) // 系统服务端连接消息代理时使用的用户名 .setSystemPasscode(admin); // 系统服务端连接消息代理时使用的密码 // 设置应用前缀所有以 /app 开头的消息将被路由到 MessageMapping 注解的方法 // 例如前端发送消息到 /app/chat.sendMessage会由带有 MessageMapping(/chat.sendMessage) 的方法处理 registry.setApplicationDestinationPrefixes(/app); } /** * 注册 STOMP 协议的 WebSocket 端点供客户端连接。 * * param registry STOMP 端点注册器 */ Override public void registerStompEndpoints(StompEndpointRegistry registry) { // 添加一个名为 /ws 的 WebSocket 端点 // 前端可通过 new SockJS(http://host:port/ws) 进行连接 registry.addEndpoint(/ws) .setAllowedOriginPatterns(http://192.168.0.100:51823) // 允许跨域来源 .withSockJS(); // 启用 SockJS 作为 WebSocket 的降级兼容方案支持不支持 WebSocket 的浏览器 } }3 处理前端通过 STOMP 协议发送的聊天消息import com.pig4cloud.ai.bi6.config.WebSocketConfig; import com.pig4cloud.ai.bi6.pojo.ChatMessage; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.messaging.handler.annotation.MessageMapping; import org.springframework.messaging.simp.SimpMessagingTemplate; import org.springframework.stereotype.Controller; /** * WebSocket 聊天控制器用于处理前端通过 STOMP 协议发送的聊天消息。 * 本控制器接收以 /app/chat.send 为目的地的消息因 WebSocketConfig 中设置了 /app 前缀 * 并将消息广播到指定的频道如 /topic/topic_1。 */ Controller public class WebSocketChatController { //Spring 提供的模板类用于向 WebSocket 客户端发送消息 Autowired private SimpMessagingTemplate messagingTemplate; /** * 接收前端发送的消息 * * SimpMessageHeaderAccessor 消息头访问器 * * param message */ MessageMapping(/chat.send) public void handleChatMessage(ChatMessage message, SimpMessageHeaderAccessor headerAccessor) { //如果 握手拦截器HandshakeInterceptor 有存入数据这里可以取过来使用 //System.out.println(headerAccessor.getSessionAttributes().get(xxxx)); // 假设前端传入了 channel 字段 (如 topic_1) String targetChannel message.getChannel(); if (targetChannel null || targetChannel.isEmpty()) { targetChannel topic_1; // 默认频道 } // 动态构建 destination: /topic/topic_1 String destination /topic/ targetChannel; // 发送给所有订阅了该频道的人 messagingTemplate.convertAndSend(destination, message); System.out.println(Message sent to destination : message.getContent()); } /** * 也可以监听特定用户的消息 (私聊逻辑预留) * MessageMapping(/chat.private) * SendToUser(/queue/private) * public ... */ }4 向 WebSocket 客户端发送消息import com.pig4cloud.ai.bi6.pojo.ChatMessage; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.messaging.simp.SimpMessagingTemplate; import org.springframework.web.bind.annotation.*; RestController RequestMapping(/api) public class ChatApiController { //Spring 提供的模板类用于向 WebSocket 客户端发送消息 Autowired private SimpMessagingTemplate messagingTemplate; /** * 测试接口调用此接口后端会主动向所有前端推送消息 */ GetMapping(/push/{zhuti}) public String pushMessage(PathVariable(zhuti) String zhuti, RequestParam(defaultValue 我是后端主动推送的消息) String content) { ChatMessage msg new ChatMessage(Server, content); // 核心代码convertAndSend(目的地, 消息对象) // 目的地必须和前端订阅的一致 (/topic/public) messagingTemplate.convertAndSend(/topic/zhuti, msg); return 消息已推送到指定topic请查看前端页面。; } }5 其它文件 pojo前台在 第3篇 附录import lombok.Data; Data public class ChatMessage { private String from; // 发送者 private String content; // 内容 private Long timestamp; // 时间戳 private String channel; // 频道,主题 public ChatMessage() {} public ChatMessage(String from, String content) { this.from from; this.content content; this.timestamp System.currentTimeMillis(); } }