快捷導(dǎo)航

Java中Retry方法的簡單實現(xiàn)

更新時間：2024年01月29日 10:18:55 作者：Smallc0de

這篇文章主要介紹了Java中Retry方法的簡單實現(xiàn),Retry主要是利用Java的lambda表達(dá)式和線程接口實現(xiàn)有返回值和無返回值的重試,思考了下就寫了一個簡易Retry功能分享出來,需要的朋友可以參考下

前言

最近有一個需求，要調(diào)用外部API滿足業(yè)務(wù)邏輯，然后就用到了重試機制，又聯(lián)想到不止API，其實很多中間件，比如Reids、各類MQ、Kafka，這些涉及到連接失敗的讀取寫入需求都應(yīng)該有重試機制。

雖然最終需求使用了Spring自帶的Retry機制，但是這一套還是太重了，思考了下就寫了一個簡易Retry功能分享出來。

主要是利用Java的lambda表達(dá)式和線程接口實現(xiàn)有返回值和無返回值的重試。

代碼

public class Retry {
    public static <T> T retry(Callable<T> callable, int num) {
        Retry.check(num >= 1, "次數(shù)需要大于0");
        Exception exception = null;
        for (int i = 0; i < num; i++) {
            try {
                return callable.call();
            } catch (Exception e) {
                exception = e;
            }
        }
        throw Retry.runtimeException(exception);
    }
    public static void retry(Runnable runnable, int num) {
        Retry.check(num >= 1, "次數(shù)需要大于0");
        Exception exception = null;
        for (int i = 0; i < num; i++) {
            try {
                runnable.run();
                return;
            } catch (Exception e) {
                exception = e;
            }
        }
        throw Retry.runtimeException(exception);
    }
    public static void check(boolean check, String message) {
        if (!check) {
            if (Objects.isNull(message)) {
                throw new IllegalArgumentException("Check Requirement Error.");
            } else {
                throw new IllegalArgumentException("Check Requirement Error:" + message);
            }
        }
    }
    public static RuntimeException runtimeException(Exception exception) {
        Retry.check(Objects.nonNull(exception), "傳入?yún)?shù)為null");
        if (exception instanceof RuntimeException) {
            throw new RuntimeException(exception);
        }
        throw new RuntimeException();
    }
}

使用樣例

public class RetryTest implements Runnable{
    public static void main(String[] args) {
        RetryTest test=new RetryTest();
        //直接用Runnable做參數(shù)，可以換成Callable的實例
        Retry.retry(test, 3);
        //使用方法做參數(shù)
        Retry.retry(()->supplierTestNoReturn("a"), 3);
        //使用方法做參數(shù)接受返回值
        String back = Retry.retry(()->supplierTest("a"), 3);
        System.out.println(back);
        //直接寫有返回值
        Retry.retry(()->{
            //TODO 業(yè)務(wù)邏輯調(diào)用外部API等等
            System.out.println("返回值");
            return "返回值";
        }, 3);
 		//直接寫無返回值
        Retry.retry(()->{
            //TODO 業(yè)務(wù)邏輯調(diào)用外部API等等
            System.out.println("無返回值");
        }, 3);
    }
    //非靜態(tài)直接使用
    public void test(){
        Retry.retry(()->normal("aa"), 3);
    }
    @Override
    public void run() {
        //TODO 業(yè)務(wù)邏輯、調(diào)用外部API等等
        System.out.println("Runnable Test");
    }
    public static String supplierTest(String str){
        //TODO 業(yè)務(wù)邏輯、調(diào)用外部API等等
        return "Test "+ str;
    }
    public static void supplierTestNoReturn(String str){
        //TODO 業(yè)務(wù)邏輯、調(diào)用外部API等等
        System.out.println("No Return "+str);
    }
    public String normal(String a){
        //TODO 業(yè)務(wù)邏輯、調(diào)用外部API等等
        return "Test";
    }
}

另一個版本

例子寫出來以后，覺得也可以用Supplier去替代Callable也是可以的，只需要一些小改動，如下一樣可以使用。

    public static <T> T retry(Supplier<T> supplier, int num) {
        Retry.check(num >= 1, "次數(shù)需要大于0");
        Exception exception = null;
        for (int i = 0; i < num; i++) {
            try {
                return supplier.get();
            } catch (Exception e) {
                exception = e;
            }
        }
        throw Retry.runtimeException(exception);
    }

到此這篇關(guān)于Java中Retry方法的簡單實現(xiàn)的文章就介紹到這了,更多相關(guān)Java的Retry方法內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: