@Override public void afterConnectionEstablished(WebSocketSession session) throws Exception { if (logger.isDebugEnabled()) { logger.debug("New " + session); } super.afterConnectionEstablished(session); }
@Override public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception { if (logger.isDebugEnabled()) { logger.debug(session + " closed with " + closeStatus); } super.afterConnectionClosed(session, closeStatus); }
@Override public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception { if (logger.isTraceEnabled()) { logger.trace("Handling " + message + " in " + session); } super.handleMessage(session, message); }
@Test public void getLastHandler() { AbstractWebSocketHandler h1 = new AbstractWebSocketHandler() { }; WebSocketHandlerDecorator h2 = new WebSocketHandlerDecorator(h1); WebSocketHandlerDecorator h3 = new WebSocketHandlerDecorator(h2); assertSame(h1, h3.getLastHandler()); }
private static SubProtocolWebSocketHandler unwrapSubProtocolWebSocketHandler(WebSocketHandler handler) { WebSocketHandler actual = WebSocketHandlerDecorator.unwrap(handler); if (!(actual instanceof SubProtocolWebSocketHandler)) { throw new IllegalArgumentException("No SubProtocolWebSocketHandler in " + handler); } return (SubProtocolWebSocketHandler) actual; }
public static WebSocketHandler unwrap(WebSocketHandler handler) { if (handler instanceof WebSocketHandlerDecorator) { return ((WebSocketHandlerDecorator) handler).getLastHandler(); } else { return handler; } }
public WebSocketHandler getLastHandler() { WebSocketHandler result = this.delegate; while (result instanceof WebSocketHandlerDecorator) { result = ((WebSocketHandlerDecorator) result).getDelegate(); } return result; }
@Override public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception { if (logger.isDebugEnabled()) { logger.debug("Transport error in " + session, exception); } super.handleTransportError(session, exception); }
public SockJsWebSocketHandler(SockJsServiceConfig serviceConfig, WebSocketHandler webSocketHandler, WebSocketServerSockJsSession sockJsSession) { Assert.notNull(serviceConfig, "serviceConfig must not be null"); Assert.notNull(webSocketHandler, "webSocketHandler must not be null"); Assert.notNull(sockJsSession, "session must not be null"); this.sockJsServiceConfig = serviceConfig; this.sockJsSession = sockJsSession; webSocketHandler = WebSocketHandlerDecorator.unwrap(webSocketHandler); this.subProtocols = ((webSocketHandler instanceof SubProtocolCapable) ? new ArrayList<>(((SubProtocolCapable) webSocketHandler).getSubProtocols()) : Collections.emptyList()); }
private WebSocketHandler unwrapWebSocketHandler(WebSocketHandler handler) { return (handler instanceof WebSocketHandlerDecorator) ? ((WebSocketHandlerDecorator) handler).getLastHandler() : handler; }
@Test public void openConnection() throws Exception { List<String> subprotocols = Arrays.asList("abc"); TestLifecycleWebSocketClient client = new TestLifecycleWebSocketClient(false); WebSocketHandler handler = new TextWebSocketHandler(); WebSocketConnectionManager manager = new WebSocketConnectionManager(client, handler , "/path/{id}", "123"); manager.setSubProtocols(subprotocols); manager.openConnection(); WebSocketHttpHeaders expectedHeaders = new WebSocketHttpHeaders(); expectedHeaders.setSecWebSocketProtocol(subprotocols); assertEquals(expectedHeaders, client.headers); assertEquals(new URI("/path/123"), client.uri); WebSocketHandlerDecorator loggingHandler = (WebSocketHandlerDecorator) client.webSocketHandler; assertEquals(LoggingWebSocketHandlerDecorator.class, loggingHandler.getClass()); assertSame(handler, loggingHandler.getDelegate()); }
@Override public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception { if (logger.isDebugEnabled()) { logger.debug("Transport error in " + session, exception); } super.handleTransportError(session, exception); }
@Override public void afterConnectionEstablished(WebSocketSession session) throws Exception { session.getAttributes().put("decorated", true); super.afterConnectionEstablished(session); } }
@Override public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception { if (logger.isDebugEnabled()) { logger.debug(session + " closed with " + closeStatus); } super.afterConnectionClosed(session, closeStatus); }
/** * Determine the sub-protocols supported by the given WebSocketHandler by * checking whether it is an instance of {@link SubProtocolCapable}. * @param handler the handler to check * @return a list of supported protocols, or an empty list if none available */ protected final List<String> determineHandlerSupportedProtocols(WebSocketHandler handler) { WebSocketHandler handlerToCheck = WebSocketHandlerDecorator.unwrap(handler); List<String> subProtocols = null; if (handlerToCheck instanceof SubProtocolCapable) { subProtocols = ((SubProtocolCapable) handlerToCheck).getSubProtocols(); } return (subProtocols != null ? subProtocols : Collections.emptyList()); }
@Override public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception { super.handleMessage(session, message); }
private static void unwrapAndCheckDecoratedHandlerType(WebSocketHandler handler, Class<?> handlerClass) { if (handler instanceof WebSocketHandlerDecorator) { handler = ((WebSocketHandlerDecorator) handler).getLastHandler(); } assertTrue(handlerClass.isInstance(handler)); } }
public WebSocketHandler getLastHandler() { WebSocketHandler result = this.delegate; while (result instanceof WebSocketHandlerDecorator) { result = ((WebSocketHandlerDecorator) result).getDelegate(); } return result; }
@Override public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception { if (logger.isDebugEnabled()) { logger.debug("Transport error in " + session, exception); } super.handleTransportError(session, exception); }
@Override public void afterConnectionEstablished(WebSocketSession wsSession) throws Exception { super.afterConnectionEstablished(wsSession); publishEvent(new SessionConnectEvent(this, wsSession)); }