|
19 | 19 | import org.apache.commons.logging.Log;
|
20 | 20 | import org.apache.commons.logging.LogFactory;
|
21 | 21 |
|
22 |
| -import org.springframework.beans.factory.ObjectProvider; |
23 |
| -import org.springframework.beans.factory.config.ConfigurableListableBeanFactory; |
24 | 22 | import org.springframework.boot.web.client.RestTemplateBuilder;
|
25 | 23 | import org.springframework.web.client.RestTemplate;
|
26 | 24 | import org.springframework.web.client.support.RestTemplateAdapter;
|
27 | 25 | import org.springframework.web.service.invoker.HttpExchangeAdapter;
|
| 26 | +import org.springframework.web.util.DefaultUriBuilderFactory; |
28 | 27 |
|
29 | 28 | /**
|
30 | 29 | * @author Olga Maciaszek-Sharma
|
31 | 30 | */
|
32 |
| -public class RestTemplateAdapterProvider implements HttpExchangeAdapterProvider { |
| 31 | +public class RestTemplateInterfaceClientsFactoryBean extends AbstractHttpInterfaceClientsFactoryBean { |
33 | 32 |
|
34 |
| - private static final Log logger = LogFactory.getLog(RestTemplateAdapterProvider.class); |
35 |
| - |
36 |
| - private final RestTemplateBuilder restTemplateBuilder; |
37 |
| - |
38 |
| - private final ObjectProvider<HttpInterfaceClientsProperties> propertiesProvider; |
39 |
| - |
40 |
| - public RestTemplateAdapterProvider(RestTemplateBuilder restTemplateBuilder, |
41 |
| - ObjectProvider<HttpInterfaceClientsProperties> propertiesProvider) { |
42 |
| - this.restTemplateBuilder = restTemplateBuilder; |
43 |
| - this.propertiesProvider = propertiesProvider; |
44 |
| - } |
| 33 | + private static final Log logger = LogFactory.getLog(RestTemplateInterfaceClientsFactoryBean.class); |
45 | 34 |
|
46 | 35 | @Override
|
47 |
| - public HttpExchangeAdapter get(ConfigurableListableBeanFactory beanFactory, String clientId) { |
48 |
| - RestTemplate userProvidedRestTemplate = QualifiedBeanProvider.qualifiedBean(beanFactory, RestTemplate.class, |
49 |
| - clientId); |
| 36 | + protected HttpExchangeAdapter exchangeAdapter() { |
| 37 | + String baseUrl = getBaseUrl(); |
| 38 | + |
| 39 | + RestTemplate userProvidedRestTemplate = QualifiedBeanProvider |
| 40 | + .qualifiedBean(this.applicationContext.getBeanFactory(), RestTemplate.class, this.clientId); |
50 | 41 | if (userProvidedRestTemplate != null) {
|
| 42 | + // If the user wants to set the baseUrl directly on the builder, |
| 43 | + // it should not be set in properties. |
| 44 | + if (baseUrl != null) { |
| 45 | + userProvidedRestTemplate.setUriTemplateHandler(new DefaultUriBuilderFactory(baseUrl)); |
| 46 | + } |
51 | 47 | return RestTemplateAdapter.create(userProvidedRestTemplate);
|
52 | 48 | }
|
53 |
| - HttpInterfaceClientsProperties properties = this.propertiesProvider.getObject(); |
54 |
| - String baseUrl = properties.getProperties(clientId).getBaseUrl(); |
55 |
| - RestTemplateBuilder userProvidedRestTemplateBuilder = QualifiedBeanProvider.qualifiedBean(beanFactory, |
56 |
| - RestTemplateBuilder.class, clientId); |
| 49 | + |
| 50 | + RestTemplateBuilder userProvidedRestTemplateBuilder = QualifiedBeanProvider |
| 51 | + .qualifiedBean(this.applicationContext.getBeanFactory(), RestTemplateBuilder.class, this.clientId); |
57 | 52 | if (userProvidedRestTemplateBuilder != null) {
|
58 |
| - // If the user wants to set the baseUrl directly on the builder, |
59 |
| - // it should not be set in properties. |
| 53 | + |
60 | 54 | if (baseUrl != null) {
|
61 | 55 | userProvidedRestTemplateBuilder.rootUri(baseUrl);
|
62 | 56 | }
|
63 | 57 | return RestTemplateAdapter.create(userProvidedRestTemplateBuilder.build());
|
64 | 58 | }
|
| 59 | + |
65 | 60 | // create a RestTemplateAdapter bean with default implementation
|
66 | 61 | if (logger.isDebugEnabled()) {
|
67 |
| - logger.debug("Creating RestTemplateAdapter for '" + clientId + "'"); |
| 62 | + logger.debug("Creating RestTemplateAdapter for '" + this.clientId + "'"); |
68 | 63 | }
|
69 |
| - RestTemplate restTemplate = this.restTemplateBuilder.rootUri(baseUrl).build(); |
| 64 | + RestTemplateBuilder builder = this.applicationContext.getBean(RestTemplateBuilder.class); |
| 65 | + RestTemplate restTemplate = builder.rootUri(baseUrl).build(); |
70 | 66 | return RestTemplateAdapter.create(restTemplate);
|
71 | 67 | }
|
72 | 68 |
|
|
0 commit comments