@scope("prototype") @loadbalanced注解负载均衡失效问题

时间:2023-03-20 马上就要下班了人气:0

正文

用RestTemplate想在有ribbon负载均衡的功能下又可以配置多例来注入一些不固定的参数，bean 单例的时候没问题多例的时候却失效了，百度很久完全没有这方面的答案，google了下starkoverflow有一篇回答解释了

工具类

@scope(“prototype”)就是单纯的new了一下，spring不管理bean，那注解就失效了，替代实现的方法就是传一个LoadBalancerInterceptor的实例，根据这个思路我写的工具类代码如下：

@Configuration
public class RestTemplateManager {
    @Resource
    private LoadBalancerInterceptor loadBalancerInterceptor;

    /**
     * Time:millisecond
     * @param readtimout
     * @param connectionRequestTimeout
     * @param connectTimeout
     * @return HttpComponentsClientHttpRequestFactory
     */
    public HttpComponentsClientHttpRequestFactory getHttpRequestFactory(int readtimout,int connectionRequestTimeout,int connectTimeout) {
        HttpComponentsClientHttpRequestFactory factory = new HttpComponentsClientHttpRequestFactory();
        factory.setReadTimeout(readtimout);
        factory.setConnectionRequestTimeout(connectionRequestTimeout);
        factory.setConnectTimeout(connectTimeout);
        factory.setBufferRequestBody(false);
        return factory;
    }

    @Bean
    @Scope("prototype")
    public RestTemplate getConfigRestTemplate(HttpComponentsClientHttpRequestFactory factory) {
        RestTemplate restTemplate = new RestTemplate();
        //解决编码问题
        restTemplate.getMessageConverters().set(1, new StringHttpMessageConverter(StandardCharsets.UTF_8));
        restTemplate.setRequestFactory(factory);
        //add loadbalancerInterceptor,list is empty default
        restTemplate.getInterceptors().add(loadBalancerInterceptor);
        return restTemplate;
    }

    @Bean
    @LoadBalanced
    public RestTemplate getRestTemplate() {
        RestTemplate restTemplate = new RestTemplate();
        //解决编码问题
        restTemplate.getMessageConverters().set(1, new StringHttpMessageConverter(StandardCharsets.UTF_8));
        return restTemplate;
    }
}

测试过默认情况 restTemplate.getInterceptors()是空的，加入loadBalancerInterceptor就可以实现@loadbalanced的功能了，LoadBalancerAutoConfiguration依赖于loadBalancerInterceptor，ribbonInterceptor又是在LoadBalancerAutoConfiguration中声明的，这就解释了为什么加入loadBalancerInterceptor可以实现负载均衡的效果，关于怎么取这个loadBalancerInterceptor我研究了下用@Resource注解注入，在没有声明spring管理的类中@Autowired的是取不了实例的，而@Resource可以直接匹配原始类型注入，而且貌似是单例的，正好符合需要就这样使用了，能力有限,有些东西也是一知半解，算是给需要的人提供一些思路吧。

加载全部内容