java 重试机制工具

我就是我 2023-02-21 12:51 137阅读 0赞

**使用场景：**业务中经常遇到需要重试机制对一段代码进行 多次执行常遇到的有 访问网络数据接口，由于网络等原因需要重试 两到多次，直到返回正确结果为止。

**已有方案：**

**    1. g**uava-retrying  开源项目

<dependency>
        <groupId>com.github.rholder</groupId>
        <artifactId>guava-retrying</artifactId>
        <version>2.0.0</version>
    </dependency>

2. spring-retry

<dependency>
        <groupId>org.springframework.retry</groupId>
        <artifactId>spring-retry</artifactId>
    </dependency>

以上两种方案都可以完美的解决 重试问题

但是上面两种方案都略显臃肿，需要依赖第三方项目（主要是没大看懂他们的源码），基于Java8 函数式编程（太香了）

开发一下工具Demo （只是个Demo,一种方案，可优化，求分享）

**上代码：**

package com.barton;
    
    import java.util.concurrent.TimeUnit;
    import java.util.function.BiFunction;
    import java.util.function.Supplier;
    
    /**
     * create by barton on 2020-6-30
     */
    public class RetryTest {
    
    
        /**
         * 重试机制
         * @param retryNumber 重试次数
         * @param waitTime 重试等待时间
         * @param retryIfExceptionType 重试一场类型
         * @param call 执行主体
         * @param retryIf 重试条件
         * @param <T>
         * @return T
         */
        public static <T> T retry(
                Integer retryNumber,
                Long waitTime,
                Class<? extends Throwable> retryIfExceptionType,
                Supplier<T> call,
                BiFunction<Integer,T,Boolean> retryIf
        ){
            T result = null;
            Integer currentNumber = 1;
            Boolean exceptionFlag = false;
            while (true){
                if(currentNumber > retryNumber){
                    return result;
                }
                exceptionFlag = false;
                try{
                    result = call.get();
                }catch (Exception e){
                    if(retryIfExceptionType != null
                            && e.getClass().getName().equals(retryIfExceptionType.getName())){
                        exceptionFlag = true;
                    }
                    e.printStackTrace();
                }
                if(exceptionFlag){
                    currentNumber++;
                    continue;
                }
    
                if(retryIf != null){
                    Boolean isRetry = retryIf.apply(currentNumber,result);
                    // true 重试，false 不重试返回
                    if(!isRetry){
                        return result;
                    }else {
                        //间隔一定时间重试
                        if(waitTime != null && waitTime > 0){
                            try {
                                TimeUnit.MILLISECONDS.sleep(waitTime);
                            } catch (InterruptedException e) {
                                e.printStackTrace();
                            }
                        }
                    }
                }
                currentNumber++;
            }
        }
    
        public static void main(String[] args) {
            String res = retry(3,
                    2000L,
                    NumberFormatException.class,
                    ()->{
                        // 执行主体
                        System.out.println("do something");
                        return "SUCCESS";
                    },(number,result)->{
                        System.out.println("retry:"+number);
                        // 判断是否需要重试，根据业务修改
                        if("SUCCESS".equals(result)){
                            return true;
                        }else{
                            return false;
                        }
                    });
            System.out.println(res);
        }
    }

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzM2NTI4Nzg4_size_16_color_FFFFFF_t_70][]

（注：只是个Demo ,需要使用的同学可优化，欢迎分享）

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzM2NTI4Nzg4_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200630214720884.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzM2NTI4Nzg4,size_16,color_FFFFFF,t_70