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

Example 1 with JettyWebSocketHandlerAdapter

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

the class JettyWebSocketClient method createJettyHandler.

private Object createJettyHandler(URI url, WebSocketHandler handler, MonoProcessor<Void> completion) {
    return new JettyWebSocketHandlerAdapter(handler, session -> {
        UpgradeResponse response = session.getUpgradeResponse();
        HttpHeaders responseHeaders = new HttpHeaders();
        response.getHeaders().forEach(responseHeaders::put);
        HandshakeInfo info = afterHandshake(url, responseHeaders);
        return new JettyWebSocketSession(session, info, this.bufferFactory, completion);
    });
}

Example 2 with JettyWebSocketHandlerAdapter

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

the class JettyRequestUpgradeStrategy 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);
    JettyWebSocketHandlerAdapter adapter = new JettyWebSocketHandlerAdapter(handler, session -> {
        HandshakeInfo info = getHandshakeInfo(exchange, subProtocol);
        DataBufferFactory factory = response.bufferFactory();
        return new JettyWebSocketSession(session, info, factory);
    });
    startLazily(servletRequest);
    boolean isUpgrade = this.factory.isUpgradeRequest(servletRequest, servletResponse);
    Assert.isTrue(isUpgrade, "Not a WebSocket handshake");
    try {
        adapterHolder.set(new WebSocketHandlerContainer(adapter, subProtocol));
        this.factory.acceptWebSocket(servletRequest, servletResponse);
    } catch (IOException ex) {
        return Mono.error(ex);
    } finally {
        adapterHolder.remove();
    }
    return Mono.empty();
}