SpringBoot从0-1集成STOMP协议快速实现消息转发-编程阁

✨重磅！盹猫的个人小站正式上线啦～诚邀各位技术大佬前来探秘！✨
这里有：
硬核技术干货：编程技巧、开发经验、踩坑指南，带你解锁技术新姿势！
趣味开发日常：代码背后的脑洞故事、工具测评，让技术圈不再枯燥～
独家资源分享：开源项目、学习资料包，助你打怪升级快人一步！
👉点击直达→盹猫猫的个人小站👈
🌟 来逛逛吧，说不定能挖到你正在找的技术宝藏哦～

🎨 前言

❓ 什么是STOMP协议？

🌲 环境准备

Pom依赖

🐎 代码步骤

1.配置类

2.用户认证

3. 消息转发

⭕ 总结

欢迎来到盹猫(>^ω^<)的博客
本篇文章主要介绍了
[SpringBoot从0-1集成STOMP协议快速实现消息转发]
❤博主广交技术好友，喜欢文章的可以关注一下❤

🎨 前言

在即时的消息通信应用开发时，经常需要用到将消息转发给特定人员或群组的功能。这时如果使用WebSocket进行开发，就需要维护发送者和接收者的对应关系，当用户过多时，维护这个关系表会是很大的成本。这时我们就可以使用基于Websocket的STOMP协议，来进行消息的转发处理。

本篇文章就是记录SpringBoot如何集成STOMP协议转发的，但在此之前先来看看什么是STOMP协议。

❓ 什么是STOMP协议？

WebSocket 是浏览器与服务器之间的一种双向通信协议（属于 TCP 层之上的应用层协议），解决了 HTTP “请求 - 响应” 单向通信的痛点。

STOMP（Simple Text Oriented Messaging Protocol，简单文本定向消息协议）基于WebSocket协议的应用层的消息规范，可以理解为 “给 WebSocket 通信制定的‘普通话，在 WebSocket 底层通道之上，定义了统一的消息格式、交互规则、路由机制。

来看看如何集成吧！

Let‘s GO！

🌲 环境准备

Pom依赖

要使用STOMP协议需要用到spring-boot-starter-websocket依赖，在Pom文件的dependencies中添加以下内容：

<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency>

🐎 代码步骤

示例中我们实现如下功能：

用户可以连接到ws://127.0.0.1:8080/ws/plot并完成用户认证,并可订阅/topic/plot/{ID}路径，当用户发布数据到/send/plot/{ID}路径时，将数据转发到所有的订阅了/topic/plot/{ID}的用户。

1.配置类

首先配置类实现WebSocketMessageBrokerConfigurer接口，并使用@EnableWebSocketMessageBroker进行标记这是一个默认的Websocket消息代理，并实现以下方法：

registerStompEndpoints：在该方法中可以配置作为STOMP的路径，并可配置允许跨域访问。（即用户的WebSocket需要连接这个路径）

configureMessageBroker: 在该方法中可以设置作为订阅的代理路径。（即用户可以在这个路径上订阅）

configureClientInboundChannel：配置消息入站通道，如可以在这里配置拦截器对发起的请求进行认证。（这里是自定义的StompTokenValidateInterceptor类作为拦截器）

具体代码如下：

