重试utils

import com.alibaba.fastjson.JSONObject;
import org.slf4j.Logger;

import java.util.function.Function;
import java.util.function.Predicate;
import java.util.function.Supplier;

/**
 * @description 重试utils
 */
public class RetryUtils {

    /**
     * 默认用法（最多重试3次），有入参，返回null表示多次重试依旧失败
     */
    public static <T, R> R retryDo(Function<T, R> doFunction, T doFunctionRequest, Predicate<R> checkFunction, Logger log, String logDesc) {
        return retryDo(doFunction, doFunctionRequest, checkFunction, 2, log, logDesc);
    }

    /**
     * 默认用法（最多重试3次），无入参，返回null表示多次重试依旧失败
     */
    public static <R> R retryDo(Supplier<R> supplier, Predicate<R> checkFunction, Logger log, String logDesc) {
        return retryDo(supplier, checkFunction, 2, log, logDesc);
    }


    /**
     * 返回null表示多次重试依旧失败
     */
    public static <T, R> R retryDo(Function<T, R> doFunction, T doFunctionRequest, Predicate<R> checkFunction, Integer maxRetryTimes, Logger log, String logDesc) {
        R result = null;

        boolean checkBool = false;
        int retryTimes = 0;
        while (!checkBool && retryTimes <= maxRetryTimes) {
            try {
                log.info("{}，request={},retryTimes={}", logDesc, JSONObject.toJSON(doFunctionRequest), retryTimes);
                result = doFunction.apply(doFunctionRequest);
                log.info("{}，request={},retryTimes={},result={}", logDesc, JSONObject.toJSON(doFunctionRequest), retryTimes, JSONObject.toJSON(result));
                checkBool = checkFunction.test(result);
                if (!checkBool) {
                    log.info("{}，request={},retryTimes={},result={}，校验不通过", logDesc, JSONObject.toJSON(doFunctionRequest), retryTimes, JSONObject.toJSON(result));
                }
            } catch (Throwable throwable) {
                log.error(logDesc + "-发生未知异常", throwable);
            }
            retryTimes++;
        }
        if (!checkBool) {
            return null;
        }
        return result;
    }

    /**
     * 返回null表示多次重试依旧失败
     */
    public static <R> R retryDo(Supplier<R> supplier, Predicate<R> checkFunction, Integer maxRetryTimes, Logger log, String logDesc) {
        R result = null;

        Boolean checkBool = false;
        int retryTimes = 0;
        while (!checkBool && retryTimes <= maxRetryTimes) {
            try {
                log.info("{},retryTimes={}", logDesc, retryTimes);
                result = supplier.get();
                log.info("{}，retryTimes={},result={}", logDesc, retryTimes, JSONObject.toJSON(result));
                checkBool = checkFunction.test(result);
                if (!checkBool) {
                    log.info("{}，retryTimes={},result={}，校验不通过", logDesc, retryTimes, JSONObject.toJSON(result));
                }
            } catch (Throwable throwable) {
                log.error(logDesc + "-发生未知异常", throwable);
            }
            retryTimes++;
        }
        if (!checkBool) {
            return null;
        }
        return result;
    }

}