跳到主要內容

Java Design - 可重複使用的Retry

Why...

當系統發生了例外情況時,夠強健的系統會重新嘗試(retry)發生問題的操作,最常見的例子就是連線中斷的重連。也有可能會另尋其它路徑,像是使用備援的系統或資料來源。一開始我採用了while/for loop的方式做retry,然而當這樣的程式碼夠多後,看了也挺令人厭煩的。Spring有提供RetryTemplate,讓你可以實做想要的Retry。但Spring實在太大包了,除非系統中一定會用到Spring,否則要包這個東西進去,也是挺OOXX的。

I Thinking & Trying

於是我開始嘗試著造輪子。我參考Reference的三篇文章,並設計了一個折中的方式,可以滿足大部分的需求(我需要的需求)。

  1. 設定重試次數
  2. 設定重試條件
  3. 設定重試延遲時間
  4. 可做Alternative操作

我參考了Reference中的三篇文章,設計了一個我認為好擴充與方便使用的RetryableTask類別,讓我可以將這些繁瑣的動作給包裝起來:

概念相當簡單:

  • RetryableTask負責執行這些retry的動作,它依賴於Callable、ISleepStrategy、IRetryablePolicy類別。
  • Callable類別讓Programmer將操作給包裝起來,管你是要從DB還是從檔案取資料,這裡就是提供功能的操作流程。
  • ISleepStrategy負責讓RetryableTask知道每次Retry時,需要先等待幾秒鐘。目前提供了BasicSleepStrategy(固定時間)與VariableSleepStrategy(變化時間)。
  • IRetryablePolicy會根據每次操作的結果,決定是否要Retry。這裡使用了Composite Pattern,讓你可以使用多個Policy去控制Retry策略。目前提供了AttemptRetryablePolicy(最大次數)、NullRetryablePolicy(不可為NULL)、ExceptioRetryablePolicy(例外情形)三種策略。

Programming

IRetryableTask

首先讓我們看看RetryableTask的member與constructor,預設是使用BasicSleepStrategy與ExceptioRetryablePolicy,當然也可以透過set method去更改。功能執行的主體則是透過Callable,Client必須把它的操作流程先實做好再丟進來。

private Callable<T> mCallable = null;
private Object mResult = null;
private ISleepStrategy mSleepStrategy = new BasicSleepStrategy();
private IRetryablePolicy mRetryablePolicy = new ExceptioRetryablePolicy();
 
public RetryableTask(Callable<T> callable){
	mCallable = callable;
}

接著是最核心的部分。基本上就是透過ISleepStrategy與IRetryablePolicy去控制流程,透過mCallable.call()去執行主要功能。萬一都重試失敗,就將結果回傳或將例外往上丟。(結果也許會由mCallable.call()回傳一個Default value)

@Override
public T call() throws Throwable {
	boolean isFirstTime = true;
	do {
		if(!isFirstTime && mSleepStrategy != null){
			ThreadUtil.sleep(mSleepStrategy.getSleepTime());
		}
		try {
			T result = mCallable.call();
			mResult = result;
		} catch( Exception e ){
			mResult = e;
		}
	} while( isNeedToRetry(mResult) );
 
	if( mResult instanceof Throwable ){
		throw (Throwable)mResult;
	}
	return (T)mResult;
}
 
private boolean isNeedToRetry(Object aData){
	if( mRetryablePolicy != null ){
		return mRetryablePolicy.needToRetry(aData);
	}
	return false;
}

ISleepStrategyISleepStrategy

就是實做每次Retry時,你要Sleep多久的規則而已。

public class VariableSleepStrategy implements ISleepStrategy {
 
	private int mCurrentIndex = 0;
	private long[] mSleepTimes;
	private int mMaxLenghth = 0;
 
	public VariableSleepStrategy(long[] sleepTimes){
		mMaxLenghth = sleepTimes.length;
		mSleepTimes = new long[mMaxLenghth];
		System.arraycopy(sleepTimes, 0, mSleepTimes, 0, mMaxLenghth);
	}
 
