原创

详解Spring Retry实现原理

本文通过一个简单的例子演示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();
 }
}
 测试 通过一个用户相关的业务方法来测试上面的代码 接口定义:
package org.java.base.springretry;
public interface UserService {

 void add() throws Exception;

 void query() throws Exception;
}
接口实现:
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();
 }
}
测试:
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();//失败重试
 }
}
add方法不添加重试注解,程序异常结束,query方法添加重试注解,设置重试3次,运行效果如下 1528270952(1) 以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持我们。
正文到此结束
Loading...