本文通过一个简单的例子演示Spring Retry的实现原理，例子中定义的注解只包含重试次数属性，实际上Spring Retry中注解可设置属性要多的多，单纯为了讲解原理，所以弄简单点，关于Spring Retry可查阅相关文档、博客。
注解定义
package org.java.base.springretry;
import java.lang.annotation.*;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Retryable {
int maxAttemps() default 0;
}
复制代码
代理实现
以Cglib作为代理工具，先来写个Callback实现，这也是重试的实现的核心逻辑
package org.java.base.springretry;
import java.lang.reflect.Method;
import org.springframework.cglib.proxy.MethodInterceptor;
import org.springframework.cglib.proxy.MethodProxy;
public class AnnotationAwareRetryOperationsInterceptor implements MethodInterceptor{
//记录重试次数
private int times = 0;
@Override
public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
//获取拦截的方法中的Retryable注解
Retryable retryable = method.getAnnotation(Retryable.class);
if(retryable == null){
return proxy.invokeSuper(obj,args);
}else{ //有Retryable注解，加入异常重试逻辑
int maxAttemps = retryable.maxAttemps();
try {
return proxy.invokeSuper(obj,args);
} catch (Throwable e) {
if(times++ == maxAttemps){
System.out.println("已达最大重试次数：" + maxAttemps + ",不再重试！");
}else{
System.out.println("调用" + method.getName() + "方法异常，开始第" + times +"次重试。。。");
//注意这里不是invokeSuper方法，invokeSuper会退出当前interceptor的处理
proxy.invoke(obj,args);
}
}
}
return null;
}
}
复制代码
然后是写个代理类，使用AnnotationAwareRetryOperationsInterceptor作为拦截器
package org.java.base.springretry;
import org.springframework.cglib.proxy.Enhancer;
public class SpringRetryProxy {
public Object newProxyInstance(Object target){
Enhancer enhancer = new Enhancer();
enhancer.setSuperclass(target.getClass());
enhancer.setCallback(new AnnotationAwareRetryOperationsInterceptor());
return enhancer.create();
}
}复制代码 |
