前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >RestTemplate 创建重试机制

RestTemplate 创建重试机制

作者头像
分享干货的你
发布2021-04-23 11:05:43
5.6K0
发布2021-04-23 11:05:43
举报
文章被收录于专栏:分享干货的你

我们使用resttemplate 调用其它服务的时候,因为网络波动造成的对方服务异常或者对方服务降级后又好了。 这里我们就需要resttempalte 的重试机制了。

这里面就要使用Apache 的httpclient 的HttpRequestRetryHandler 。

在通过HttpComponentsClientHttpRequestFactory 这个request 工厂构建。

具体代码

代码语言:javascript
复制
public class RetryRestTemplate {

    /**
     * 失败重试默认3次,重试间隔时间=次数*1000 ms. 默认毫秒
     * @return
     */
    public static RestTemplate build(){
        return build(3,1000);
    }

    /**
     * @param retryTimes  重试次数
     * @param retryIntervalTime 每次重试间隔时间=重试次数*retryIntervalTime
     * @return
     */
    public static RestTemplate build(int retryTimes, long retryIntervalTime){
        RestTemplate restTemplate = new RestTemplate();
        //用UTF-8 StringHttpMessageConverter替换默认StringHttpMessageConverter
        List<HttpMessageConverter<?>> newMessageConverters = new ArrayList<>();
        for(HttpMessageConverter<?> converter : restTemplate.getMessageConverters()){
            if(converter instanceof StringHttpMessageConverter){
                // 默认的是ios 8859-1
                StringHttpMessageConverter messageConverter = new StringHttpMessageConverter(StandardCharsets.UTF_8);
                newMessageConverters.add(messageConverter);
            }else {
                newMessageConverters.add(converter);
            }
        }
        restTemplate.setMessageConverters(newMessageConverters);

        HttpClientBuilder httpClientBuilder = HttpClients.custom();
        // 只有io异常才会触发重试
        HttpRequestRetryHandler handler = new HttpRequestRetryHandler() {
            @Override
            public boolean retryRequest(IOException exception, int curRetryCount, HttpContext context) {
                // curRetryCount 每一次都会递增,从1开始
                if (curRetryCount > retryTimes) {
                    return false;
                }
                try {
                    //重试延迟
                    Thread.sleep(curRetryCount*retryIntervalTime);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                if (exception instanceof ConnectTimeoutException
                        || exception instanceof NoHttpResponseException || exception instanceof ConnectException) {
                    return true;
                }

                HttpClientContext clientContext = HttpClientContext.adapt(context);
                org.apache.http.HttpRequest request = clientContext.getRequest();
                boolean idempotent = !(request instanceof HttpEntityEnclosingRequest);
                if (idempotent) {
                    // 如果请求被认为是幂等的,那么就重试。即重复执行不影响程序其他效果的
                    return true;
                }
                return false;
            }
        };
        httpClientBuilder.setRetryHandler(handler).setMaxConnTotal(400);

        // httpClient连接配置,底层是配置RequestConfig
        HttpComponentsClientHttpRequestFactory clientHttpRequestFactory = new HttpComponentsClientHttpRequestFactory(httpClientBuilder.build());
        // 连接超时
        clientHttpRequestFactory.setConnectTimeout(30000);
        // 数据读取超时时间,即SocketTimeout
        clientHttpRequestFactory.setReadTimeout(30000);
        // 连接不够用的等待时间,不宜过长,必须设置,比如连接不够用时,时间过长将是灾难性的
        clientHttpRequestFactory.setConnectionRequestTimeout(10000);
        // 缓冲请求数据,默认值是true。通过POST或者PUT大量发送数据时,建议将此属性更改为false,以免耗尽内存。
         clientHttpRequestFactory.setBufferRequestBody(false);
        restTemplate.setRequestFactory(clientHttpRequestFactory);
        return restTemplate;
    }
}

测试结果:

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2021-04-12,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 分享干货的你 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档