Resolve UriComponentsBuilder argument with contextPath

Closes gh-25822
This commit is contained in:
Rossen Stoyanchev
2020-10-15 22:08:32 +01:00
parent 424e28bc35
commit a2ff139edd
4 changed files with 22 additions and 7 deletions

View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2019 the original author or authors.
* Copyright 2002-2020 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -111,7 +111,8 @@ public class ServerWebExchangeMethodArgumentResolver extends HandlerMethodArgume
}
else if (UriBuilder.class == paramType || UriComponentsBuilder.class == paramType) {
URI uri = exchange.getRequest().getURI();
return UriComponentsBuilder.fromUri(uri).replacePath(null).replaceQuery(null);
String contextPath = exchange.getRequest().getPath().contextPath().value();
return UriComponentsBuilder.fromUri(uri).replacePath(contextPath).replaceQuery(null);
}
else {
// should never happen...

View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2019 the original author or authors.
* Copyright 2002-2020 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -97,9 +97,23 @@ public class ServerWebExchangeMethodArgumentResolverTests {
assertThat(value).isNotNull();
assertThat(value.getClass()).isEqualTo(UriComponentsBuilder.class);
assertThat(((UriComponentsBuilder) value).path("/next").toUriString()).isEqualTo("https://example.org:9999/next");
assertThat(((UriComponentsBuilder) value).path("/next").toUriString())
.isEqualTo("https://example.org:9999/next");
}
@Test // gh-25822
public void resolveUriComponentsBuilderWithContextPath() {
ServerWebExchange exchange = MockServerWebExchange.from(
MockServerHttpRequest.get("https://example.org:9999/app/path?q=foo").contextPath("/app"));
MethodParameter param = this.testMethod.arg(UriComponentsBuilder.class);
Object value = this.resolver.resolveArgument(param, new BindingContext(), exchange).block();
assertThat(value).isNotNull();
assertThat(value.getClass()).isEqualTo(UriComponentsBuilder.class);
assertThat(((UriComponentsBuilder) value).path("/next").toUriString())
.isEqualTo("https://example.org:9999/app/next");
}
@SuppressWarnings("unused")

View File

@@ -1903,8 +1903,8 @@ and others) and is equivalent to `required=false`.
See <<webflux-ann-sessionattributes>> for more details.
| `UriComponentsBuilder`
| For preparing a URL relative to the current request's host, port, scheme, and path.
See <<webflux-uri-building>>.
| For preparing a URL relative to the current request's host, port, scheme, and
context path. See <<webflux-uri-building>>.
| `@SessionAttribute`
| For access to any session attribute -- in contrast to model attributes stored in the session

View File

@@ -2085,7 +2085,7 @@ and others) and is equivalent to `required=false`.
| `UriComponentsBuilder`
| For preparing a URL relative to the current request's host, port, scheme, context path, and
the literal part of the servlet mapping. See <<mvc-uri-building>>.
the literal part of the servlet mapping. See <<mvc-uri-building>>.
| `@SessionAttribute`
| For access to any session attribute, in contrast to model attributes stored in the session