作者简介:
华哥
10年+后端开发工作经验,
主要分享:关于java体系的知识,如:java基础知识/数据结算/算法,Spring/MyBatis/Netty源码分析,高并发/高性能/分布式/微服务架构的原理,JVM性能优化等。
公众号:java杂记
本文通过一个简单的例子演示Spring Retry的实现原理,例子中定义的注解只包含重试次数属性,实际上Spring Retry中注解可设置属性要多的多,单纯为了讲解原理,所以弄简单点,关于Spring Retry可查阅相关文档、博客。
注解定义
<pre style="margin: 0px; padding: 0px; max-width: 100%; box-sizing: border-box; overflow-wrap: break-word !important; overflow: auto;">package org.java.base.springretry;
import java.lang.annotation.*;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Retryable {int maxAttemps() default 0;
}</pre>
代理实现
以Cglib作为代理工具,先来写个Callback实现,这也是重试的实现的核心逻辑
<pre style="margin: 0px; padding: 0px; max-width: 100%; box-sizing: border-box; overflow-wrap: break-word !important; overflow: auto;">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;
}
}</pre>
然后是写个代理类,使用AnnotationAwareRetryOperationsInterceptor作为拦截器
<pre style="margin: 0px; padding: 0px; max-width: 100%; box-sizing: border-box; overflow-wrap: break-word !important; overflow: auto;">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();
}
}</pre>
** 测试**
通过一个用户相关的业务方法来测试上面的代码
接口定义:
<pre style="margin: 0px; padding: 0px; max-width: 100%; box-sizing: border-box; overflow-wrap: break-word !important; overflow: auto;">package org.java.base.springretry;
public interface UserService {void add() throws Exception;
void query() throws Exception;
}</pre>
接口实现:
<pre style="margin: 0px; padding: 0px; max-width: 100%; box-sizing: border-box; overflow-wrap: break-word !important; overflow: auto;">package org.java.base.springretry;
public class UserServiceImpl implements UserService {
@Override
public void add() throws Exception {
System.out.println("添加用户。。。");
throw new RuntimeException();
}@Override
@Retryable(maxAttemps = 3)
public void query() {
System.out.println("查询用户。。。");
throw new RuntimeException();
}
}</pre>
测试:
<pre style="margin: 0px; padding: 0px; max-width: 100%; box-sizing: border-box; overflow-wrap: break-word !important; overflow: auto;">package org.java.base.springretry;
public class TestRetry {public static void main(String[] args) throws Exception{
UserServiceImpl user = new UserServiceImpl();
//SpringRetry代理测试
SpringRetryProxy springRetryProxy = new SpringRetryProxy();
UserService u = (UserService)springRetryProxy.newProxyInstance(user);
//u.add();//失败不重试
u.query();//失败重试
}
}</pre>
add方法不添加重试注解,程序异常结束,query方法添加重试注解,设置重试3次,运行效果如下
作者简介:
华哥
10年+后端开发工作经验,
主要分享:关于java体系的知识,如:java基础知识/数据结算/算法,Spring/MyBatis/Netty源码分析,高并发/高性能/分布式/微服务架构的原理,JVM性能优化等。
公众号:java杂记
本文已经获得华哥授权转发,其他人若有兴趣转载,请直接联系作者授权。