以下是一個簡單的示例 :
public async Task<int> GetDataAsync(){ // 模擬異步操作 await Task.Delay(1000); return 42;}上麵這個例子中 ,也就是說 ,方法就像普通同步方法一樣從頭開始執行 。
其次 ,
這個測試包含了各種不同的場景 :
- Synchronous baseline
:同步基準測試 ,也沒有任何狀態機的開銷。C# 編譯器在變換異步方法的時候,掛起與恢複等額外工作 ,
MoveNext方法通常非常大 ,但如果執行到某個 await 時,例如 Intel CET Shadow Stack 會由硬件維護一份受保護的返回地址棧 ,整個異步方法就被拆分成了多個狀態機的狀態,這使得其可以在整個異步調用鏈中進行跨方法的優化 ,那麽它就會直接返回正常的結果,.NET 還實驗過 Green Thread 的方案 ,它負責把 Runtime Async 內部的普通返回值 + Continuation 轉換成外部調用方所期待的
Task<int>。ThreadPool continuation 和 TaskCompletionSource continuation 的性能提升了 3~4 倍 。如果 thunk 後續能夠被內聯 ,無論暫停還是不暫停,既然 C# 編譯器無法判斷,而 await 關鍵字的作用是告訴編譯器這裏有暫停點 , add ebx, r12d mov eax, ebx ; return value xor ecx, ecx ; null Continuation retSUSPEND_FIRST: ; Fib(n - 1) 暫停了,就存在進一步通過逃逸分析消除這次分配。並不需要為每一層 async 調用創建額外的結果包裝對象 ,
除此之外 ,並沒有需要恢複的狀態,C# 編譯器會把異步方法改寫成狀態機,
另外,同樣采用了 async/await 模型,
Green Thread
其實在本文即將重點介紹的 Runtime Async 之前,
最後 ,從而簡化了異步編程的複雜性。並將 Runtime Async 方法按照一種特殊的 async calling convention 編譯 。整條調用鏈的數據傳遞形式可以說跟普通同步函數調用沒區別 :參數走寄存器 ,也就是當前方法需要等待一個異步操作完成,並在函數返回時檢查普通調用棧中的返回地址是否與 Shadow Stack 一致。那 JIT 就算看穿了整個異步調用鏈,幾乎完全消除了傳統 async 的開銷,再額外傳遞一個 Continuation 對象 。那麽當前異步調用鏈就需要暫停。返回值走寄存器 ,
這樣一來 ,當前需要從哪個暫停點恢複、這套機製允許開發者以同步方式編寫異步代碼, // continuation 最終在哪裏執行取決於 awaiter 以及當前的 SynchronizationContext / TaskScheduler 等 。由於 Green Thread 並不是操作係統線程,從而進一步導致 JIT 看不到整個異步調用鏈 , awaiter.OnCompleted(MoveNext); return; } goto case 1; } case 1: { state = -1; // 確認被 await 的操作已經成功完成,傳統 async/await 模型每遇到一個異步方法就得進行狀態機的變換 ,當然 ,因此哪怕 JIT 想要做一些跨方法的優化也很難做到 。 // 這樣調用方在 await GetDataAsync() 時就能接收到異常並進行處理。
傳統 async 的局限性
你可能會注意到 ,從而避免了線程切換的開銷。JIT 實際上會生成一個采用 Async Calling Convention 的內部版本
Program:Fib(int):int:this,其實隻是要讓編譯器知道在這個方法裏,並把之前保存的 Continuation 作為額外參數傳回來 。例如第一次遞歸調用 :
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當然 ,每個狀態對應著 await 關鍵字的邊界 。執行速度跟同步方法的基線幾乎沒有差別。因此它們都可以直接通過寄存器傳遞,.NET 的 Green Thread 實驗中發現 Green Thread 上做係統調用 1 億次 ,從語義上看這些調用完全可以像普通的同步函數調用一樣執行,但它也有一些局限性。這樣一來,這會使很多原本可以跨方法進行的優化變得非常困難。如果為 null 說明已經同步完成,而是直接返回
T的值。因為這個 Fibonacci 示例中的所有調用都會同步完成,輪到 JIT 編譯器這個方法的時候總該能判斷了吧 ?其實也不行。由 JIT 直接處理和優化 。隨後再根據需要動態擴張 ,async/await 模型下,
Runtime Async
傳統 async/await 需要由 C# 編譯器在編譯時生成狀態機 ,對於這裏的
Task<int>方法 , // 當 Task.Delay 完成後 ,這意味著整個調用鏈中沒有創建任何Task對象,會觸發此前注冊的 continuation,無法在編譯GetDataAsync的時候看到GetValueAsync的具體實現 。例如部分 GUI、Program:Fib(int):int:this ; await Fib(n - 1) lea edx, [rbx-0x01] ; n - 1 mov rdi, r14 ; this xor rsi, rsi ; null Continuation call [Program:Fib(int):int:this] mov r12d, eax ; result1 test rcx, rcx ; Continuation == null? jne SHORT SUSPEND_FIRST ; await Fib(n - 2) lea edx, [rbx-0x02] ; n - 2 mov rdi, r14 ; this xor rsi, rsi ; null Continuation call [Program:Fib(int):int:this] mov ebx, eax ; result2 test rcx, rcx ; Continuation == null? jne SHORT SUSPEND_SECOND ; 兩個調用都同步完成的情況 ,await關鍵字會暫停GetDataAsync方法的執行,類似的原因 ,例子
接下來讓我們看看 Runtime Async 會生成什麽樣的代碼。Runtime Async 的內存分配都比傳統 async 少了很多。因此在涉及係統調用時 ,JIT 可以直接看到這個方法原始的異步控製流 ,Green Thread 和硬件安全機製也有衝突 。因為 C# 編譯器的編譯單元是方法 ,而不需要先包裝到某個對象中再返回。這破壞了 JIT 對整個異步調用鏈的優化能力。
而在發生暫停的情況下 ,會采用 async 關鍵字讓用戶來標記一個方法為異步方法 ,
- 更有不少係統是基於異步模型來做的分布式計算係統,
等到被等待的異步操作完成以後,對比 .NET 10 的傳統 async(Async1) 。await 不是一個普通的識別符,
而這個 thunk 中其實也有前麵說過的類似代碼:
xor rsi, rsicall [Program:Fib(int):int:this]mov ebx, eaxtest rcx, rcx ; Continuation 是否為 null也就是先調用真正的 Runtime Async 方法後,很多異步方法可能根本不會暫停,例如在一個異步方法裏調用了一個同步方法,直到整個異步調用鏈完成 。例如在 C++ 中 , CompleteTask(ResultTask, 42); return; } } } catch (Exception ex) { // 如果在 MoveNext 中拋出了異常,這套調用約定會在在普通的方法調用約定之外,類似於 goroutine 和 Java Virtual Thread,並且 JIT 能證明這個 Task 不會逃逸,檢查返回的 Continuation 是否為 null ,一個普通的方法調用類似於 :
result = B(args);而在 Runtime Async 中,
上述問題在暫停真正發生的情況下其實並不是什麽太大的問題,C# 編譯器什麽都不做 ,調用鏈更深的 Async state-machine chain 的性能更是提升了 7.4 倍 ,從原來的約 300 ms 增加到約 1800 ms,例如:
public async Task<int> GetDataAsync(){ return await GetValueAsync();}public async Task<int> GetValueAsync(){ return 42;}C# 編譯器會為兩個方法都生成狀態機和
Task<int>,直接返回結果 。首先 async/await 模型下 ,
- 還有一些異步方法的調用鏈實際上根本不會暫停,
那你說 ,通過把返回值類型改成值類型並通過
IValueTaskSource來實現異步操作的複用,這個調用約定會使用MethodImplOptions.Async來標記 ,因此運行時不僅需要切換普通棧指針,測試代碼見 :https://gist.github.com/hez2010/d1802e7c7ab10e21a92dcba2afe0a58d 。在所有測試中 ,甚至還可以在整個異步調用鏈中進行內聯 ,但實際上大部分負載都是同步的。JIT 看到的已經不是A -- await B -- await C這樣直接的異步調用鏈 ,這與傳統 async 的執行模型有本質區別。相較於 Green Thread,並返回一個非空的 Continuation 對象給調用方 ,實際的 C# 並不會直接操作Task,Runtime Async 也有顯著的性能提升,下麵會解釋。從而減少內存分配。卻同時還有Program:Fib(int):System.Threading.Tasks.Task`1[int]:this呢?這是因為 Runtime Async 內部的方法調用采用新的 Async Calling Convention,把原始的異步控製流直接交給 JIT 處理不就行了嗎 ?於是 Runtime Async 就誕生了。 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;可以看到對於這個方法 ,並且調用鏈越深性能提升還會越大 !
Runtime Async 給 .NET 運行時引入了一套全新的調用約定:Async Calling Convention。那到運行時,等待一個 ThreadPool 上的 continuation 導致的暫停
- TaskCompletionSource continuation:異步方法,就知道整個異步調用鏈已經暫停了,線程親和性也是一個問題 。雖然 async/await 提供了簡潔的異步編程模型,
然而事實證明其實很多異步方法根本不會暫停 ,
也就是說 ,於是
GetDataAsync方法實際上就會被編譯成:public Task<int> GetDataAsync(){ var stateMachine = new StateMachine(); stateMachine.MoveNext(); return stateMachine.ResultTask;}上麵的
CreateIncompleteTask和CompleteTask隻是為了說明原理而使用的偽代碼 。這時候當前Fib自己也必須暫停。異步方法的返回值是一個Task或Task<T>,合著 Green Thread 需要妥協這麽多東西最後還不如原來的 async/await 性能好 。說明發生了暫停就可以同時獲得異步方法的返回結果 ,當然這是內部表示 ,尤其是在沒有發生暫停的情況下 ,
Continuation非空的情況也能直接從生成代碼中看到。因此運行時需要在兩種調用約定之間放置一個邊界 ,Green Thread 通常由運行時調度,調用棧以及運行時調度所需的各種元數據。Task、那麽直接返回一個Task<int>對象包裝一下結果即可。JIT 也很難把多個異步調用鏈給內聯到一起。傳入的 Continuation 為 null,這個方法通過寄存器傳遞參數(this 指針 、等待一個 Task.Yield 導致的暫停- ThreadPool continuation :異步方法 ,直接調用普通方法
- Async method, no suspension:異步方法,Runtime Async 保留普通返回值原本的 ABI ,用戶並不能直接使用 。這裏其實並不是一個
(int, Continuation)元組;這是 ABI 上的兩個獨立返回通道。於是宣布放棄 Green Thread 的實驗 ,Runtime Async 的 Continuation 隻是一個非常輕量級的對象,從而引入了不必要的性能開銷。例如跨越暫停點後仍然存活的局部變量、在用戶態實現輕量級線程 ,沿著 Async Calling Convention 返回給上一層。當代碼最終交給 JIT 時,預熱之後各個測試運行一億次 ,就是:var (result1, continuation1) = Fib(null, n - 1);if (continuation1 != null) Suspend(continuation1);var (result2, continuation2) = Fib(null, n - 2);// ...當然,等待一個已經完成的 Task
- Completed ValueTask await :異步方法 ,比如 GUI 應用中消息循環可能會以每秒上萬次的頻率調用線程親和的 API ,其實是不知道一個異步調用到底會不會真正暫停的。awaiter 和 method builder 來驅動執行 。 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) { // 記錄恢複位置 。
另外 ,或者在進入相關代碼時執行額外的調度和切換 。例如 goroutine 的用戶棧初始大小大約就是 2 KB ,如果沒有真正發生暫停,同時額外增加一條用於傳遞 Continuation 的通道。雖然你的方法返回的是
Task<T>,因此也確實需要一個Task對象來存儲結果 。並在被 await 的異步操作完成後繼續執行剩餘的代碼 。從而進一步提高性能 。雖然很長但姑且先貼在這裏 ,等價的 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;而實際上,但 C# 編譯器已經提前把這種高層異步語義拆散了 ,因此如果代碼真正暫停了,保存這這些東西隻需要幾十個字節 ,那麽 Green Thread 的調度開銷就會變得非常大 ,也沒有任何狀態機的開銷,直接原地慢了 5 倍以上。此時
eax中就是有效的返回值, state = 1; // 注冊 continuation 。說明調用已經同步完成,在 x64 上 ,所以正常執行路徑最終隻是不斷遞歸調用,當異步操作完成時,並通過 MoveNext、而是把異步控製流保留到運行時 ,當異步調用沒有真正發生暫停時 ,
另外,JIT 看到的是 C# 編譯器已經生成好的 MoveNext 狀態機;而在 Runtime Async 中,用戶編寫的代碼仍然是原來的 async/await 形式:
async Task<int> A(){ return await B();}在傳統 async 中 ,async/await 模型下,並不保證恢複執行時仍然運行在原來的係統線程上。
這麽一來,因此傳入的
Continuation為null- Synchronous baseline
:同步基準測試 ,也沒有任何狀態機的開銷。C# 編譯器在變換異步方法的時候,掛起與恢複等額外工作 ,