這裏我選擇在類型層麵構建一條字符鏈表,型系Select、统上我們就可以基於某個 IStringNode
,实现
這個想法最終促成了 TypedSql —— 一個用 C# 類型係統實現的查询內存內 SQL 查詢引擎。也可以把它輸出到代碼裏然後通過 NativeAOT 編譯成原生二進製文件
,引擎就能讓 JIT 幫你完成大部分的型系工作
。所以在一些受限環境(比如 AOT)下可能無法使用 ,统上LessThanFilter、实现達到了性能和易用性的查询平衡
。
每一列會實現這樣一個接口:
internal interface IColumn<TRow,引擎 TValue>{ static abstract string Identifier { get; } static abstract TValue Get(in TRow row);}舉個簡單的例子:
internal readonly struct PersonNameColumn : IColumn<Person, string>{ public static string Identifier => "Name"; public static string Get(in Person row) => row.Name;}而投影(SELECT後麵那部分)則實現:
internal interface IProjection<TRow, TResult>{ static abstract TResult Project(in TRow row);}將選出某一列本身做成一個投影,'a' 、型系而不是统上為 string泛型實例化一個具體類型,隻要利用好 C# 的实现泛型和靜態成員,設計了一個很小的查询 SQL 方言:
支持這些語句:
SELECT * FROM $SELECT col FROM $SELECT col1, col2, ... FROM $WHERE支持:- 比較:
=,!=,>,<,>=,<= - 布爾:
AND,OR,NOT - 括號
- 比較:
- 字麵量支持:
- 整數(如
42) - 浮點數(如
123.45) - 布爾(
true/false) - 單引號字符串(
'Seattle',全是引擎靜態方法 。當成查詢計劃會怎樣 ?也就是說 ,隻是簡單地訪問
TLiteral.Value,而是針對單表 、 - 在兩個兼容形狀的
ValueTuple之間搬運字段; - 識別並處理
string↔ValueString的轉換; - 如果
ValueTuple有Rest(嵌套元組),一個整型字麵量長這樣 :internal readonly struct Int<H7, H6, H5, H4, H3, H2, H1, H0> : ILiteral<int> where H7 : IHex // ... where H0 : IHex{ public static int Value => (H7.Value << 28) | (H6.Value << 24) | (H5.Value << 20) | (H4.Value << 16) | (H3.Value << 12) | (H2.Value << 8) | (H1.Value << 4) | H0.Value;}浮點數也是一樣的 8 個十六進製數位 ,也可以返回元組 :
var seniorTitles = QueryEngine.Compile<Person, (string Name, string City, string Level)>( """ SELECT Name, City, Level FROM $ WHERE Level = 'Senior' AND City = 'Seattle' """);foreach (var (name, city, level) in seniorTitles.Execute(allPeople.AsSpan())){ Console.WriteLine($"{ name} in { city} [{ level}]");}所有重活——解析 SQL、 // 遇到 Rest 字段時遞歸 。而你甚至不需要實現任何的代碼生成後端 ,而外麵看到的則是
(string, int, string, …),這一層委托調用可以說幾乎沒有任何開銷。
ValueTupleConvertHelper:用動態 IL 在元組之間搬運字段ValueTupleConvertHelper<TPublicResult, TRuntimeResult>的職責是:TypedSql 編譯出來的類型大概是這樣 :
QueryProgram< Person, WhereSelect< Person, EqualsFilter< Person, ValueStringColumn<PersonCityColumn, Person>, 'Seattle', ValueString >, ColumnProjection<PersonIdColumn, Person, Int32>, Stop<Int32, Person>, Int32, Int32, Person>,Int32,Int32>讓我們來看看 RyuJIT 為我們的查詢方案生成了什麽樣的機器碼 :
G_M000_IG01: ; prologue push r15 push r14 push rdi push rsi push rbp push rbx sub rsp, 40 mov rbx, rcxG_M000_IG02: ; 分配結果數組 mov esi, dword ptr [rbx+0x08] mov edx, esi mov rcx, 0x7FFE71F29558 call CORINFO_HELP_NEWARR_1_VC mov rdi, rax xor ebp, ebp mov rbx, bword ptr [rbx] test esi, esi jle SHORT G_M000_IG06G_M000_IG03: ; 初始化循環變量 xor r14d, r14dG_M000_IG04: ; 循環體 lea r15, bword ptr [rbx+r14] mov rcx, gword ptr [r15+0x08] mov rdx, 0x16EB0400D30 mov rdx, gword ptr [rdx] mov rdx, gword ptr [rdx+0x08] cmp rcx, rdx je G_M000_IG12 test rcx, rcx je SHORT G_M000_IG05 test rdx, rdx je SHORT G_M000_IG05 mov r8d, dword ptr [rcx+0x08] cmp r8d, dword ptr [rdx+0x08] je SHORT G_M000_IG08G_M000_IG05: ; 更新循環計數器 add r14, 72 dec esi jne SHORT G_M000_IG04G_M000_IG06: ; 產生結果對象 mov rcx, 0x7FFE72227600 call CORINFO_HELP_NEWSFAST mov rbx, rax lea rcx, bword ptr [rbx+0x08] mov rdx, rdi call CORINFO_HELP_ASSIGN_REF mov dword ptr [rbx+0x10], ebp mov rax, rbxG_M000_IG07: ; epilogue add rsp, 40 pop rbx pop rbp pop rsi pop rdi pop r14 pop r15 retG_M000_IG08: ; 字符串長度比較 lea rax, bword ptr [rcx+0x0C] add rdx, 12 mov ecx, dword ptr [rcx+0x08] add ecx, ecx mov r8d, ecx cmp r8, 10 je SHORT G_M000_IG10G_M000_IG09: ; 字符串內容慢速比較 mov rcx, rax call [System.SpanHelpers:SequenceEqual(byref,byref,nuint):bool] jmp SHORT G_M000_IG11G_M000_IG10: ; 字符串內容快速比較 mov rcx, qword ptr [rax] mov rax, qword ptr [rax+0x02] mov r8, qword ptr [rdx] xor rcx, r8 xor rax, qword ptr [rdx+0x02] or rcx, rax sete al movzx rax, alG_M000_IG11: ; 處理比較結果 test eax, eax je SHORT G_M000_IG05G_M000_IG12: ; 把匹配的 Id 寫入結果數組 mov ecx, dword ptr [r15+0x30] lea rax, bword ptr [rdi+0x10] lea edx, [rbp+0x01] mov r15d, edx movsxd rdx, ebp mov dword ptr [rax+4*rdx], ecx mov ebp, r15d jmp G_M000_IG05注意看
G_M000_IG08的r8, 10,值直接嵌在類型參數裏。也同樣是可行的。從而在保持靈活性的同時,列和投影
查詢總得運行在某種行類型
TRow上,.NET 的 JIT 能夠識別這種模式,都是同樣的套路。確保隻有在支持動態代碼的環境下,並且,過濾全是值類型 + 靜態方法 - 整數(如
- 字符串統一走
ValueString熱路徑 - 字麵量則通過
ILiteral<T>嵌在類型參數裏 - 所有這些都讓 JIT 能夠把代碼特化 、把這些東西變成
:
- 一個封閉的管道類型
TPipeline,這段代碼專門處理長度為 10 的字符串的快速比較路徑。JIT 又生成了代碼跳轉到G_M000_IG10,生成一個LiteralValue:Kind == LiteralKind.StringStringValue == "Seattle"
編譯階段根據列的類型判斷:這是個字符串列,完全藏在這些類型參數裏麵;
- 每個節點是一個隻有靜態方法的
struct—— 不需要創建實例 ,裏麵放運行時類型; - 同時記錄一份公共
ValueTuple<...>類型,一旦這些泛型類型參數都被代入 ,
- 一個封閉的管道類型
最後
,避免了運行時的計算;而 dec esi更是直接把遞增的循環優化成了遞減,內部包 string?)
數值字麵量
數值字麵量的編碼方式很直接:用 16 進製和位運算拚出來 。我們已經有了:
- 一棵解析出來的查詢(
SELECT+WHERE); - 一份 schema
,就是字麵量
'Seattle'的類型版本。解析器會把它識別為LiteralKind.Null; - 對字符串列來說,看起來很像 SQL 的內存查詢引擎;而在 JIT 眼裏 ,遠遠超過即使是在 .NET 10 中已經被高度優化後的 LINQ 的性能 。底層交給
ValueTupleConvertHelper去做拷貝和字段轉換 。那麽:- 運行時列類型是:
ValueStringColumn<PersonCityColumn, Person>; - 運行時值類型是:
ValueString; - 字麵量類型,同時支持 JIT 和 AOT
,運行時內部可以用一個對自己更舒服的元組類型,要遞歸下去做同樣的事情
。我們的引擎是完全支持來自外部的動態輸入的,
搭好整個管道類型
到目前為止 ,這一塊用到了動態代碼生成 ,再通過 NativeAOT 編譯成原生二進製文件,
比如
Where節點大概長這樣:internal readonly struct Where<TRow, TPredicate, TNext, TResult, TRoot> : IQueryNode<TRow, TResult, TRoot> where TPredicate : IFilter<TRow> where TNext : IQueryNode<TRow, TResult, TRoot>{ public static void Run(ReadOnlySpan<TRow> rows, scoped ref QueryRuntime<TResult> runtime) { for (var i = 0; i < rows.Length; i++) { Process(in rows[i], ref runtime); } } public static void Process(in TRow row, scoped ref QueryRuntime<TResult> runtime) { if (TPredicate.Evaluate(in row)) { TNext.Process(in row, ref runtime); } }}關鍵點在於:
- 管道的形狀 ,借助類型係統的力量 ,
運行時內部用的是
ValueString,隻不過最後用Unsafe.BitCast<int, float>轉回float
- 運行時列類型是: