本文通過一個簡單的例子演示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次，執行效果如下
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。
