转载

神坑中间件：spring-boot-starter-websocket

引言

简易地使用 WebSocket 时，使用 spring-boot-starter-websocket 没什么问题，虽然路由部分设计得有些缺陷，但不影响正常使用。

但当我使用 spring-boot-starter-websocket 实现复杂业务的时候，发现这个中间件虽然是 spring 官方提供的中间件，但是却像是从来没有用过 spring 的人写出来的一样。

灵异事件

需求描述

想实现一个外网向企业内局域网转发数据的雏形，就是下面这张图：

secret 为内网服务， server 为外网服务，该服务向 server 注册，建立 WebSocket 连接，这样在外网的 server 接收到指令就能通过 WebSocket 通道转发给内网的 secret 。

神坑中间件：spring-boot-starter-websocket

神奇代码

WebSocket 服务端 Endpoint ，路由映射 /websocket/{name} ， name 为注册的服务实例的名字。

客户端连接 ws://127.0.0.1:8000/websocket/HEBUT ，注册一个名为 HEBUT 的服务实例。

服务端将服务实例名称到 Session 的映射存到了一个 ConcurrentHashMap 里。

@Component
@ServerEndpoint("/websocket/{name}")
public class YunzhiWebSocket {

    private static final Logger logger = LoggerFactory.getLogger(YunzhiWebSocket.class);

    private Map<String, Session> sessionMap = new ConcurrentHashMap<>();

    @OnOpen
    public void onOpen(@PathParam(value = "name") String name, Session session) throws IOException {
        if (name != null && !name.equals("")) {
            logger.debug("名称合法，添加到Map中");
            sessionMap.put(name, session);
        } else {
            logger.debug("关闭连接");
            session.close();
        }
    }

    @OnMessage
    public void onMessage(String message) {
        logger.error("接收到消息 {}", message);
    }

    @OnError
    public void onError(@PathParam(value = "name") String name, Throwable throwable) {
        logger.error("连接发生错误 {}", throwable.getMessage());
        sessionMap.remove(name);
    }

    @OnClose
    public void onClose(@PathParam(value = "name") String name) {
        logger.debug("关闭连接");
        sessionMap.remove(name);
    }

    public Map<String, Session> getSessionMap() {
        return sessionMap;
    }
}

一个映射 ** 的方法，将所有其他的请求都交给当前 action 处理，根据要访问的实例名去 SessionMap 里找相应的 Session 。

@RequestMapping("{name}/**")
public void dispatcher(@PathVariable String name, HttpServletRequest request) {
    logger.debug("根据服务名查询Session");
    Session session = yunzhiWebSocket.getSessionMap().get(name);

    logger.debug("未找到服务，抛出异常");
    if (session == null) {
        throw new ServiceNotFoundException("找不到该服务实例");
    }
}