工具類:
import java.lang.reflect.Method;
public class RetryUtil {
private static ThreadLocal<Integer> retryTimesInThread = new ThreadLocal<>();
/**
* 設置當前方法重試次數
*
* @param retryTimes
* @return
*/
public static RetryUtil setRetryTimes(Integer retryTimes) {
if (retryTimesInThread.get() == null)
retryTimesInThread.set(retryTimes);
return new RetryUtil();
}
/**
* 重試當前方法
* <p>按順序傳入調用者方法的所有參數</p>
* @param args
* @return
*/
public Object retry(Object... args) {
try {
Integer retryTimes = retryTimesInThread.get();
if (retryTimes <= 0) {
retryTimesInThread.remove();
return null;
}
retryTimesInThread.set(--retryTimes);
String upperClassName = Thread.currentThread().getStackTrace()[2].getClassName();
String upperMethodName = Thread.currentThread().getStackTrace()[2].getMethodName();
Class clazz = Class.forName(upperClassName);
Object targetObject = clazz.newInstance();
Method targetMethod = null;
for (Method method : clazz.getDeclaredMethods()) {
if (method.getName().equals(upperMethodName)) {
targetMethod = method;
break;
}
}
if (targetMethod == null)
return null;
targetMethod.setAccessible(true);
return targetMethod.invoke(targetObject, args);
} catch (Exception e) {
e.printStackTrace();
return null;
}
}
}
調用:
RetryUtil.setRetryTimes(3).retry(url, workOrderPost);
為了防止多線程情況下出現並發問題,這里定義了一個 ThreadLocal 變量來存儲當前線程的重試次數。然后通過 setRetryTimes ,一個靜態方法來設置這個重試次數,並返回一個 RetryUtil 對象。
調用者通過返回的 RetryUtil 對象調用 retry 方法實現重試。retry 方法接收一個可變參數,因為調用者實際的參數不確定,這里要求按順序傳入調用者方法的所有參數。
接下來判斷 ThreadLocal 變量是否小於等於 0 ,如果是,則說明重復次數已達到,返回 null;如果不是,則讓 ThreadLocal 變量減一。接下來:
String upperClassName = Thread.currentThread().getStackTrace()[2].getClassName(); String upperMethodName = Thread.currentThread().getStackTrace()[2].getMethodName();
來獲取當前方法(retry)的上層方法名和上層類名。Thread.currentThread().getStackTrace() 得到線程的方法棧數組,數組的第二個元素 Thread.currentThread().getStackTrace() [1] 為當前方法棧,第三個元素 Thread.currentThread().getStackTrace() [2] 為上層方法棧,通過上層方法的棧幀得到上層方法的方法名和類名。
下面就是通過反射獲取該類的所有方法,循環判斷方法名是否等於所要重復執行的方法,如果是的話,執行該方法,參數就是傳入可變參數。
注意睡眠重試確保之前操作事務提交,避免超時。