study/Spring-Cloud-Alibaba
1
0
Fork 0
mirror of https://gitee.com/mirrors/Spring-Cloud-Alibaba.git synced 2021-06-26 13:25:11 +08:00
Code Issues Releases Wiki Activity

Merge pull request #905 from echooymxq/master

Browse Source 
Advance adapt Dubbo restTemplates
This commit is contained in:
Mercy Ma 2019-09-11 12:47:54 +08:00 committed by GitHub
commit 08cce947fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 34 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
spring-cloud-alibaba-dubbo/src/main/java/com/alibaba/cloud/dubbo/autoconfigure/DubboLoadBalancedRestTemplateAutoConfiguration.java
View File

@ -25,16 +25,19 @@ import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory; import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.boot.autoconfigure.AutoConfigureAfter; import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass; import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.context.event.ApplicationStartedEvent;
import org.springframework.cloud.client.loadbalancer.LoadBalanced; import org.springframework.cloud.client.loadbalancer.LoadBalanced;
import org.springframework.cloud.client.loadbalancer.LoadBalancerInterceptor; import org.springframework.cloud.client.loadbalancer.LoadBalancerInterceptor;
import org.springframework.cloud.client.loadbalancer.RestTemplateCustomizer; import org.springframework.cloud.client.loadbalancer.RestTemplateCustomizer;
import org.springframework.cloud.client.loadbalancer.RetryLoadBalancerInterceptor; import org.springframework.cloud.client.loadbalancer.RetryLoadBalancerInterceptor;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.context.event.EventListener; import org.springframework.context.event.EventListener;
import org.springframework.core.env.Environment; import org.springframework.core.env.Environment;
import org.springframework.core.type.MethodMetadata; import org.springframework.core.type.MethodMetadata;
import org.springframework.http.client.ClientHttpRequestInterceptor; import org.springframework.http.client.ClientHttpRequestInterceptor;
import org.springframework.lang.Nullable;
import org.springframework.util.CollectionUtils; import org.springframework.util.CollectionUtils;
import org.springframework.web.client.RestTemplate; import org.springframework.web.client.RestTemplate;
@ -56,8 +59,8 @@ import com.alibaba.cloud.dubbo.service.DubboGenericServiceFactory;
@ConditionalOnClass(name = { "org.springframework.web.client.RestTemplate" }) @ConditionalOnClass(name = { "org.springframework.web.client.RestTemplate" })
@AutoConfigureAfter(name = { @AutoConfigureAfter(name = {
"org.springframework.cloud.client.loadbalancer.LoadBalancerAutoConfiguration" }) "org.springframework.cloud.client.loadbalancer.LoadBalancerAutoConfiguration" })
public class DubboLoadBalancedRestTemplateAutoConfiguration public class DubboLoadBalancedRestTemplateAutoConfiguration implements
implements BeanClassLoaderAware, SmartInitializingSingleton { BeanClassLoaderAware, ApplicationContextAware, SmartInitializingSingleton {
private static final Class<DubboTransported> DUBBO_TRANSPORTED_CLASS = DubboTransported.class; private static final Class<DubboTransported> DUBBO_TRANSPORTED_CLASS = DubboTransported.class;
@ -89,6 +92,9 @@ public class DubboLoadBalancedRestTemplateAutoConfiguration
@Autowired(required = false) @Autowired(required = false)
private Map<String, RestTemplate> restTemplates = Collections.emptyMap(); private Map<String, RestTemplate> restTemplates = Collections.emptyMap();
@Nullable
private ApplicationContext applicationContext;
private ClassLoader classLoader; private ClassLoader classLoader;
/** /**
@ -111,18 +117,21 @@ public class DubboLoadBalancedRestTemplateAutoConfiguration
* {@link SmartInitializingSingleton} beans or * {@link SmartInitializingSingleton} beans or
* {@link RestTemplateCustomizer#customize(RestTemplate) customization}) * {@link RestTemplateCustomizer#customize(RestTemplate) customization})
*/ */
@EventListener(ApplicationStartedEvent.class) @EventListener(ContextRefreshedEvent.class)
public void adaptRestTemplates() { public void adaptRestTemplates(ContextRefreshedEvent event) {
DubboTransportedAttributesResolver attributesResolver = new DubboTransportedAttributesResolver( if (event.getApplicationContext() == this.applicationContext) {
environment);
for (Map.Entry<String, RestTemplate> entry : restTemplates.entrySet()) { DubboTransportedAttributesResolver attributesResolver = new DubboTransportedAttributesResolver(
String beanName = entry.getKey(); environment);
Map<String, Object> dubboTranslatedAttributes = getDubboTranslatedAttributes(
beanName, attributesResolver); for (Map.Entry<String, RestTemplate> entry : restTemplates.entrySet()) {
if (!CollectionUtils.isEmpty(dubboTranslatedAttributes)) { String beanName = entry.getKey();
adaptRestTemplate(entry.getValue(), dubboTranslatedAttributes); Map<String, Object> dubboTranslatedAttributes = getDubboTranslatedAttributes(
beanName, attributesResolver);
if (!CollectionUtils.isEmpty(dubboTranslatedAttributes)) {
adaptRestTemplate(entry.getValue(), dubboTranslatedAttributes);
}
} }
} }
} }
@ -132,7 +141,7 @@ public class DubboLoadBalancedRestTemplateAutoConfiguration
* {@link DubboTransported @DubboTransported} * {@link DubboTransported @DubboTransported}
* *
* @param beanName the bean name of {@link LoadBalanced @LoadBalanced} * @param beanName the bean name of {@link LoadBalanced @LoadBalanced}
* {@link RestTemplate} * {@link RestTemplate}
* @param attributesResolver {@link DubboTransportedAttributesResolver} * @param attributesResolver {@link DubboTransportedAttributesResolver}
* @return non-null {@link Map} * @return non-null {@link Map}
*/ */
@ -144,10 +153,10 @@ public class DubboLoadBalancedRestTemplateAutoConfiguration
AnnotatedBeanDefinition annotatedBeanDefinition = (AnnotatedBeanDefinition) beanDefinition; AnnotatedBeanDefinition annotatedBeanDefinition = (AnnotatedBeanDefinition) beanDefinition;
MethodMetadata factoryMethodMetadata = annotatedBeanDefinition MethodMetadata factoryMethodMetadata = annotatedBeanDefinition
.getFactoryMethodMetadata(); .getFactoryMethodMetadata();
attributes = factoryMethodMetadata != null attributes = factoryMethodMetadata != null ? Optional
? Optional.ofNullable(factoryMethodMetadata .ofNullable(factoryMethodMetadata
.getAnnotationAttributes(DUBBO_TRANSPORTED_CLASS_NAME)).orElse(attributes) .getAnnotationAttributes(DUBBO_TRANSPORTED_CLASS_NAME))
: Collections.emptyMap(); .orElse(attributes) : Collections.emptyMap();
} }
return attributesResolver.resolve(attributes); return attributesResolver.resolve(attributes);
} }
@ -158,8 +167,8 @@ public class DubboLoadBalancedRestTemplateAutoConfiguration
* *
* @param restTemplate {@link LoadBalanced @LoadBalanced} {@link RestTemplate} Bean * @param restTemplate {@link LoadBalanced @LoadBalanced} {@link RestTemplate} Bean
* @param dubboTranslatedAttributes the annotation dubboTranslatedAttributes * @param dubboTranslatedAttributes the annotation dubboTranslatedAttributes
* {@link RestTemplate} bean being annotated * {@link RestTemplate} bean being annotated
* {@link DubboTransported @DubboTransported} * {@link DubboTransported @DubboTransported}
*/ */
private void adaptRestTemplate(RestTemplate restTemplate, private void adaptRestTemplate(RestTemplate restTemplate,
Map<String, Object> dubboTranslatedAttributes) { Map<String, Object> dubboTranslatedAttributes) {
@ -188,4 +197,9 @@ public class DubboLoadBalancedRestTemplateAutoConfiguration
this.classLoader = classLoader; this.classLoader = classLoader;
} }
@Override
public void setApplicationContext(ApplicationContext applicationContext) {
this.applicationContext = applicationContext;
}
} }