	@Override
	public long getSleepTime() {
		if( mCurrentIndex == mMaxLenghth ){
			throw new RuntimeException("Over the max length.");
		}
		return mSleepTimes[mCurrentIndex++];
	}
}

IRetryablePolicy

IRetryablePolicy會根據執行結果決定是否要Retry,我以ExceptionRetryablePolicy為例。ExceptionRetryablePolicy提供三個建構子,一個支援如果執行結果為Exception類別或子類別就要Retry;另外兩個會根據你給定的例外類別列表,結果有在其中才Retry。

public class ExceptionRetryablePolicy implements IRetryablePolicy {
 
	private List<Class<? extends Throwable>> mExceptinList = null;
 
	public ExceptionRetryablePolicy() {
		this(Exception.class);
	}
 
	public ExceptionRetryablePolicy(Class<? extends Throwable> throwableClass) {
		mExceptinList = new ArrayList<Class<? extends Throwable>>();
		mExceptinList.add(throwableClass);
	}
 
	public ExceptionRetryablePolicy(List<Class<? extends Throwable>> exceptinList) {
		mExceptinList = exceptinList;
	}
 
	@Override
	public boolean needToRetry(Object data) {
		if( data == null || mExceptinList == null )
			return false;
		for( Class<? extends Throwable> throwableClass : mExceptinList ){
			if( throwableClass.isInstance(data)){
				return true;
			}
		}
		return false;
	}
}

CompositeRetryablePolicy支援多個策略,像是你可以同時支援最大次數、例外情況或NULL情況。實做就是去呼叫各Policy的needToRetry去決定。

public class CompositeRetryablePolicy implements IRetryablePolicy {
 
	private List<IRetryablePolicy> mPolicyList = null;
 
	public CompositeRetryablePolicy(){
 
	}
 
	public CompositeRetryablePolicy(List<IRetryablePolicy> policyList){
		mPolicyList = policyList;
	}
 
	@Override
	public boolean needToRetry(Object data) {
		if( mPolicyList == null ){
			return false;
		}
		boolean needRetry = !mPolicyList.isEmpty();
		for( IRetryablePolicy policy : mPolicyList ){
			if(!policy.needToRetry(data)){
				needRetry = false;
			}
		}
		return needRetry;
	}
}

NullRetryablePolicy是在結果為NULL時,去做Retry;AttemptRetryablePolicy則是用來控制最大的重試次數。

Testing

我透過Powermock,並實做一個Alternative retry給大家看看。首先讓我們mock要呼叫的method: userDao1.getUser()與userDao2.getUser(),假設userDao1 from DB,userDao2 from file。userDao1為第一次執行使用,會拋出一個例外;userDao2在第二次使用,會回傳正確結果。

IUser user_expect = PowerMock.createMock(IUser.class);
 
String errorMsg = "Testing error msg";
final IUserDao userDao1 = PowerMock.createStrictMock(IUserDao.class);
userDao1.getUser(EasyMock.anyObject(String.class));
PowerMock.expectLastCall().andThrow(new RuntimeException(errorMsg)).once();
 
final IUserDao userDao2 = PowerMock.createStrictMock(IUserDao.class);
userDao2.getUser(EasyMock.anyObject(String.class));
PowerMock.expectLastCall().andReturn(user_expect).once();
 
PowerMock.replayAll();

Callable call()的實做透過userDao1與userDao2去交互執行,若userDao1.getUser執行失敗就會用userDao2.getUser。像要取得一個local port,也許就可以透過遞增或遞減port number去實做。

Callable<IUser> platformUtil = new Callable<IUser>() {
	private boolean switchFlag = true;
	@Override
	public IUser call() throws Exception {
		switchFlag = !switchFlag;
		if( !switchFlag )
			return userDao1.getUser("1234");
		else {
			return userDao2.getUser("1234");
		}
	}
};

