WebClient uri method with uriTemplate and UriBuilder

See gh-22705

Author: Jeoffrey Haeyaert

spring-webflux/src/main/java/org/springframework/web/reactive/function/client/DefaultWebClient.java

@@ -184,6 +184,12 @@ public RequestBodySpec uri(String uriTemplate, Map<String, ?> uriVariables) {
 			return uri(uriBuilderFactory.expand(uriTemplate, uriVariables));
 		}
 
+		@Override
+		public RequestBodySpec uri(String uriTemplate, Function<UriBuilder, URI> uriFunction) {
+			attribute(URI_TEMPLATE_ATTRIBUTE, uriTemplate);
+			return uri(uriFunction.apply(uriBuilderFactory.uriString(uriTemplate)));
+		}
+
 		@Override
 		public RequestBodySpec uri(Function<UriBuilder, URI> uriFunction) {
 			return uri(uriFunction.apply(uriBuilderFactory.builder()));

spring-webflux/src/main/java/org/springframework/web/reactive/function/client/WebClient.java

@@ -348,9 +348,16 @@ interface UriSpec<S extends RequestHeadersSpec<?>> {
 		 */
 		S uri(String uri, Map<String, ?> uriVariables);
 
+		/**
+		 * Build the URI for the request using the {@link UriBuilderFactory}
+		 * configured for this client and initialized with the specified <code>uri</code>.
+		 */
+		S uri(String uri, Function<UriBuilder, URI> uriFunction);
+
 		/**
 		 * Build the URI for the request using the {@link UriBuilderFactory}
 		 * configured for this client.
+		 * @see #uri(String, Function)
 		 */
 		S uri(Function<UriBuilder, URI> uriFunction);
 	}

spring-webflux/src/test/java/org/springframework/web/reactive/function/client/DefaultWebClientTests.java

@@ -84,6 +84,18 @@ public void uriBuilder() {
 		assertEquals("/base/path?q=12", request.url().toString());
 	}
 
+
+	@Test
+	public void uriBuilderWithUriTemplate() {
+		this.builder.build().get()
+					.uri("/path/{id}", builder -> builder.queryParam("q", "12").build("identifier"))
+					.exchange().block(Duration.ofSeconds(10));
+
+		ClientRequest request = verifyAndGetRequest();
+		assertEquals("/base/path/identifier?q=12", request.url().toString());
+		assertEquals("/path/{id}", request.attribute(WebClient.class.getName() + ".uriTemplate").<String>get());
+	}
+
 	@Test
 	public void uriBuilderWithPathOverride() {
 		this.builder.build().get()