Spring容器整合WebSocket

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(); }

Spring与WebScoket的配置即如上两个方法。