Merge branch '3.1.x'

pull/36016/head
Phillip Webb 1 year ago
commit 2ce36b15b3

@ -16,7 +16,10 @@
package org.springframework.boot.autoconfigure.web.reactive.function.client; package org.springframework.boot.autoconfigure.web.reactive.function.client;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Supplier; import java.util.function.Supplier;
import java.util.stream.Collectors;
import java.util.stream.Stream; import java.util.stream.Stream;
import javax.net.ssl.SSLException; import javax.net.ssl.SSLException;
@ -36,6 +39,7 @@ import org.springframework.util.function.ThrowingConsumer;
* {@link ClientHttpConnectorFactory} for {@link ReactorClientHttpConnector}. * {@link ClientHttpConnectorFactory} for {@link ReactorClientHttpConnector}.
* *
* @author Phillip Webb * @author Phillip Webb
* @author Fernando Cappi
*/ */
class ReactorClientHttpConnectorFactory implements ClientHttpConnectorFactory<ReactorClientHttpConnector> { class ReactorClientHttpConnectorFactory implements ClientHttpConnectorFactory<ReactorClientHttpConnector> {
@ -55,20 +59,19 @@ class ReactorClientHttpConnectorFactory implements ClientHttpConnectorFactory<Re
@Override @Override
public ReactorClientHttpConnector createClientHttpConnector(SslBundle sslBundle) { public ReactorClientHttpConnector createClientHttpConnector(SslBundle sslBundle) {
ReactorNettyHttpClientMapper mapper = this.mappers.get() List<ReactorNettyHttpClientMapper> mappers = this.mappers.get()
.reduce((before, after) -> (client) -> after.configure(before.configure(client))) .collect(Collectors.toCollection(ArrayList::new));
.orElse((client) -> client);
if (sslBundle != null) { if (sslBundle != null) {
mapper = new SslConfigurer(sslBundle)::configure; mappers.add(new SslConfigurer(sslBundle));
} }
return new ReactorClientHttpConnector(this.reactorResourceFactory, mapper::configure); return new ReactorClientHttpConnector(this.reactorResourceFactory,
ReactorNettyHttpClientMapper.of(mappers)::configure);
} }
/** /**
* Configures the Netty {@link HttpClient} with SSL. * Configures the Netty {@link HttpClient} with SSL.
*/ */
private static class SslConfigurer { private static class SslConfigurer implements ReactorNettyHttpClientMapper {
private final SslBundle sslBundle; private final SslBundle sslBundle;
@ -76,7 +79,8 @@ class ReactorClientHttpConnectorFactory implements ClientHttpConnectorFactory<Re
this.sslBundle = sslBundle; this.sslBundle = sslBundle;
} }
HttpClient configure(HttpClient httpClient) { @Override
public HttpClient configure(HttpClient httpClient) {
return httpClient.secure(ThrowingConsumer.of(this::customizeSsl).throwing(IllegalStateException::new)); return httpClient.secure(ThrowingConsumer.of(this::customizeSsl).throwing(IllegalStateException::new));
} }

@ -1,5 +1,5 @@
/* /*
* Copyright 2012-2020 the original author or authors. * Copyright 2012-2023 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -16,15 +16,19 @@
package org.springframework.boot.autoconfigure.web.reactive.function.client; package org.springframework.boot.autoconfigure.web.reactive.function.client;
import java.util.Collection;
import reactor.netty.http.client.HttpClient; import reactor.netty.http.client.HttpClient;
import org.springframework.http.client.reactive.ReactorClientHttpConnector; import org.springframework.http.client.reactive.ReactorClientHttpConnector;
import org.springframework.util.Assert;
/** /**
* Mapper that allows for custom modification of a {@link HttpClient} before it is used as * Mapper that allows for custom modification of a {@link HttpClient} before it is used as
* the basis for a {@link ReactorClientHttpConnector}. * the basis for a {@link ReactorClientHttpConnector}.
* *
* @author Brian Clozel * @author Brian Clozel
* @author Phillip Webb
* @since 2.3.0 * @since 2.3.0
*/ */
@FunctionalInterface @FunctionalInterface
@ -37,4 +41,31 @@ public interface ReactorNettyHttpClientMapper {
*/ */
HttpClient configure(HttpClient httpClient); HttpClient configure(HttpClient httpClient);
/**
* Return a new {@link ReactorNettyHttpClientMapper} composed of the given mappers.
* @param mappers the mappers to compose
* @return a composed {@link ReactorNettyHttpClientMapper} instance
* @since 3.1.1
*/
static ReactorNettyHttpClientMapper of(Collection<ReactorNettyHttpClientMapper> mappers) {
Assert.notNull(mappers, "Mappers must not be null");
return of(mappers.toArray(ReactorNettyHttpClientMapper[]::new));
}
/**
* Return a new {@link ReactorNettyHttpClientMapper} composed of the given mappers.
* @param mappers the mappers to compose
* @return a composed {@link ReactorNettyHttpClientMapper} instance
* @since 3.1.1
*/
static ReactorNettyHttpClientMapper of(ReactorNettyHttpClientMapper... mappers) {
Assert.notNull(mappers, "Mappers must not be null");
return (httpClient) -> {
for (ReactorNettyHttpClientMapper mapper : mappers) {
httpClient = mapper.configure(httpClient);
}
return httpClient;
};
}
} }

@ -24,6 +24,10 @@ import org.eclipse.jetty.util.thread.Scheduler;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import org.springframework.boot.autoconfigure.AutoConfigurations; import org.springframework.boot.autoconfigure.AutoConfigurations;
import org.springframework.boot.ssl.SslBundle;
import org.springframework.boot.ssl.SslBundleKey;
import org.springframework.boot.ssl.jks.JksSslStoreBundle;
import org.springframework.boot.ssl.jks.JksSslStoreDetails;
import org.springframework.boot.test.context.FilteredClassLoader; import org.springframework.boot.test.context.FilteredClassLoader;
import org.springframework.boot.test.context.runner.ReactiveWebApplicationContextRunner; import org.springframework.boot.test.context.runner.ReactiveWebApplicationContextRunner;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
@ -34,6 +38,8 @@ import org.springframework.test.util.ReflectionTestUtils;
import static org.assertj.core.api.Assertions.assertThat; import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.mock; import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.verify;
/** /**
* Tests for {@link ClientHttpConnectorFactoryConfiguration}. * Tests for {@link ClientHttpConnectorFactoryConfiguration}.
@ -80,12 +86,16 @@ class ClientHttpConnectorFactoryConfigurationTests {
@Test @Test
void shouldApplyHttpClientMapper() { void shouldApplyHttpClientMapper() {
JksSslStoreDetails storeDetails = JksSslStoreDetails.forLocation("classpath:test.jks");
JksSslStoreBundle stores = new JksSslStoreBundle(storeDetails, storeDetails);
SslBundle sslBundle = spy(SslBundle.of(stores, SslBundleKey.of("password")));
new ReactiveWebApplicationContextRunner() new ReactiveWebApplicationContextRunner()
.withConfiguration(AutoConfigurations.of(ClientHttpConnectorFactoryConfiguration.ReactorNetty.class)) .withConfiguration(AutoConfigurations.of(ClientHttpConnectorFactoryConfiguration.ReactorNetty.class))
.withUserConfiguration(CustomHttpClientMapper.class) .withUserConfiguration(CustomHttpClientMapper.class)
.run((context) -> { .run((context) -> {
context.getBean(ReactorClientHttpConnectorFactory.class).createClientHttpConnector(); context.getBean(ReactorClientHttpConnectorFactory.class).createClientHttpConnector(sslBundle);
assertThat(CustomHttpClientMapper.called).isTrue(); assertThat(CustomHttpClientMapper.called).isTrue();
verify(sslBundle).getManagers();
}); });
} }

@ -0,0 +1,99 @@
/*
* Copyright 2012-2023 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.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.boot.autoconfigure.web.reactive.function.client;
import java.util.Collection;
import java.util.List;
import org.junit.jupiter.api.Test;
import reactor.netty.http.client.HttpClient;
import reactor.netty.http.client.HttpClientConfig;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
/**
* Tests for {@link ReactorNettyHttpClientMapper}.
*
* @author Phillip Webb
*/
class ReactorNettyHttpClientMapperTests {
@Test
void ofWithCollectionCreatesComposite() {
ReactorNettyHttpClientMapper one = (httpClient) -> new TestHttpClient(httpClient, "1");
ReactorNettyHttpClientMapper two = (httpClient) -> new TestHttpClient(httpClient, "2");
ReactorNettyHttpClientMapper three = (httpClient) -> new TestHttpClient(httpClient, "3");
ReactorNettyHttpClientMapper compose = ReactorNettyHttpClientMapper.of(List.of(one, two, three));
TestHttpClient httpClient = (TestHttpClient) compose.configure(new TestHttpClient());
assertThat(httpClient.getContent()).isEqualTo("123");
}
@Test
void ofWhenCollectionIsNullThrowsException() {
Collection<ReactorNettyHttpClientMapper> mappers = null;
assertThatIllegalArgumentException().isThrownBy(() -> ReactorNettyHttpClientMapper.of(mappers))
.withMessage("Mappers must not be null");
}
@Test
void ofWithArrayCreatesComposite() {
ReactorNettyHttpClientMapper one = (httpClient) -> new TestHttpClient(httpClient, "1");
ReactorNettyHttpClientMapper two = (httpClient) -> new TestHttpClient(httpClient, "2");
ReactorNettyHttpClientMapper three = (httpClient) -> new TestHttpClient(httpClient, "3");
ReactorNettyHttpClientMapper compose = ReactorNettyHttpClientMapper.of(one, two, three);
TestHttpClient httpClient = (TestHttpClient) compose.configure(new TestHttpClient());
assertThat(httpClient.getContent()).isEqualTo("123");
}
@Test
void ofWhenArrayIsNullThrowsException() {
ReactorNettyHttpClientMapper[] mappers = null;
assertThatIllegalArgumentException().isThrownBy(() -> ReactorNettyHttpClientMapper.of(mappers))
.withMessage("Mappers must not be null");
}
private static class TestHttpClient extends HttpClient {
private final String content;
TestHttpClient() {
this.content = "";
}
TestHttpClient(HttpClient httpClient, String content) {
this.content = (httpClient instanceof TestHttpClient testHttpClient) ? testHttpClient.content + content
: content;
}
@Override
public HttpClientConfig configuration() {
throw new UnsupportedOperationException("Auto-generated method stub");
}
@Override
protected HttpClient duplicate() {
throw new UnsupportedOperationException("Auto-generated method stub");
}
String getContent() {
return this.content;
}
}
}
Loading…
Cancel
Save