SpringCloud/spring-cloud-gateway
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
spencergibb
2021-03-12 15:40:41 -05:00
parent 40402173e0
commit 685ff1585b
2 changed files with 13 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
spring-cloud-gateway-server/src/main/java/org/springframework/cloud/gateway/filter/factory/AddRequestHeaderGatewayFilterFactory.java
View File

@@ -38,8 +38,7 @@ public class AddRequestHeaderGatewayFilterFactory extends AbstractNameValueGatew
public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
String value = ServerWebExchangeUtils.expand(exchange, config.getValue());
ServerHttpRequest request = exchange.getRequest().mutate()
.headers(httpHeaders -> httpHeaders.add(config.getName(), value))
.build();
.headers(httpHeaders -> httpHeaders.add(config.getName(), value)).build();
return chain.filter(exchange.mutate().request(request).build());
}

31
spring-cloud-gateway-server/src/test/java/org/springframework/cloud/gateway/filter/factory/AddRequestHeaderGatewayFilterFactoryTests.java
View File

@@ -62,13 +62,10 @@ public class AddRequestHeaderGatewayFilterFactoryTests extends BaseWebClientTest
@Test
public void addRequestHeaderFilterWorksMultipleValues() {
testClient.get().uri("/multivalueheaders")
.header("Host", "www.addrequestheader.org").exchange()
testClient.get().uri("/multivalueheaders").header("Host", "www.addrequestheader.org").exchange()
.expectBody(Map.class).consumeWith(result -> {
Map<String, Object> headers = getMap(result.getResponseBody(),
"headers");
assertThat(headers).containsEntry("X-Request-Example",
Arrays.asList("ValueA", "ValueB"));
Map<String, Object> headers = getMap(result.getResponseBody(), "headers");
assertThat(headers).containsEntry("X-Request-Example", Arrays.asList("ValueA", "ValueB"));
});
}
@@ -83,13 +80,10 @@ public class AddRequestHeaderGatewayFilterFactoryTests extends BaseWebClientTest
@Test
public void addRequestHeaderFilterMultipleValuesWorksJavaDsl() {
testClient.get().uri("/multivalueheaders")
.header("Host", "www.addrequestheaderjava.org").exchange()
testClient.get().uri("/multivalueheaders").header("Host", "www.addrequestheaderjava.org").exchange()
.expectBody(Map.class).consumeWith(result -> {
Map<String, Object> headers = getMap(result.getResponseBody(),
"headers");
assertThat(headers).containsEntry("X-Request-Acme",
Arrays.asList("ValueB-www", "ValueC-www"));
Map<String, Object> headers = getMap(result.getResponseBody(), "headers");
assertThat(headers).containsEntry("X-Request-Acme", Arrays.asList("ValueB-www", "ValueC-www"));
});
}
@@ -114,13 +108,12 @@ public class AddRequestHeaderGatewayFilterFactoryTests extends BaseWebClientTest
r -> r.path("/headers").and().host("{sub}.addrequestheaderjava.org")
.filters(f -> f.prefixPath("/httpbin").addRequestHeader("X-Request-Acme", "ValueB-{sub}"))
.uri(uri))
.route("add_multiple_request_header_java_test", r -> r
.path("/multivalueheaders").and()
.host("{sub}.addrequestheaderjava.org")
.filters(f -> f.prefixPath("/httpbin")
.addRequestHeader("X-Request-Acme", "ValueB-{sub}")
.addRequestHeader("X-Request-Acme", "ValueC-{sub}"))
.uri(uri))
.route("add_multiple_request_header_java_test",
r -> r.path("/multivalueheaders").and().host("{sub}.addrequestheaderjava.org")
.filters(f -> f.prefixPath("/httpbin")
.addRequestHeader("X-Request-Acme", "ValueB-{sub}")
.addRequestHeader("X-Request-Acme", "ValueC-{sub}"))
.uri(uri))
.build();
}