package com.uav.qd.config; import com.uav.qd.interceptor.StompTokenValidateInterceptor; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.Configuration; import org.springframework.messaging.simp.config.ChannelRegistration; 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; /** * @author nodcat * @version 1.0 * @since 2025/12/29 下午4:44 */ @Configuration @EnableWebSocketMessageBroker public class WebSocketPlotConfig implements WebSocketMessageBrokerConfigurer { @Autowired StompTokenValidateInterceptor stompTokenValidateInterceptor; /** * 注册STOMP端点，供前端连接 */ @Override public void registerStompEndpoints(StompEndpointRegistry registry) { // 标绘协同端点，允许跨域访问，支持WebSocket握手 registry.addEndpoint("/ws/plot") .setAllowedOriginPatterns("*"); } /** * 配置消息代理，实现消息广播 */ @Override public void configureMessageBroker(MessageBrokerRegistry registry) { // 订阅地址 registry.enableSimpleBroker("/topic/plot"); } @Override public void configureClientInboundChannel(ChannelRegistration registration) { // registration.interceptors(stompTokenValidateInterceptor); } }

2.用户认证

在用户认证时需要让StompTokenValidateInterceptor实现ChannelInterceptor接口，并重写preSend方法，这里的认证就可以自定义了，包括从请求头中获取token(或JWT TOKEN)，然后查数据库（如果是JWT TOKEN则直接解析认证），通过后返回message即可完成连接。

代码如下：

package com.uav.qd.interceptor; import com.uav.qd.common.constant.Constant; import com.uav.qd.common.exception.SysException; import com.uav.qd.common.security.SecurityUser; import com.uav.qd.entity.SysUserTokenEntity; import com.uav.qd.service.ProjectService; import com.uav.qd.service.SysShiroService; import com.uav.qd.service.SysUserService; import org.apache.commons.lang3.StringUtils; import org.jetbrains.annotations.NotNull; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.messaging.Message; import org.springframework.messaging.MessageChannel; import org.springframework.messaging.simp.stomp.StompCommand; import org.springframework.messaging.simp.stomp.StompHeaderAccessor; import org.springframework.messaging.support.ChannelInterceptor; import org.springframework.messaging.support.MessageHeaderAccessor; import org.springframework.stereotype.Component; import java.util.Objects; /** * @author nodcat * @version 1.0 * @since 2026/1/4 上午11:43 */ @Component public class StompTokenValidateInterceptor implements ChannelInterceptor { @Autowired SysShiroService shiroService; private static final String AUTHORIZATION_HEADER = "token"; @Override public Message<?> preSend(@NotNull Message<?> message, MessageChannel channel) { // 1. 获取 STOMP 头访问器，用于读取 STOMP 协议头 StompHeaderAccessor accessor = MessageHeaderAccessor.getAccessor(message, StompHeaderAccessor.class); if (accessor == null) { return message; } if (StompCommand.CONNECT.equals(accessor.getCommand())) { // 3. 提取 STOMP 连接头中的 Token String token = accessor.getFirstNativeHeader(AUTHORIZATION_HEADER); if (token == null || token.trim().isEmpty()) { throw new SysException("STOMP认证失败：未携带有效Token"); } //这里是查询数据库获取，可根据实际需要修改逻辑 SysUserTokenEntity userEntity = shiroService.getByToken(token); if (userEntity==null){ throw new SysException("STOMP认证失败：Token不存在或已失效"); } Long userId = userEntity.getUserId(); Objects.requireNonNull(accessor.getSessionAttributes()).put(Constant.USER_KEY,userId); } return message; } }

⚠️ 注意：这里可以将用户id等数据放到accessor中，以方便在后面的收到消息中直接获取这些关键数据。

3. 消息转发

消息转发时只需要使用到两个注解进行消息的转发：

@SendTo("/topic/plot/{projectId}")：将消息转发到订阅了该主题的用户。

@MessageMapping("/send/plot/{projectId}")：接收该路径发布的消息。

package com.uav.qd.controller.dash; import com.uav.qd.common.constant.Constant; import com.uav.qd.common.exception.SysException; import com.uav.qd.common.security.SecurityUser; import com.uav.qd.dto.emplan.EmPlanMessageDTO; import com.uav.qd.dto.emplan.OperationMessageDTO; import com.uav.qd.service.PlotProjectService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.messaging.handler.annotation.DestinationVariable; import org.springframework.messaging.handler.annotation.MessageMapping; import org.springframework.messaging.handler.annotation.SendTo; import org.springframework.messaging.simp.SimpMessageHeaderAccessor; import org.springframework.stereotype.Controller; import java.util.Map; import java.util.Objects; /** * STOMP方案消息处理器 * @author nodcat * @version 1.0 * @since 2025/12/30 上午10:11 */ @Controller public class DashPlotMessageController { @Autowired PlotProjectService plotProjectService; /** * 向其它用户转发当前项目编辑消息 * @param projectId 项目ID * @param message 消息 * @param accessor 认证消息 * @return 共享操作消息 */ @SendTo("/topic/plot/{projectId}") @MessageMapping("/send/plot/{projectId}") public EmPlanMessageDTO handlePlotMessage( @DestinationVariable String projectId, OperationMessageDTO message, SimpMessageHeaderAccessor accessor ) throws Exception { Long userId = getUserId(accessor); //更新缓存内容 plotProjectService.updateCacheData(projectId,message); return EmPlanMessageDTO.builder() .userId(userId) .projectId(projectId) .operate(message.getOperate()) .data(message.getData()) .build(); } /** * 获取认证消息内的用户ID * @param accessor 认证消息 * @return 用户ID */ public Long getUserId(SimpMessageHeaderAccessor accessor){ Map<String, Object> attributes = accessor.getSessionAttributes(); return (Long) Objects.requireNonNull(attributes).get(Constant.USER_KEY); } }

整个写法与RestController的写法类似，可以根据实际需要对消息进行缓存操作等，这里返回的类会以Json字符串的格式返回。