例如第一次遞歸調用 :
await Fib(n - 1)被編譯成 :
lea edx, [rbx-0x01] ; n - 1mov rdi, r14 ; thisxor rsi, rsi ; Continuation = nullcall [Program:Fib(int):int:this]而 Fib(n - 1)實際上返回了兩個值
:
eax = Fib 的 int 返回值rcx = Continuation當然 ,
以下是一個簡單的示例:
public async Task<int> GetDataAsync(){ // 模擬異步操作 await Task.Delay(1000); return 42;}上麵這個例子中 ,並返回一個非空的 Continuation 對象給調用方,因此運行時需要在兩種調用約定之間放置一個邊界, mov rdi, rcx mov rsi, 0x... ; Continuation call [CORINFO_HELP_ALLOC_CONTINUATION] mov r12, rax mov dword ptr [r12+0x48], ebx ; 保存 n 的值 ; ... 保存其他需要保存的狀態 ... mov rcx, r12 ; return Continuation retSUSPEND_SECOND: ; Fib(n - 2) 暫停了,等待一個 Task.Yield 導致的暫停
例子
接下來讓我們看看 Runtime Async 會生成什麽樣的代碼。Green Thread 和硬件安全機製也有衝突。Runtime Async 都能以最小的開銷執行。直接調用普通方法
async/await 機製本質上是利用 CPS(Continuation Passing Style)變換來實現的
。執行速度跟同步方法的基線幾乎沒有差別 。沿著 Async Calling Convention 返回給上一層。所以正常執行路徑最終隻是不斷遞歸調用,Task.Delay(1000)是一個異步操作
,await 不是一個普通的識別符,尤其是在整個異步調用鏈實際上都沒有發生暫停的情況下,並把之前保存的 Continuation 作為額外參數傳回來。而上層的異步方法隻是簡單地把結果傳遞下去 。但有這 2KB 都夠創建幾百個 async 狀態機了
。awaiter 和 method builder 來驅動執行
。並不保證恢複執行時仍然運行在原來的係統線程上。下麵會解釋。從而減少內存分配 。因此 Runtime Async 的開銷遠小於 Green Thread。
Runtime Async 給 .NET 運行時引入了一套全新的調用約定 :Async Calling Convention 。就存在進一步通過逃逸分析消除這次分配。
而在發生暫停的情況下,等待一個已經完成的 Task
最後 ,
運行後 ,從而避免了線程切換的開銷。並通過 MoveNext、直到 Task.Delay完成 ,那 JIT 就算看穿了整個異步調用鏈 ,同時返回一個空的 Continuation 表示整個調用鏈沒有發生暫停。被標記的方法則會作為 CPS 變換的入口點 。等價的 C# 偽代碼類似於:
var (result1, continuation1) = Fib(null, n - 1);if (continuation1 != null) Suspend(continuation1);var (result2, continuation2) = Fib(null, n - 2);if (continuation2 != null) Suspend(continuation2);return result1 + result2;而實際上,
這一套機製也真正實現了 pay for play:不暫停就不為異步抽象付費 ,因此如果代碼真正暫停了, add ebx, r12d mov eax, ebx ; return value xor ecx, ecx ; null Continuation retSUSPEND_FIRST: ; Fib(n - 1) 暫停了,並不需要為每一層 async 調用創建額外的結果包裝對象,例如在一個異步方法裏調用了一個同步方法 ,C# 編譯器會把異步方法改寫成狀態機 ,
Green Thread
其實在本文即將重點介紹的 Runtime Async 之前 ,於是我們必須創建一個 Continuation 來保存當前的執行狀態。把原始的異步控製流直接交給 JIT 處理不就行了嗎 ?於是 Runtime Async 就誕生了。而是通過 AsyncTaskMethodBuilder<int>來創建並完成代表整個異步方法的 Task<int>
。而是一係列狀態機、而且這樣一來 ,async/await 模型下,但在整個異步調用鏈中,並判斷這次調用是否發生了暫停。說明發生了暫停
就可以同時獲得異步方法的返回結果,例如:
public async Task<int> GetDataAsync(){ return await GetValueAsync();}public async Task<int> GetValueAsync(){ return 42;}C# 編譯器會為兩個方法都生成狀態機和 Task<int>
,因為 C# 編譯器的編譯單元是方法,被等待操作的返回值或異常狀態等等。從而進一步提高性能。返回值類型已經不是原來的 Task<int>了
。幾乎完全消除了傳統 async 的開銷 , awaiter.GetResult(); // 把 Task<int> 完成並把結果設置成 42。這意味著整個調用鏈中沒有創建任何 Task對象
,返回值和 Continuation 都可以被放進寄存器裏。例如部分 GUI
、OS 以及各種依賴 thread-local 的代碼 。它隻需要保存非常少量的東西,於是 GetDataAsync方法實際上就會被編譯成:
public Task<int> GetDataAsync(){ var stateMachine = new StateMachine(); stateMachine.MoveNext(); return stateMachine.ResultTask;}上麵的 CreateIncompleteTask和 CompleteTask隻是為了說明原理而使用的偽代碼。那麽 Green Thread 的調度開銷就會變得非常大 ,當代碼最終交給 JIT 時,由於 JIT 能夠直接看到完整的異步調用控製流,
而 await 關鍵字的作用是告訴編譯器這裏有暫停點,例如在 C++ 中,調用鏈更深的 Async state-machine chain 的性能更是提升了 7.4 倍,JIT 實際上會生成一個采用 Async Calling Convention 的內部版本 Program:Fib(int):int:this,C# 編譯器什麽都不做 ,其實是不知道一個異步調用到底會不會真正暫停的。很多異步方法可能根本不會暫停,卻同時還有 Program:Fib(int):System.Threading.Tasks.Task`1[int]:this呢 ?這是因為 Runtime Async 內部的方法調用采用新的 Async Calling Convention
,因為它包含了整個異步方法的邏輯
。.NET 還實驗過 Green Thread 的方案
,async 關鍵字其實並不是必須的 ,雖然很長但姑且先貼在這裏
,調用方在收到非空的 Continuation 後 ,而 Green Thread 通常會在用戶態自行切換調用棧,如果為 null 說明已經同步完成,隻要目標架構的調用約定允許
,一旦大量代碼具有這種要求 ,因此也確實需要一個 Task對象來存儲結果
。從原來的約 300 ms 增加到約 1800 ms,調用棧以及運行時調度所需的各種元數據。
第一次遞歸調用之後:
call [Program:Fib(int):int:this]mov r12d, eaxtest rcx, rcxjne SHORT SUSPEND如果 rcx != null
,雖然它們的調用鏈看起來是異步的 ,於是程序可以立即繼續執行:
lea edx, [rbx-0x02]mov rdi, r14xor rsi, rsicall [Program:Fib(int):int:this] ; 進行第二次遞歸調用 Fib(n - 2)換成接近 C# 的偽代碼,甚至需要操作係統提供專門的支持 。Runtime Async 直接把內存分配和 GC 全都降到了 0,整個異步方法就被拆分成了多個狀態機的狀態 ,
傳統 async 的局限性
你可能會注意到,
然而事實證明其實很多異步方法根本不會暫停 ,這個 Task<int> 會在當前異步方法完成時被設置為完成狀態 。例如 Intel CET Shadow Stack 會由硬件維護一份受保護的返回地址棧,既然 C# 編譯器無法判斷, awaiter.OnCompleted(MoveNext); return; } goto case 1; } case 1: { state = -1; // 確認被 await 的操作已經成功完成 ,因為這個 Fibonacci 示例中的所有調用都會同步完成,
然而這種方案有天然的缺陷 :
Green Thread 再輕量其本質上仍然是一個完整的執行上下文,
首先 async/await 模型下
,性能提升了近 20 倍,於是誕生了諸如 ValueTask這樣的優化方案
,
也就是說,
另外,於是宣布放棄 Green Thread 的實驗,
首先,實際的 C# 並不會直接操作 Task,並且 JIT 能證明這個 Task 不會逃逸,但 C++ 並不要求 async 關鍵字。Runtime Async 的 Continuation 隻是一個非常輕量級的對象,它負責把 Runtime Async 內部的普通返回值 + Continuation 轉換成外部調用方所期待的 Task<int>
。例如 goroutine 的用戶棧初始大小大約就是 2 KB,直接原地慢了 5 倍以上 。這套機製允許開發者以同步方式編寫異步代碼,用戶編寫的代碼仍然是原來的 async/await 形式:
async Task<int> A(){ return await B();}在傳統 async 中 ,直到整個異步調用鏈完成。再額外傳遞一個 Continuation 對象。用戶並不能直接使用。 // 這樣調用方在 await GetDataAsync() 時就能接收到異常並進行處理。導致開發者無法自由地控製調度行為。傳入的 Continuation 為 null ,在用戶態實現輕量級線程, mov rdi, rcx mov rsi, 0x... ; Continuation type call [CORINFO_HELP_ALLOC_CONTINUATION] mov r15, rax mov dword ptr [r15+0x4C], r12d ; 保存 Fib(n - 1) 的結果 ; ... 保存其他需要保存的狀態 ... mov rcx, r15 ; return Continuation ret; --------------------------------------------Program:Fib(int):Task<int>:this mov rdi, rbx ; this mov edx, r15d ; n xor rsi, rsi ; null Continuation call [Program:Fib(int):int:this] ; 調用真正的 Runtime Async 方法 mov ebx, eax ; result test rcx, rcx ; Continuation == null? jne THUNK_SUSPENDED ; return Task.FromResult(ebx) mov rax, <Task<int>> retTHUNK_SUSPENDED: ; var task = new RuntimeAsyncTask<int>(); ; 把 continuation 連接到 task; ; return task;
可以看到對於這個方法,當然 ,這會使很多原本可以跨方法進行的優化變得非常困難。調度行為和運行時高度耦合 ,而真正暫停時也隻需要為實際使用的狀態付費 。也就是當前方法需要等待一個異步操作完成 ,這使得其可以在整個異步調用鏈中進行跨方法的優化,C# 編譯器在變換異步方法的時候,
但如果執行到某個 await 時 ,每個部分在 await 處暫停,保存這這些東西隻需要幾十個字節,
class StateMachine{ private int state = 0; // 創建一個用來存儲結果的 Task<int>,但現實中存在大量依賴特定係統線程的 API,也無法做任何優化,如果 thunk 後續能夠被內聯
,無法在編譯 GetDataAsync的時候看到 GetValueAsync的具體實現。於是我們必須創建一個 Continuation 來保存當前的執行狀態。並將 Runtime Async 方法按照一種特殊的 async calling convention 編譯。而不需要先包裝到某個對象中再返回
。傳統 async/await 模型每遇到一個異步方法就得進行狀態機的變換,整個調用鏈中根本沒有創建任何 Task對象
,並在被 await 的異步操作完成後繼續執行剩餘的代碼。這個方法通過寄存器傳遞參數(this 指針、最裏層由 Task.Yield 導致暫停測試目前最新的 .NET 11 每日構建版本的 Runtime Async(Async2),其實隻是要讓編譯器知道在這個方法裏 ,
傳統 async/await
.NET 自古以來就提供了 async/await 異步編程模型,因此哪怕 JIT 想要做一些跨方法的優化也很難做到。如果失敗則會在這裏拋出異常。它不再讓 C# 編譯器提前把 async 方法展開成狀態機,Fib 的簽名仍然是 Task<int> Fib(int)
。返回值走寄存器,於是實際上等價為 :
var result1 = Fib(n - 1);var result2 = Fib(n - 2);return result1 + result2;你會發現 ,並且調用鏈越深性能提升還會越大!在所有測試中 ,那麽它就會直接返回正常的結果,
等到被等待的異步操作完成以後 ,C# 之所以要求 async 關鍵字,
再有,awaiter 和 continuation 之間的交互。這在高性能場景下可能會帶來額外的內存分配。類似於 goroutine 和 Java Virtual Thread ,這樣的調用鏈實際上是同步的。
而這個 thunk 中其實也有前麵說過的類似代碼:
xor rsi, rsicall [Program:Fib(int):int:this]mov ebx, eaxtest rcx, rcx ; Continuation 是否為 null也就是先調用真正的 Runtime Async 方法後 ,
JIT 才會在這一刻真正創建保存當前執行狀態所需要的 Continuation:
mov rdi, rcxmov rsi, 0x... ; Continuation typecall [CORINFO_HELP_ALLOC_CONTINUATION]mov r12, rax隨後把恢複執行時仍然需要的局部狀態保存進去:
mov dword ptr [r12+0x48], ebx最後:
mov rcx, r12ret把剛剛創建好的 Continuation放進 rcx,await關鍵字會暫停 GetDataAsync方法的執行,如果沒有真正發生暫停
,因此在涉及係統調用時,這時候當前 Fib自己也必須暫停
。因此運行時不僅需要切換普通棧指針 ,而是把異步控製流保留到運行時,Continuation非空的情況也能直接從生成代碼中看到。轉而開發 Runtime Async。一個普通的方法調用類似於 :
result = B(args);而在 Runtime Async 中,每個狀態對應著 await 關鍵字的邊界。例如跨越暫停點後仍然存活的局部變量
、而這個同步方法又調用了另一個異步方法,當異步調用沒有真正發生暫停時,雖然你的方法返回的是 Task<T>,整個調用鏈就像普通的同步函數調用一樣執行 。無論暫停還是不暫停
, CompleteTask(ResultTask, 42); return; } } } catch (Exception ex) { // 如果在 MoveNext 中拋出了異常 ,掛起與恢複等額外工作, // continuation 最終在哪裏執行取決於 awaiter 以及當前的 SynchronizationContext / TaskScheduler 等
。async/await 模型下,正常返回值和額外的 Continuation 都屬於調用約定的一部分,從語義上看這些調用完全可以像普通的同步函數調用一樣執行,使狀態機再次執行 MoveNext。 public Task<int> ResultTask { get; } = CreateIncompleteTask<int>(); private TaskAwaiter awaiter; public void MoveNext() { try { switch (state) { case 0: { awaiter = Task.Delay(1000).GetAwaiter(); if (!awaiter.IsCompleted) { // 記錄恢複位置
。因此至少需要保存寄存器狀態 、當異步操作完成時,也就是說
,就知道整個異步調用鏈已經暫停了 ,Green Thread 需要運行時在用戶態實現線程調度
,等待一個已經完成的 ValueTask
例如
,異步方法的返回值是一個 Task或 Task<T>