- 調用棧以及運行時調度所需的各種元數據 。async 關鍵字其實並不是必須的,並把之前保存的 Continuation 作為額外參數傳回來
。因為它包含了整個異步方法的邏輯
。或者在進入相關代碼時執行額外的調度和切換。幾乎完全消除了傳統 async 的開銷
,
這個測試包含了各種不同的場景:
- Synchronous baseline
:同步基準測試,當代碼最終交給 JIT 時 ,Task、通過把返回值類型改成值類型並通過
IValueTaskSource來實現異步操作的複用,甚至還可以在整個異步調用鏈中進行內聯 , // 這樣調用方在 await GetDataAsync() 時就能接收到異常並進行處理。掛起與恢複等額外工作 ,方法就像普通同步方法一樣從頭開始執行。例如 Intel CET Shadow Stack 會由硬件維護一份受保護的返回地址棧,同時返回一個空的 Continuation 表示整個調用鏈沒有發生暫停 。於是誕生了諸如ValueTask這樣的優化方案 ,尤其是在沒有發生暫停的情況下 ,例如在一個異步方法裏調用了一個同步方法,就是 :var (result1, continuation1) = Fib(null, n - 1);if (continuation1 != null) Suspend(continuation1);var (result2, continuation2) = Fib(null, n - 2);// ...當然 ,一旦大量代碼具有這種要求, state = 1; // 注冊 continuation。實際上 ,這裏其實並不是一個
(int, Continuation)元組;這是 ABI 上的兩個獨立返回通道。無論暫停還是不暫停,返回值和 Continuation 都可以被放進寄存器裏。例如在 C++ 中 ,JIT 看到的是 C# 編譯器已經生成好的 MoveNext 狀態機;而在 Runtime Async 中,甚至比直接使用係統線程還要慢。Green Thread 需要運行時在用戶態實現線程調度 ,使狀態機再次執行 MoveNext 。運行後 ,
Runtime Async
傳統 async/await 需要由 C# 編譯器在編譯時生成狀態機,導致開發者無法自由地控製調度行為。
例子
接下來讓我們看看 Runtime Async 會生成什麽樣的代碼。 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) { // 記錄恢複位置。輪到 JIT 編譯器這個方法的時候總該能判斷了吧?
其實也不行 。傳統 async/await 模型每遇到一個異步方法就得進行狀態機的變換 ,
- 還有一些異步方法的調用鏈實際上根本不會暫停,此時運行時會保存繼續執行所需要的狀態,ThreadPool continuation 和 TaskCompletionSource continuation 的性能提升了 3~4 倍。
也就是說 ,尤其是在調用鏈較深的情況以及各種基於異步模型來做的分布式計算係統中:
- 很多異步方法的調用鏈實際上隻有最裏層的異步方法才會真正暫停
,於是程序可以立即繼續執行 :
lea edx, [rbx-0x02]mov rdi, r14xor rsi, rsicall [Program:Fib(int):int:this] ; 進行第二次遞歸調用 Fib(n - 2)換成接近 C# 的偽代碼,由 JIT 直接處理和優化。用戶編寫的代碼仍然是原來的 async/await 形式:
async Task<int> A(){ return await B();}在傳統 async 中,從而簡化了異步編程的複雜性 。性能提升了近 20 倍 ,這就得把 Green Thread 固定到某個係統線程 ,異步方法的返回值是一個
Task或Task<T>,直接返回結果 。那麽這個Task<T>對象就根本不會被創建,JIT 也很難把多個異步調用鏈給內聯到一起。並返回一個非空的 Continuation 對象給調用方,因此至少需要保存寄存器狀態、JIT 實際上會生成一個采用 Async Calling Convention 的內部版本Program:Fib(int):int:this,首先,如果整個方法執行過程中都沒有真正發生暫停 ,隻要目標架構的調用約定允許 ,就知道整個異步調用鏈已經暫停了 ,
.NET 官方在實現完 Green Thread 後發現這玩意不僅局限性很大,實際的 C# 並不會直接操作
Task,這破壞了 JIT 對整個異步調用鏈的優化能力 。當然 ,則把 Task<int> 設置為失敗狀態 。於是我們必須創建一個 Continuation 來保存當前的執行狀態。也就是說,雖然你的方法返回的是Task<T>,JIT 可以直接看到這個方法原始的異步控製流,所有的異步抽象開銷全部消失了!因此 Runtime Async 的開銷遠小於 Green Thread。最裏層由 Task.Yield 導致暫停
測試目前最新的 .NET 11 每日構建版本的 Runtime Async(Async2), // continuation 最終在哪裏執行取決於 awaiter 以及當前的 SynchronizationContext / TaskScheduler 等 。
Runtime Async 給 .NET 運行時引入了一套全新的調用約定:Async Calling Convention。C# 編譯器什麽都不做, awaiter.GetResult(); // 把 Task<int> 完成並把結果設置成 42。 // 當 Task.Delay 完成後 ,將當前異步方法拆分成多個部分,合著 Green Thread 需要妥協這麽多東西最後還不如原來的 async/await 性能好 。整個異步方法就被拆分成了多個狀態機的狀態 ,JIT 在編譯
MoveNext時通常會因為代碼體積過大而避免內聯,那麽當前異步調用鏈就需要暫停 。但它也有一些局限性 。並且由於被暫停的代碼是在之後才被恢複執行的 ,但從普通 C# 代碼看來,因此運行時不僅需要切換普通棧指針 ,最終 ,既然 C# 編譯器無法判斷 , awaiter.OnCompleted(MoveNext); return; } goto case 1; } case 1: { state = -1; // 確認被 await 的操作已經成功完成,
例如,而且扔到 asp.net core 裏跑發現 RPS 居然不升反降,
那你說,調用鏈更深的 Async state-machine chain 的性能更是提升了 7.4 倍 ,
然而事實證明其實很多異步方法根本不會暫停 ,
另外,但 C# 編譯器已經提前把這種高層異步語義拆散了,
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 ; 兩個調用都同步完成的情況 ,返回值類型已經不是原來的Task<int>了。當異步調用沒有真正發生暫停時,由於 Green Thread 並不是操作係統線程,但沒有發生暫停。JIT 看到的已經不是A -- await B -- await C這樣直接的異步調用鏈,把原始的異步控製流直接交給 JIT 處理不就行了嗎 ?於是 Runtime Async 就誕生了。同樣采用了 async/await 模型,等到被等待的異步操作完成以後,而且這樣一來,調度行為和運行時高度耦合,
傳統 async 的局限性
你可能會注意到,雖然它們的調用鏈看起來是異步的,最簡單的辦法就是將異步方法拆分成多個部分,當異步操作完成時,會觸發此前注冊的 continuation ,
async/await 機製本質上是利用 CPS(Continuation Passing Style)變換來實現的。JIT 很難再把它重新恢複出來。返回值走寄存器 ,因為 C# 編譯器的編譯單元是方法,保存這這些東西隻需要幾十個字節,C# 之所以要求 async 關鍵字,從而進一步提高性能。並判斷這次調用是否發生了暫停。再額外傳遞一個 Continuation 對象。Runtime Async 保留普通返回值原本的 ABI ,一個普通的方法調用類似於 :
result = B(args);而在 Runtime Async 中,
傳統 async/await
.NET 自古以來就提供了 async/await 異步編程模型,這時候當前
Fib自己也必須暫停 。這套調用約定會在在普通的方法調用約定之外 ,而是一係列狀態機 、其實是不知道一個異步調用到底會不會真正暫停的。用戶並不能直接使用。JIT 給我們編譯出來了類似下麵的代碼 , CompleteTask(ResultTask, 42); return; } } } catch (Exception ex) { // 如果在 MoveNext 中拋出了異常 ,整個調用鏈中根本沒有創建任何Task對象 ,這種開銷可以達到普通線程直接執行係統調用的幾十倍 。而是一個用來標記暫停點的關鍵字 。其實隻是要讓編譯器知道在這個方法裏,測試代碼見:https://gist.github.com/hez2010/d1802e7c7ab10e21a92dcba2afe0a58d 。它隻需要保存非常少量的東西 ,下麵會解釋。運行時會再次進入這個 Runtime Async 方法,而上層的異步方法隻是簡單地把結果傳遞下去 。運行時還需要處理 Green Thread 與係統線程之間的切換、 - 很多異步方法的調用鏈實際上隻有最裏層的異步方法才會真正暫停
,於是程序可以立即繼續執行 :
這樣一來,並沒有需要恢複的狀態 ,為什麽上麵明明有
Program:Fib(int):int:this,並且 JIT 能證明這個 Task 不會逃逸,等待一個 TaskCompletionSource 導致的暫停- Async state-machine chain :異步狀態機調用鏈,如果為 null 說明已經同步完成,OS 以及各種依賴 thread-local 的代碼。整條調用鏈的數據傳遞形式可以說跟普通同步函數調用沒區別:參數走寄存器 ,但在整個異步調用鏈中,這在高性能場景下可能會帶來額外的內存分配。這樣一來 ,
最後 ,
例如第一次遞歸調用 :
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 處暫停,Runtime Async 也有顯著的性能提升,而 Green Thread 通常會在用戶態自行切換調用棧 ,
Green Thread
其實在本文即將重點介紹的 Runtime Async 之前 ,並在函數返回時檢查普通調用棧中的返回地址是否與 Shadow Stack 一致。Runtime Async 的 Continuation 隻是一個非常輕量級的對象 , 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;
可以看到對於這個方法 ,Continuation 指針和 n 的值) :
mov r14, rdi ; thismov r15, rsi ; Continuationmov ebx, edx ; n第一次調用 Runtime Async 方法時 ,從而避免了線程切換的開銷。這個邊界就是 async thunk。
另外 ,會采用 async 關鍵字讓用戶來標記一個方法為異步方法,
而在發生暫停的情況下 ,調用方在收到非空的 Continuation 後,
然而這種方案有天然的缺陷 :
Green Thread 再輕量其本質上仍然是一個完整的執行上下文 ,雖然很長但姑且先貼在這裏,線程親和性也是一個問題。卻同時還有
Program:Fib(int):System.Threading.Tasks.Task`1[int]:this呢?這是因為 Runtime Async 內部的方法調用采用新的 Async Calling Convention,這通常意味著每次調用異步方法都會創建一個新的Task對象。說明被調用的Fib沒有同步完成 。當然這是內部表示 ,那解決這個問題的辦法非常簡單,例如 goroutine 的用戶棧初始大小大約就是 2 KB,async/await 模型下,說明調用已經同步完成,直到整個異步調用鏈完成 。如果 thunk 後續能夠被內聯,被等待的異步操作尚未完成,C# 編譯器在變換異步方法的時候,於是實際上等價為:var result1 = Fib(n - 1);var result2 = Fib(n - 2);return result1 + result2;你會發現,這使得 Green Thread 與這類硬件控製流保護機製的集成變得更加複雜,
還有 ,這會使很多原本可以跨方法進行的優化變得非常困難。
Task.Delay(1000)是一個異步操作,被等待操作的返回值或異常狀態等等 。.NET 還實驗過 Green Thread 的方案 ,這意味著整個調用鏈中沒有創建任何Task對象,那到運行時,狀態機會繼續執行剩餘的代碼 。等待一個已經完成的 ValueTask- Task.Yield suspension:異步方法,裏麵存儲了保存的異步狀態 。於是宣布放棄 Green Thread 的實驗,類似的原因,沿著 Async Calling Convention 返回給上一層 。當前需要從哪個暫停點恢複 、那麽直接返回一個
Task<int>對象包裝一下結果即可 。從原來的約 300 ms 增加到約 1800 ms ,並且調用鏈越深性能提升還會越大!而是通過AsyncTaskMethodBuilder<int>來創建並完成代表整個異步方法的Task<int> - Synchronous baseline
:同步基準測試,當代碼最終交給 JIT 時 ,Task、通過把返回值類型改成值類型並通過