跳到主要內容

CompletableFuture - Carefully avoid transition task from one thread to another. It costs.

Problem

這個問題我使用以下範例來說明:

private ExecutorService es = Executors.newFixedThreadPool(2);
@Test
public void transitionTaskOfCallback(){
	CompletableFuture.supplyAsync(()->{
		return new Response();
	}, es)
	.thenAcceptAsync(response->{
		System.out.println(response);
	}, es);
}

這段程式透過supplyAsync將工作送給es去執行,當執行完畢後,會再透過thenAcceptAsync把後續處理交給es執行;或許你的程式執行起來,可以很順利地拿到你要的結果,但這裡做了白工。

由於這兩段都使用了async的方式,也都是把工作交由es去執行。假如es中有多個thread,這可能會造成額外的thread context switch;因為第一個工作與第二個工作並不一定會在同一個thread上執行。

How to resolve?

首先你必須要搞清楚,你到底在做什麼.. 假如你是很單純的在supplyAsync中做些計算,然後callback是要針對計算結果做處理,那幹嘛需要多此一舉把它宣告成async的寫法呢? 只要改成以下寫法,就可以讓你維持原本的Thread繼續執行:

@Test
public void fixTransitionTaskOfCallback(){
	CompletableFuture.supplyAsync(()->{
		return new Response();
	}, es)
	.thenAccept(response->{
		System.out.println(response);
	});
}

但假如你的使用情況像之前文章中,會將Blocking工作送到另外一個Thread,接著透過compose串接,就要考慮使用Async方式去延續工作:

@Test
public void testSendAsync(){
	CompletableFuture<Response> sendAsync = CompletableFuture.supplyAsync(()->{
		dumpCurrentThreadName("supplyAsync");
		return launchTaskWithAuxThread(()->new BlockingJob().invoke());
	}, es)
	.thenCompose((CompletableFuture<Response> responseReceived)->{
		dumpCurrentThreadName("thenCompose");
		return responseReceived;
	})
	.thenApplyAsync((Response x)->{
		dumpCurrentThreadName("thenApply");
		return x;
	}, es);
 
	sendAsync.join();
}

上面的是我在Java8上實做的範例。如果以作者的範例來說,可以參考我在Java11做的範例程式碼。它使用了一個共享的變數responseReceived;當Blocking工作結束後,使用responseReceived.completeAsync去串接thenApply,這樣可以讓工作的執行回到共用的ExecutorService:

public static class BlockingJob2 {
	private CompletableFuture<Response> responseReceived;
	public BlockingJob2(CompletableFuture<Response> responseReceived) {
		this.responseReceived = responseReceived;
	}
	public void invoke() {
		try {
			dumpCurrentThreadName("before blocking job");
			Thread.sleep(2*1000);
			// io blocking job and get response
			responseReceived.completeAsync(()->new Response(), es);
		} catch( Exception e ) {
			// log
		} finally {
			dumpCurrentThreadName("after blocking job");
		}
	}
}
 
public static CompletableFuture<Response> send2(){
	return CompletableFuture.supplyAsync(()->{
		dumpCurrentThreadName("supplyAsync");
 
		CompletableFuture<Response> responseReceived = new CompletableFuture<>();
		launchTaskWithAuxThread(()->{
			new BlockingJob2(responseReceived).invoke();
		});
		return responseReceived;
	}, es)
	.thenCompose((CompletableFuture<Response> responseReceived)->{
		dumpCurrentThreadName("thenCompose");
		return responseReceived;
	})
	.thenApply((Response x)->{
		dumpCurrentThreadName("whenComplete: " + x.getClass().getName());
		return x;
	});
}

async寫法帶來的是thread的控制性;sync則是因為減少context switch而帶來的則是較好的效能。要選擇哪一種方式,一定要先確定好你要的是什麼。

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