java - Spring-boot 中路由 websocket 目的地

Question

拥有原始的 websocket 实现:

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new MessageHandler(), "/websocket")
                .setAllowedOrigins("*")
                .addInterceptors();;
    }
}

处理程序:

public class MessageHandler extends TextWebSocketHandler {
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        // The WebSocket has been closed
    }
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        String auth = (String) session.getAttributes().get("auth");
        System.out.println(auth);

        session.sendMessage(new TextMessage("You are now connected to the server. This is the first message."));
    }
    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage textMessage) throws Exception {
        // A message has been received

    }
}

Websocket 客户端使用 /websocket url 连接到服务器(握手等)，例如 ws://localhost:8080/websocket

但是，现在连接已建立，是否有办法路由消息？假设我有一个提供聊天和一些弹出功能的应用程序(为简单起见，假设用户向应用程序中的所有 friend 发送弹出消息和一些弹出窗口)。

当然，我想将聊天消息路由到 /chat 并将弹出窗口路由到 /popup。

实现此目的的一种方法是将 json 消息发送到服务器并在那里解析它，例如:

    protected void handleTextMessage(WebSocketSession session, TextMessage textMessage) throws Exception {
        String path = getRouteFromJsonMessage(textMessage);
        if( ! "".equals(path) && path.equals("chat")
           ....
        if( ! "".equals(path) && path.equals("popup")
          ....

    }

但这似乎太慢了，在每条消息上解析 json。还有其他更好的方法来实现路由吗？

感谢您的帮助!

Answer 1

为什么不注册两个不同的 MessageHandler

public class WebSocketConfig implements WebSocketConfigurer {
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new ChatMessageHandler(), "/chat")
                    .setAllowedOrigins("*")
                    .addInterceptors()
                .addHandler(new PopUpHandler(), "/popup") //etc;

    }
}