时间戳

首页 > 解决方案 > 在 Spring-boot 中路由 websocket 目标

java - 在 Spring-boot 中路由 websocket 目标

问题描述

具有原始 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

    }
}

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

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

当然，我想将聊天消息路由到/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。还有其他更好的方法来实现路由吗？

感谢帮助！

标签： javaspringspring-bootwebsocket

解决方案

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

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

    }
}

推荐阅读