Examples with StandardWebSocketSession org.springframework.web.reactive.socket.adapter.StandardWebSocketSession used on opensource projects

Example 1 with StandardWebSocketSession

use of org.springframework.web.reactive.socket.adapter.StandardWebSocketSession in project spring-framework by spring-projects.

the class StandardWebSocketClient method createEndpoint.

private StandardWebSocketHandlerAdapter createEndpoint(URI url, WebSocketHandler handler, MonoProcessor<Void> completion, DefaultConfigurator configurator) {
    return new StandardWebSocketHandlerAdapter(handler, session -> {
        HttpHeaders responseHeaders = configurator.getResponseHeaders();
        HandshakeInfo info = afterHandshake(url, responseHeaders);
        return new StandardWebSocketSession(session, info, this.bufferFactory, completion);
    });
}

Example 2 with StandardWebSocketSession

use of org.springframework.web.reactive.socket.adapter.StandardWebSocketSession in project spring-framework by spring-projects.

the class TomcatRequestUpgradeStrategy method upgrade.

@Override
public Mono<Void> upgrade(ServerWebExchange exchange, WebSocketHandler handler, Optional<String> subProtocol) {
    ServerHttpRequest request = exchange.getRequest();
    ServerHttpResponse response = exchange.getResponse();
    HttpServletRequest servletRequest = getHttpServletRequest(request);
    HttpServletResponse servletResponse = getHttpServletResponse(response);
    Endpoint endpoint = new StandardWebSocketHandlerAdapter(handler, session -> {
        HandshakeInfo info = getHandshakeInfo(exchange, subProtocol);
        DataBufferFactory factory = response.bufferFactory();
        return new StandardWebSocketSession(session, info, factory);
    });
    String requestURI = servletRequest.getRequestURI();
    DefaultServerEndpointConfig config = new DefaultServerEndpointConfig(requestURI, endpoint);
    config.setSubprotocols(subProtocol.map(Collections::singletonList).orElse(Collections.emptyList()));
    try {
        WsServerContainer container = getContainer(servletRequest);
        container.doUpgrade(servletRequest, servletResponse, config, Collections.emptyMap());
    } catch (ServletException | IOException ex) {
        return Mono.error(ex);
    }
    return Mono.empty();
}