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

Example 11 with HandshakeInfo

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

the class TomcatRequestUpgradeStrategy method upgrade.

// for old doUpgrade variant in Tomcat 9.0.55
@SuppressWarnings("deprecation")
@Override
public Mono<Void> upgrade(ServerWebExchange exchange, WebSocketHandler handler, @Nullable String subProtocol, Supplier<HandshakeInfo> handshakeInfoFactory) {
    ServerHttpRequest request = exchange.getRequest();
    ServerHttpResponse response = exchange.getResponse();
    HttpServletRequest servletRequest = ServerHttpRequestDecorator.getNativeRequest(request);
    HttpServletResponse servletResponse = ServerHttpResponseDecorator.getNativeResponse(response);
    HandshakeInfo handshakeInfo = handshakeInfoFactory.get();
    DataBufferFactory bufferFactory = response.bufferFactory();
    // Trigger WebFlux preCommit actions and upgrade
    return exchange.getResponse().setComplete().then(Mono.deferContextual(contextView -> {
        Endpoint endpoint = new StandardWebSocketHandlerAdapter(ContextWebSocketHandler.decorate(handler, contextView), session -> new TomcatWebSocketSession(session, handshakeInfo, bufferFactory));
        String requestURI = servletRequest.getRequestURI();
        DefaultServerEndpointConfig config = new DefaultServerEndpointConfig(requestURI, endpoint);
        config.setSubprotocols(subProtocol != null ? Collections.singletonList(subProtocol) : Collections.emptyList());
        WsServerContainer container = getContainer(servletRequest);
        try {
            container.doUpgrade(servletRequest, servletResponse, config, Collections.emptyMap());
        } catch (Exception ex) {
            return Mono.error(ex);
        }
        return Mono.empty();
    }));
}

Example 12 with HandshakeInfo

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

the class UndertowWebSocketClient method handleChannel.

private void handleChannel(URI url, WebSocketHandler handler, Sinks.Empty<Void> completionSink, DefaultNegotiation negotiation, WebSocketChannel channel) {
    HandshakeInfo info = createHandshakeInfo(url, negotiation);
    DataBufferFactory bufferFactory = DefaultDataBufferFactory.sharedInstance;
    UndertowWebSocketSession session = new UndertowWebSocketSession(channel, info, bufferFactory, completionSink);
    UndertowWebSocketHandlerAdapter adapter = new UndertowWebSocketHandlerAdapter(session);
    channel.getReceiveSetter().set(adapter);
    channel.resumeReceives();
    handler.handle(session).checkpoint(url + " [UndertowWebSocketClient]").subscribe(session);
}

Example 13 with HandshakeInfo

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

the class UndertowWebSocketClient method createHandshakeInfo.

private HandshakeInfo createHandshakeInfo(URI url, DefaultNegotiation negotiation) {
    HttpHeaders responseHeaders = negotiation.getResponseHeaders();
    String protocol = responseHeaders.getFirst("Sec-WebSocket-Protocol");
    return new HandshakeInfo(url, responseHeaders, Mono.empty(), protocol);
}

Example 14 with HandshakeInfo

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

the class JettyWebSocketClient method createHandshakeInfo.

private HandshakeInfo createHandshakeInfo(URI url, Session jettySession) {
    HttpHeaders headers = new HttpHeaders();
    jettySession.getUpgradeResponse().getHeaders().forEach(headers::put);
    String protocol = headers.getFirst("Sec-WebSocket-Protocol");
    return new HandshakeInfo(url, headers, Mono.empty(), protocol);
}

Example 15 with HandshakeInfo

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

the class HandshakeWebSocketService method createHandshakeInfo.

private HandshakeInfo createHandshakeInfo(ServerWebExchange exchange, ServerHttpRequest request, @Nullable String protocol, Map<String, Object> attributes) {
    URI uri = request.getURI();
    // Copy request headers, as they might be pooled and recycled by
    // the server implementation once the handshake HTTP exchange is done.
    HttpHeaders headers = new HttpHeaders();
    headers.addAll(request.getHeaders());
    MultiValueMap<String, HttpCookie> cookies = request.getCookies();
    Mono<Principal> principal = exchange.getPrincipal();
    String logPrefix = exchange.getLogPrefix();
    InetSocketAddress remoteAddress = request.getRemoteAddress();
    return new HandshakeInfo(uri, headers, cookies, principal, protocol, remoteAddress, attributes, logPrefix);
}