Examples with DefaultHandshakeHandler org.springframework.web.socket.server.support.DefaultHandshakeHandler used on opensource projects

Example 6 with DefaultHandshakeHandler

use of org.springframework.web.socket.server.support.DefaultHandshakeHandler in project spring-framework by spring-projects.

the class WebSocketHandlerRegistrationTests method handshakeHandler.

@Test
public void handshakeHandler() {
    WebSocketHandler handler = new TextWebSocketHandler();
    HandshakeHandler handshakeHandler = new DefaultHandshakeHandler();
    this.registration.addHandler(handler, "/foo").setHandshakeHandler(handshakeHandler);
    List<Mapping> mappings = this.registration.getMappings();
    assertThat(mappings.size()).isEqualTo(1);
    Mapping mapping = mappings.get(0);
    assertThat(mapping.webSocketHandler).isEqualTo(handler);
    assertThat(mapping.path).isEqualTo("/foo");
    assertThat(mapping.handshakeHandler).isSameAs(handshakeHandler);
}

Example 7 with DefaultHandshakeHandler

use of org.springframework.web.socket.server.support.DefaultHandshakeHandler in project spring-framework by spring-projects.

the class WebSocketHandlerRegistrationTests method handshakeHandlerPassedToSockJsRegistration.

@Test
public void handshakeHandlerPassedToSockJsRegistration() {
    WebSocketHandler handler = new TextWebSocketHandler();
    HandshakeHandler handshakeHandler = new DefaultHandshakeHandler();
    this.registration.addHandler(handler, "/foo").setHandshakeHandler(handshakeHandler).withSockJS();
    this.registration.getSockJsServiceRegistration().setTaskScheduler(this.taskScheduler);
    List<Mapping> mappings = this.registration.getMappings();
    assertThat(mappings.size()).isEqualTo(1);
    Mapping mapping = mappings.get(0);
    assertThat(mapping.webSocketHandler).isEqualTo(handler);
    assertThat(mapping.path).isEqualTo("/foo/**");
    assertThat(mapping.sockJsService).isNotNull();
    WebSocketTransportHandler transportHandler = (WebSocketTransportHandler) mapping.sockJsService.getTransportHandlers().get(TransportType.WEBSOCKET);
    assertThat(transportHandler.getHandshakeHandler()).isSameAs(handshakeHandler);
}

Example 8 with DefaultHandshakeHandler

use of org.springframework.web.socket.server.support.DefaultHandshakeHandler in project jhipster-sample-app-websocket by jhipster.

the class WebsocketConfiguration method defaultHandshakeHandler.

private DefaultHandshakeHandler defaultHandshakeHandler() {
    return new DefaultHandshakeHandler() {

        @Override
        protected Principal determineUser(ServerHttpRequest request, WebSocketHandler wsHandler, Map<String, Object> attributes) {
            Principal principal = request.getPrincipal();
            if (principal == null) {
                Collection<SimpleGrantedAuthority> authorities = new ArrayList<>();
                authorities.add(new SimpleGrantedAuthority(AuthoritiesConstants.ANONYMOUS));
                principal = new AnonymousAuthenticationToken("WebsocketConfiguration", "anonymous", authorities);
            }
            return principal;
        }
    };
}

Example 9 with DefaultHandshakeHandler

use of org.springframework.web.socket.server.support.DefaultHandshakeHandler in project ma-core-public by infiniteautomation.

the class MangoWebSocketConfigurer method handshakeHandler.

@Bean
public DefaultHandshakeHandler handshakeHandler() {
    WebSocketPolicy policy = new WebSocketPolicy(WebSocketBehavior.SERVER);
    policy.setInputBufferSize(8192);
    // We don't want timeouts..
    policy.setIdleTimeout(Integer.MAX_VALUE);
    WebSocketServerFactory factory = new WebSocketServerFactory(policy);
    return new DefaultHandshakeHandler(new JettyRequestUpgradeStrategy(factory));
}

Example 10 with DefaultHandshakeHandler

use of org.springframework.web.socket.server.support.DefaultHandshakeHandler in project spring-framework by spring-projects.

the class WebMvcStompWebSocketEndpointRegistrationTests method handshakeHandlerInterceptorWithSockJsServiceAndAllowedOrigins.

@Test
public void handshakeHandlerInterceptorWithSockJsServiceAndAllowedOrigins() {
    WebMvcStompWebSocketEndpointRegistration registration = new WebMvcStompWebSocketEndpointRegistration(new String[] { "/foo" }, this.handler, this.scheduler);
    DefaultHandshakeHandler handshakeHandler = new DefaultHandshakeHandler();
    HttpSessionHandshakeInterceptor interceptor = new HttpSessionHandshakeInterceptor();
    String origin = "https://mydomain.com";
    registration.setHandshakeHandler(handshakeHandler).addInterceptors(interceptor).setAllowedOrigins(origin).withSockJS();
    MultiValueMap<HttpRequestHandler, String> mappings = registration.getMappings();
    assertThat(mappings.size()).isEqualTo(1);
    Map.Entry<HttpRequestHandler, List<String>> entry = mappings.entrySet().iterator().next();
    assertThat(entry.getValue()).isEqualTo(Arrays.asList("/foo/**"));
    SockJsHttpRequestHandler requestHandler = (SockJsHttpRequestHandler) entry.getKey();
    assertThat(requestHandler.getWebSocketHandler()).isNotNull();
    DefaultSockJsService sockJsService = (DefaultSockJsService) requestHandler.getSockJsService();
    assertThat(sockJsService).isNotNull();
    Map<TransportType, TransportHandler> handlers = sockJsService.getTransportHandlers();
    WebSocketTransportHandler transportHandler = (WebSocketTransportHandler) handlers.get(TransportType.WEBSOCKET);
    assertThat(transportHandler.getHandshakeHandler()).isSameAs(handshakeHandler);
    assertThat(sockJsService.getHandshakeInterceptors().size()).isEqualTo(2);
    assertThat(sockJsService.getHandshakeInterceptors().get(0)).isEqualTo(interceptor);
    assertThat(sockJsService.getHandshakeInterceptors().get(1).getClass()).isEqualTo(OriginHandshakeInterceptor.class);
    assertThat(sockJsService.getAllowedOrigins().contains(origin)).isTrue();
}