Examples with ServerHttpRequest org.springframework.http.server.reactive.ServerHttpRequest used on opensource projects

Example 21 with ServerHttpRequest

use of org.springframework.http.server.reactive.ServerHttpRequest 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 22 with ServerHttpRequest

use of org.springframework.http.server.reactive.ServerHttpRequest in project spring-framework by spring-projects.

the class WebHttpHandlerBuilderTests method httpHandlerDecorator.

@Test
void httpHandlerDecorator() {
    BiFunction<ServerHttpRequest, String, ServerHttpRequest> mutator = (req, value) -> req.mutate().headers(headers -> headers.add("My-Header", value)).build();
    AtomicBoolean success = new AtomicBoolean();
    HttpHandler httpHandler = WebHttpHandlerBuilder.webHandler(exchange -> {
        HttpHeaders headers = exchange.getRequest().getHeaders();
        assertThat(headers.get("My-Header")).containsExactlyInAnyOrder("1", "2", "3");
        success.set(true);
        return Mono.empty();
    }).httpHandlerDecorator(handler -> (req, res) -> handler.handle(mutator.apply(req, "1"), res)).httpHandlerDecorator(handler -> (req, res) -> handler.handle(mutator.apply(req, "2"), res)).httpHandlerDecorator(handler -> (req, res) -> handler.handle(mutator.apply(req, "3"), res)).build();
    httpHandler.handle(MockServerHttpRequest.get("/").build(), new MockServerHttpResponse()).block();
    assertThat(success.get()).isTrue();
}

Example 23 with ServerHttpRequest

use of org.springframework.http.server.reactive.ServerHttpRequest in project spring-framework by spring-projects.

the class ForwardedHeaderTransformerTests method removeOnly.

@Test
void removeOnly() {
    this.requestMutator.setRemoveOnly(true);
    HttpHeaders headers = new HttpHeaders();
    headers.add("Forwarded", "for=192.0.2.60;proto=http;by=203.0.113.43");
    headers.add("X-Forwarded-Host", "example.com");
    headers.add("X-Forwarded-Port", "8080");
    headers.add("X-Forwarded-Proto", "http");
    headers.add("X-Forwarded-Prefix", "prefix");
    headers.add("X-Forwarded-Ssl", "on");
    headers.add("X-Forwarded-For", "203.0.113.195");
    ServerHttpRequest request = this.requestMutator.apply(getRequest(headers));
    assertForwardedHeadersRemoved(request);
}

Example 24 with ServerHttpRequest

use of org.springframework.http.server.reactive.ServerHttpRequest in project spring-framework by spring-projects.

the class ForwardedHeaderTransformerTests method forwardedForNotPresent.

@Test
public void forwardedForNotPresent() throws URISyntaxException {
    HttpHeaders headers = new HttpHeaders();
    headers.add("Forwarded", "host=84.198.58.199;proto=https");
    InetSocketAddress remoteAddress = new InetSocketAddress("example.client", 47011);
    ServerHttpRequest request = MockServerHttpRequest.method(HttpMethod.GET, new URI("https://example.com/a%20b?q=a%2Bb")).remoteAddress(remoteAddress).headers(headers).build();
    request = this.requestMutator.apply(request);
    assertThat(request.getRemoteAddress()).isEqualTo(remoteAddress);
}

Example 25 with ServerHttpRequest

use of org.springframework.http.server.reactive.ServerHttpRequest in project spring-framework by spring-projects.

the class ForwardedHeaderTransformerTests method xForwardedPrefixShouldNotLeadToDecodedPath.

// gh-23305
@Test
void xForwardedPrefixShouldNotLeadToDecodedPath() throws Exception {
    HttpHeaders headers = new HttpHeaders();
    headers.add("X-Forwarded-Prefix", "/prefix");
    ServerHttpRequest request = MockServerHttpRequest.method(HttpMethod.GET, new URI("https://example.com/a%20b?q=a%2Bb")).headers(headers).build();
    request = this.requestMutator.apply(request);
    assertThat(request.getURI()).isEqualTo(new URI("https://example.com/prefix/a%20b?q=a%2Bb"));
    assertThat(request.getPath().value()).isEqualTo("/prefix/a%20b");
    assertForwardedHeadersRemoved(request);
}