RetryableTask的部分使用了AttemptRetryablePolicy與ExceptionRetryablePolicy,要求重試次數小於3且發生例外時要Retry。最後我們期望的是能夠透過userDao2.getUser取得與user_expect相同的結果,因此使用了PowerMock.verifyAll()去確認那些mock object都有被呼叫到。

IRetryableTask<IUser> retryableTask = new RetryableTask<IUser>(platformUtil);
IRetryablePolicy compositeRetryablePolicy = new CompositeRetryablePolicy(Arrays.asList(new IRetryablePolicy[]{
		new AttemptRetryablePolicy(3),
		new ExceptionRetryablePolicy()
}));
retryableTask.setRetryablePolicy(compositeRetryablePolicy);
try {
	assertEquals(user_expect, retryableTask.call());
} catch (Throwable e) {
	fail();
}
 
PowerMock.verifyAll();

Summary

就目前的需求,這是我所能想到的設計。也許可以把更多的東西做抽像化來增加更多的擴充性,但是目前這樣就能滿足我了。

友藏內心獨白: 有先請教過學姐,再稍稍做修改。也許還達不到最令人滿意的設計。

Reference

留言

這個網誌中的熱門文章

PostgreSQL - Unattended installation on windows

Introduction 要將別人軟體包裝到自己軟體中,不可或缺的東西就是Unattended installation。以Unattended installation來說,我們可以選擇透過Installer的silent mode安裝,也可以透過把目標軟體做成portable的版本。本篇文章分享這兩種方法,教導大家如何將PostgreSQL透過Unattended installation方式安裝到目標系統成為service。 Note. 本篇以PostgreSQL 10.7為例。 Install with installer Tips 安裝程式或反安裝程式的參數,除了可以直接上官網搜尋Installation User Guide以外,也可以直接使用help參數查詢: postgresql- 10.7 - 2 -windows-x64.exe --help Windows安裝程式主要有EnterpriseDB與BigSQL兩種。BigSQL版本安裝元件是透過網路下載且支援參數不如EnterpriseDB版本多,以我們需求來說,我們傾向於使用EnterpriseDB版本。接下來分享給大家安裝與反安裝方法。 Installation @ echo off set INSTALL_DIR =C:\postgres10 set INSTALLER =postgresql- 10.7 - 2 -windows-x64.exe   rem options for installation set SSMDB_SERVICE =postgresql- 10 set MODE =--unattendedmodeui none --mode unattended   set DB_PASSWD =--superpassword postgres set DB_PORT =--serverport 5432   set SERVICE_NAME =--servicename % SSMDB_SERVICE %   set PREFIX =--prefix "%INSTALL_DIR%" set DATA_DIR =--datadir "%INSTALL_DIR%\data"   set OPTIONS =

How to install RIDE on Windows?

Introduction 多年沒在Windows上開發RobotFramework,趁著這次整理一下RIDE安裝方法。 目前RIDE最新版本與Python對應版本如下: (3.6 < python <= 3.11) Install current released version (2.0.8.1) with: pip install -U robotframework-ride 安裝Python 直接到Python官網找尋最新的3.11版本,我使用3.11.9: link 。安裝就是一直下一步而已。 安裝wxPython 每次安裝RIDE最困難的都是wxPython。看了一下 官網 描述,我就姑且相信一下: 接著進入下 載頁面 就有安裝教學。基本上就是到Python目錄下的Scripts直接執行以下command: pip install -U wxPython 安裝RIDE 接著就如RIDE官網所說,執行以下command: pip install -U robotframework-ride 啟動RIDE 直接在相同目錄下執行ride就可以啟動了,你也可以直接在桌面建ride連結,加快下次啟動時間。 沒想到這次這麼順利就安裝完成了。因為我是使用java去啟動robot framework,就不特別講要怎麼使用pip安裝robot framework了。

Hello World!

即將要搬家,因此舊網頁內容將慢慢轉移至Blogger。 如果要存取舊網頁,可以使用以下連結: https://wiki.tonylin.idv.tw/dokuwiki/doku.php