Spring容器整合WebSocket

453次阅读

共计 1450 个字符，预计需要花费 4 分钟才能阅读完成。

WebSocker 是一个保持 web 客户端与服务器长链接的技术。这样在两者通信过程中如果服务器有消息发送给客户端，就无需等待 web 客户端发送一个请求了 (HTTP 协议是请求相应式，如果没有 Web 客户端的请求，服务器是无法对客户端做相应的)。

单独的部署一个 WebSocket 服务器很简单，只需要几个注解即可。那么当在 Spring 项目中如何使用 WebSocket 呢? 也行你会想，没什么不同啊。一样通过 WebSocket 注解就可以了啊。但我们使用 Spring 的目的就是通过 Spring 的容器来管理 Bean，将 WebSocket 独立出来如果 WebSocket 中需要用到 Spring 容器中的 bean 该怎么办？

Spring 提供了两种将 WebSocket 部署在 Spring 容器中的方法:

方法一:

使用 org.springframework.web.socket.server.standard.ServerEndpointExporter.ServerEndpointExporter 类。在 Spring 容器中添加一个该类的实例:

@Bean
    public ServerEndpointExporter serverEndpointExporter() {return new ServerEndpointExporter();}

Spring 容器会去扫描包，将 WebSocket 加入容器。此时对于 WebSocket 则使用 Spring 的注解:

@ServerEndpoint(value="/getLocation",configurator = SpringConfigurator.class)
@Component
public class TransmissionLocationWebSocket {xxxx}

SpringConfigurator.class 为 Spring 提供的一个类。通过这个类, 该 WebSocket 中就可以注入 Spring 容器的 bean。

方法二：

使用 Spring 封装的 WebSocket 接口 TextWebSocketHandler。

public class WebSocketHandler extends TextWebSocketHandler {
    /**
     * 连接成功时候
     */
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception       { }
     /**
      * 连接关闭时候
      */
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {}}

将类添加到 Spring 容器中，并配置:

@Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {registry.addHandler(myHandler(), "/getLocation").setAllowedOrigins("*");
    }

    @Bean
    public WebSocketHandler myHandler() {return new WebSocketHandler();}