chore: 优化 WebSocket 相关配置和校验

basic-websocket/src/main/java/com/yiring/websocket/config/WebSocketStompConfig.java

@@ -39,24 +39,26 @@ public class WebSocketStompConfig implements WebSocketMessageBrokerConfigurer {
     final ClientOutboundChannelInterceptor clientOutboundChannelInterceptor;
 
     public static Integer stompPort;
-    public static boolean simpleMode;
+    public static boolean stompMode;
     public static String mode;
 
     @PostConstruct
     public void init() {
         stompPort = Convert.toInt(SpringUtil.getProperty("spring.rabbitmq.stomp-port"));
-        simpleMode = Objects.isNull(stompPort);
-        mode = simpleMode ? "Simple" : "STOMP";
+        stompMode = Objects.nonNull(stompPort);
+        mode = stompMode ? "STOMP" : "Simple";
     }
 
     @Override
     public void registerStompEndpoints(StompEndpointRegistry registry) {
+        // SockJS 连接
         registry
             .addEndpoint("/stomp/sock-js")
             .setAllowedOriginPatterns("*")
             .addInterceptors(new HttpSessionHandshakeInterceptor())
             .withSockJS();
 
+        // 原生 WebSocket 连接
         registry
             .addEndpoint("/stomp/ws")
             .setAllowedOriginPatterns("*")
@@ -68,6 +70,7 @@ public class WebSocketStompConfig implements WebSocketMessageBrokerConfigurer {
     @Override
     public void configureMessageBroker(MessageBrokerRegistry registry) {
         // 启动前先删除掉可能存在的残留STOMP连接缓存数据
+        // FIXME: 没有考虑多服务部署场景，仅单机模式
         redis.del(RedisKey.STOMP_ONLINE_USERS);
         log.info("WebSocket(Mode: {}) clear online user info cache of redis.", mode);
 
@@ -75,14 +78,12 @@ public class WebSocketStompConfig implements WebSocketMessageBrokerConfigurer {
         registry.setUserDestinationPrefix("/user");
         registry.setApplicationDestinationPrefixes("/app");
 
-        if (simpleMode) {
-            // 1. 使用内存方式处理消息
-            registry.enableSimpleBroker("/topic", "/queue");
-        } else {
-            // 2. 使用 RabbitMQ 处理消息（需要安装 STOMP 插件）
+        String[] destinationPrefixes = { "/topic", "/queue" };
+        if (stompMode) {
+            // 1. 使用 RabbitMQ 处理消息（需要安装 STOMP 插件）
             RabbitProperties rabbitProperties = SpringUtil.getBean(RabbitProperties.class);
             registry
-                .enableStompBrokerRelay("/topic", "/queue")
+                .enableStompBrokerRelay(destinationPrefixes)
                 .setRelayPort(stompPort)
                 .setRelayHost(rabbitProperties.getHost())
                 .setVirtualHost(rabbitProperties.getVirtualHost())
@@ -90,6 +91,9 @@ public class WebSocketStompConfig implements WebSocketMessageBrokerConfigurer {
                 .setClientPasscode(rabbitProperties.getPassword())
                 .setSystemLogin(rabbitProperties.getUsername())
                 .setSystemPasscode(rabbitProperties.getPassword());
+        } else {
+            // 2. 使用内存方式处理消息
+            registry.enableSimpleBroker(destinationPrefixes);
         }
 
         log.info("WebSocket(Mode: {}) init messageBroker success.", mode);

basic-websocket/src/main/java/com/yiring/websocket/interceptor/AbstractMessageHandler.java

@@ -18,9 +18,9 @@ import org.springframework.stereotype.Component;
 public class AbstractMessageHandler {
 
     public SimpMessageHeaderAccessor getAccessor(@NonNull Message<?> message) {
-        Class<? extends SimpMessageHeaderAccessor> clazz = WebSocketStompConfig.simpleMode
-            ? SimpMessageHeaderAccessor.class
-            : StompHeaderAccessor.class;
+        Class<? extends SimpMessageHeaderAccessor> clazz = WebSocketStompConfig.stompMode
+            ? StompHeaderAccessor.class
+            : SimpMessageHeaderAccessor.class;
         return MessageHeaderAccessor.getAccessor(message, clazz);
     }
 

basic-websocket/src/main/java/com/yiring/websocket/interceptor/ClientInboundChannelInterceptor.java

@@ -19,6 +19,7 @@ import org.springframework.messaging.simp.SimpMessageHeaderAccessor;
 import org.springframework.messaging.support.ChannelInterceptor;
 import org.springframework.messaging.support.NativeMessageHeaderAccessor;
 import org.springframework.stereotype.Component;
+import org.springframework.util.Assert;
 
 /**
  * ClientInboundChannelInterceptor
@@ -43,7 +44,7 @@ public class ClientInboundChannelInterceptor implements ChannelInterceptor {
     @Override
     public Message<?> preSend(@NonNull Message<?> message, @NonNull MessageChannel channel) {
         SimpMessageHeaderAccessor accessor = handler.getAccessor(message);
-        assert accessor != null;
+        Assert.state(accessor != null, "No accessor");
 
         if (handler.isConnect(accessor)) {
             Object raw = message.getHeaders().get(NativeMessageHeaderAccessor.NATIVE_HEADERS);
@@ -53,11 +54,16 @@ public class ClientInboundChannelInterceptor implements ChannelInterceptor {
 
                 Object tokens = ((Map<?, ?>) raw).get("token");
                 if (tokens instanceof Collection<?>) {
-                    String token = Convert.toList(String.class, tokens).get(0);
-                    User user = auths.getUserByToken(token);
-                    principal.setToken(token);
-                    principal.setUser(user.getRealName());
-                    principal.setType(StompPrincipal.Type.LOGIN_USER);
+                    try {
+                        String token = Convert.toList(String.class, tokens).get(0);
+                        User user = auths.getUserByToken(token);
+                        principal.setToken(token);
+                        principal.setUser(user.getRealName());
+                        principal.setType(StompPrincipal.Type.LOGIN_USER);
+                    } catch (Exception e) {
+                        log.warn("WebSocket(Mode: {}) connect failed: {}", WebSocketStompConfig.mode, e.getMessage());
+                        throw e;
+                    }
                 } else {
                     principal.setUser("Guest." + principal.getSession());
                     principal.setType(StompPrincipal.Type.GUEST_USER);

basic-websocket/src/main/java/com/yiring/websocket/interceptor/ClientOutboundChannelInterceptor.java

@@ -12,6 +12,7 @@ import org.springframework.messaging.simp.SimpMessageHeaderAccessor;
 import org.springframework.messaging.support.ChannelInterceptor;
 import org.springframework.messaging.support.MessageBuilder;
 import org.springframework.stereotype.Component;
+import org.springframework.util.Assert;
 
 /**
  * ClientOutboundChannelInterceptor
@@ -32,7 +33,7 @@ public class ClientOutboundChannelInterceptor implements ChannelInterceptor {
     @Override
     public Message<?> preSend(@NonNull Message<?> message, @NonNull MessageChannel channel) {
         SimpMessageHeaderAccessor accessor = handler.getAccessor(message);
-        assert accessor != null;
+        Assert.state(accessor != null, "No accessor");
 
         if (handler.isConnected(accessor)) {
             StompPrincipal principal = (StompPrincipal) accessor.getUser();