Spring Retry實現原理的示例分析

這篇文章主要為大家展示了“Spring Retry實現原理的示例分析”，內容簡而易懂，條理清晰，希望能夠幫助大家解決疑惑，下面讓小編帶領大家一起研究并學習一下“Spring Retry實現原理的示例分析”這篇文章吧。

注解定義

package retry.annotation;

import java.lang.annotation.*;

/**
 * Created by Jack.wu on 2016/9/30.
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Retryable {

  int maxAttemps() default 0;

}

代理實現

以Cglib作為代理工具，先來寫個Callback實現，這也是重試的實現的核心邏輯

package retry.interceptor;

import net.sf.cglib.proxy.MethodInterceptor;
import net.sf.cglib.proxy.MethodProxy;
import retry.annotation.Retryable;

import java.lang.reflect.Method;

/**
 * Created by Jack.wu on 2016/9/30.
 */
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 retry.core;
import net.sf.cglib.proxy.Enhancer;
import retry.interceptor.AnnotationAwareRetryOperationsInterceptor;

/**
 * Created by Jack.wu on 2016/9/30.
 */
public class SpringRetryProxy {

  public Object newProxyInstance(Object target){
    Enhancer enhancer = new Enhancer();
    enhancer.setSuperclass(target.getClass());
    enhancer.setCallback(new AnnotationAwareRetryOperationsInterceptor());
    return enhancer.create();
  }
}

 測試

通過一個用戶相關的業務方法來測試上面的代碼

接口定義：

package facade;

/**
 * Created by Jack.wu on 2016/9/26.
 */
public interface UserFacade {

  void add() throws Exception;

  void query() throws Exception;
}

接口實現：package facade.impl;

import facade.UserFacade;
import retry.annotation.Retryable;
/**
 * Created by Jack.wu on 2016/9/26.
 */
public class UserFacadeImpl implements UserFacade {
  @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();
  }
}

測試：

public class Main {

  public static void main(String[] args) throws Exception{
    UserFacadeImpl user = new UserFacadeImpl();
    //SpringRetry代理測試
    SpringRetryProxy springRetryProxy = new SpringRetryProxy();
    UserFacade u = (UserFacade)springRetryProxy.newProxyInstance(user);
    //u.add();//失敗不重試
    u.query();//失敗重試
  }
}

add方法不添加重試注解，程序異常結束，query方法添加重試注解，設置重試3次，運行效果如下

以上是“Spring Retry實現原理的示例分析”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！