Java中Retry方法的简单实现
作者：Smallc0de
这篇文章主要介绍了Java中Retry方法的简单实现,Retry主要是利用Java的lambda表达式和线程接口实现有返回值和无返回值的重试,思考了下就写了一个简易Retry功能分享出来,需要的朋友可以参考下
前言
最近有一个需求，要调用外部API满足业务逻辑，然后就用到了重试机制，又联想到不止API，其实很多中间件，比如Reids、各类MQ、Kafka，这些涉及到连接失败的读取写入需求都应该有重试机制。
虽然最终需求使用了Spring自带的Retry机制，但是这一套还是太重了，思考了下就写了一个简易Retry功能分享出来。
主要是利用Java的lambda表达式和线程接口实现有返回值和无返回值的重试。
代码
public class Retry {
public static <T> T retry(Callable<T> callable, int num) {
Retry.check(num >= 1, "次数需要大于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, "次数需要大于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), "传入参数为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做参数，可以换成Callable的实例
Retry.retry(test, 3);
//使用方法做参数
Retry.retry(()->supplierTestNoReturn("a"), 3);
//使用方法做参数接受返回值
String back = Retry.retry(()->supplierTest("a"), 3);
System.out.println(back);
//直接写有返回值
Retry.retry(()->{
//TODO 业务逻辑调用外部API等等
System.out.println("返回值");
return "返回值";
}, 3);
//直接写无返回值
Retry.retry(()->{
//TODO 业务逻辑调用外部API等等
System.out.println("无返回值");
}, 3);
}
//非静态直接使用
public void test(){
Retry.retry(()->normal("aa"), 3);
}
@Override
public void run() {
//TODO 业务逻辑、调用外部API等等
System.out.println("Runnable Test");
}
public static String supplierTest(String str){
//TODO 业务逻辑、调用外部API等等
return "Test "+ str;
}
public static void supplierTestNoReturn(String str){
//TODO 业务逻辑、调用外部API等等
System.out.println("No Return "+str);
}
public String normal(String a){
//TODO 业务逻辑、调用外部API等等
return "Test";
}
}另一个版本
例子写出来以后，觉得也可以用Supplier去替代Callable也是可以的，只需要一些小改动，如下一样可以使用。
public static <T> T retry(Supplier<T> supplier, int num) {
Retry.check(num >= 1, "次数需要大于0");
Exception exception = null;
for (int i = 0; i < num; i++) {
try {
return supplier.get();
} catch (Exception e) {
exception = e;
}
}
throw Retry.runtimeException(exception);
}到此这篇关于Java中Retry方法的简单实现的文章就介绍到这了,更多相关Java的Retry方法内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！