@Override public Mono<Void> handle(ServerHttpRequest request, ServerHttpResponse response) { // Remove underlying context path first (e.g. Servlet container) String path = request.getPath().pathWithinApplication().value(); return this.handlerMap.entrySet().stream() .filter(entry -> path.startsWith(entry.getKey())) .findFirst() .map(entry -> { String contextPath = request.getPath().contextPath().value() + entry.getKey(); ServerHttpRequest newRequest = request.mutate().contextPath(contextPath).build(); return entry.getValue().handle(newRequest, response); }) .orElseGet(() -> { response.setStatusCode(HttpStatus.NOT_FOUND); return response.setComplete(); }); }
@Override public Mono<Void> handle(ServerHttpRequest request, ServerHttpResponse response) { // Remove underlying context path first (e.g. Servlet container) String path = request.getPath().pathWithinApplication().value(); return this.handlerMap.entrySet().stream() .filter(entry -> path.startsWith(entry.getKey())) .findFirst() .map(entry -> { String contextPath = request.getPath().contextPath().value() + entry.getKey(); ServerHttpRequest newRequest = request.mutate().contextPath(contextPath).build(); return entry.getValue().handle(newRequest, response); }) .orElseGet(() -> { response.setStatusCode(HttpStatus.NOT_FOUND); return response.setComplete(); }); }
@Override public GatewayFilter apply(Config config) { return (exchange, chain) -> { ServerWebExchange filteredExchange = ofNullable((Throwable) exchange .getAttribute(HYSTRIX_EXECUTION_EXCEPTION_ATTR)) .map(executionException -> { ServerHttpRequest.Builder requestBuilder = exchange.getRequest().mutate(); requestBuilder.header(config.executionExceptionTypeHeaderName, executionException.getClass().getName()); requestBuilder.header(config.executionExceptionMessageHeaderName, executionException.getMessage()); ofNullable(getRootCause(executionException)).ifPresent(rootCause -> { requestBuilder.header(config.rootCauseExceptionTypeHeaderName, rootCause.getClass().getName()); requestBuilder.header(config.rootCauseExceptionMessageHeaderName, rootCause.getMessage()); }); return exchange.mutate().request(requestBuilder.build()).build(); }).orElse(exchange); return chain.filter(filteredExchange); }; }
@Override public HttpHeaders filter(HttpHeaders input, ServerWebExchange exchange) { if (log.isDebugEnabled()) { log.debug("Will instrument the HTTP request headers"); } ServerHttpRequest.Builder builder = exchange.getRequest().mutate(); Span span = this.handler.handleSend(this.injector, builder); if (log.isDebugEnabled()) { log.debug( "Client span " + span + " created for the request. New headers are " + builder.build().getHeaders().toSingleValueMap()); } exchange.getAttributes().put(SPAN_ATTRIBUTE, span); HttpHeaders headersWithInput = new HttpHeaders(); headersWithInput.setAll(input.toSingleValueMap()); headersWithInput.setAll(builder.build().getHeaders().toSingleValueMap()); return headersWithInput; }
@Override public GatewayFilter apply(Config config) { UriTemplate uriTemplate = new UriTemplate(config.template); return (exchange, chain) -> { ServerHttpRequest req = exchange.getRequest(); addOriginalRequestUrl(exchange, req.getURI()); Map<String, String> uriVariables = getUriTemplateVariables(exchange); URI uri = uriTemplate.expand(uriVariables); String newPath = uri.getRawPath(); exchange.getAttributes().put(GATEWAY_REQUEST_URL_ATTR, uri); ServerHttpRequest request = req.mutate() .path(newPath) .build(); return chain.filter(exchange.mutate().request(request).build()); }; }
@Override public GatewayFilter apply(Config config) { return (exchange, chain) -> { ServerHttpRequest request = exchange.getRequest(); addOriginalRequestUrl(exchange, request.getURI()); String path = request.getURI().getRawPath(); String newPath = "/" + Arrays.stream(StringUtils.tokenizeToStringArray(path, "/")) .skip(config.parts).collect(Collectors.joining("/")); newPath += (newPath.length() > 1 && path.endsWith("/") ? "/" : ""); ServerHttpRequest newRequest = request.mutate() .path(newPath) .build(); exchange.getAttributes().put(GATEWAY_REQUEST_URL_ATTR, newRequest.getURI()); return chain.filter(exchange.mutate().request(newRequest).build()); }; }
/** * Apply and remove, or remove Forwarded type headers. * @param request the request */ @Override public ServerHttpRequest apply(ServerHttpRequest request) { if (hasForwardedHeaders(request)) { ServerHttpRequest.Builder builder = request.mutate(); if (!this.removeOnly) { URI uri = UriComponentsBuilder.fromHttpRequest(request).build(true).toUri(); builder.uri(uri); String prefix = getForwardedPrefix(request); if (prefix != null) { builder.path(prefix + uri.getPath()); builder.contextPath(prefix); } } removeForwardedHeaders(builder); request = builder.build(); } return request; }
@Override public GatewayFilter apply(Config config) { String replacement = config.replacement.replace("$\\", "$"); return (exchange, chain) -> { ServerHttpRequest req = exchange.getRequest(); addOriginalRequestUrl(exchange, req.getURI()); String path = req.getURI().getRawPath(); String newPath = path.replaceAll(config.regexp, replacement); ServerHttpRequest request = req.mutate() .path(newPath) .build(); exchange.getAttributes().put(GATEWAY_REQUEST_URL_ATTR, request.getURI()); return chain.filter(exchange.mutate().request(request).build()); }; }
@Override public String url(ServerHttpRequest.Builder request) { return request.build().getURI().toString(); }
@Override public String method(ServerHttpRequest.Builder request) { return request.build().getMethodValue(); }
@Override public String requestHeader(ServerHttpRequest.Builder request, String name) { Object result = request.build().getHeaders().get(name); return result != null ? result.toString() : ""; }
@Override public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) { Route route = exchange.getAttribute(GATEWAY_ROUTE_ATTR); URI routeUri = route.getUri(); String scheme = routeUri.getScheme(); if (isAlreadyRouted(exchange) || !"forward".equals(scheme)) { return chain.filter(exchange); } exchange = exchange.mutate().request( exchange.getRequest().mutate().path(routeUri.getPath()).build()) .build(); return chain.filter(exchange); }
@Override public GatewayFilter apply(NameConfig config) { return (exchange, chain) -> { ServerHttpRequest request = exchange.getRequest().mutate() .headers(httpHeaders -> httpHeaders.remove(config.getName())) .build(); return chain.filter(exchange.mutate().request(request).build()); }; } }
@Override public GatewayFilter apply(NameValueConfig config) { return (exchange, chain) -> { ServerHttpRequest request = exchange.getRequest().mutate() .header(config.getName(), config.getValue()) .build(); return chain.filter(exchange.mutate().request(request).build()); }; }
@Override public GatewayFilter apply(NameValueConfig config) { return (exchange, chain) -> { ServerHttpRequest request = exchange.getRequest().mutate() .headers(httpHeaders -> httpHeaders.set(config.name, config.value)) .build(); return chain.filter(exchange.mutate().request(request).build()); }; } }
@Override public ServerWebExchange.Builder request(Consumer<ServerHttpRequest.Builder> consumer) { ServerHttpRequest.Builder builder = this.delegate.getRequest().mutate(); consumer.accept(builder); return request(builder.build()); }
/** * Apply and remove, or remove Forwarded type headers. * @param request the request */ @Override public ServerHttpRequest apply(ServerHttpRequest request) { if (hasForwardedHeaders(request)) { ServerHttpRequest.Builder builder = request.mutate(); if (!this.removeOnly) { URI uri = UriComponentsBuilder.fromHttpRequest(request).build(true).toUri(); builder.uri(uri); String prefix = getForwardedPrefix(request); if (prefix != null) { builder.path(prefix + uri.getPath()); builder.contextPath(prefix); } } removeForwardedHeaders(builder); request = builder.build(); } return request; }
@Test public void mutateRequest() throws Exception { SslInfo sslInfo = mock(SslInfo.class); ServerHttpRequest request = createHttpRequest("/").mutate().sslInfo(sslInfo).build(); assertSame(sslInfo, request.getSslInfo()); request = createHttpRequest("/").mutate().method(HttpMethod.DELETE).build(); assertEquals(HttpMethod.DELETE, request.getMethod()); String baseUri = "http://aaa.org:8080/a"; request = createHttpRequest(baseUri).mutate().uri(URI.create("http://bbb.org:9090/b")).build(); assertEquals("http://bbb.org:9090/b", request.getURI().toString()); request = createHttpRequest(baseUri).mutate().path("/b/c/d").build(); assertEquals("http://aaa.org:8080/b/c/d", request.getURI().toString()); request = createHttpRequest(baseUri).mutate().path("/app/b/c/d").contextPath("/app").build(); assertEquals("http://aaa.org:8080/app/b/c/d", request.getURI().toString()); assertEquals("/app", request.getPath().contextPath().value()); }
@Override public ServerWebExchange.Builder request(Consumer<ServerHttpRequest.Builder> consumer) { ServerHttpRequest.Builder builder = this.delegate.getRequest().mutate(); consumer.accept(builder); return request(builder.build()); }
@Test // SPR-16434 public void mutatePathWithEncodedQueryParams() throws Exception { ServerHttpRequest request = createHttpRequest("/path?name=%E6%89%8E%E6%A0%B9"); request = request.mutate().path("/mutatedPath").build(); assertEquals("/mutatedPath", request.getURI().getRawPath()); assertEquals("name=%E6%89%8E%E6%A0%B9", request.getURI().getRawQuery()); }