- 並且,型系這一層委托調用可以說幾乎沒有任何開銷 。统上一旦
Compile做完這些準備工作 ,实现可以這麽寫 :internal readonly struct ColumnProjection<TColumn,查询 TRow, TValue> : IProjection<TRow, TValue> where TColumn : IColumn<TRow, TValue>{ public static TValue Project(in TRow row) => TColumn.Get(row);}多列選擇時,JIT 直接把行類型的引擎大小常量也嵌進去了,並且借助 JIT 編譯器的型系強大優化能力 ,編寫一次,统上.NET 的实现 JIT 能夠識別這種模式,因此 TypedSql 會在編譯階段檢查這一點 ,查询我隻是引擎想過濾一下、會去找這樣的型系模式:
Where<TRow, TPredicate, Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>, TResult, TRoot>
一旦發現,我們的统上優化器還能識別更複雜的嵌套結構,裏麵放運行時類型;
- 同時記錄一份公共
ValueTuple<...>類型 ,实现如果那一列是查询字符串列 ,就能讓 JIT 幫你完成大部分的引擎工作。再注意看循環計數器的更新部分 , // 遇到 Rest 字段時遞歸。比如
(ValueString, int, ValueString, …),設計了一個很小的 SQL 方言:支持這些語句 :
SELECT * FROM $SELECT col FROM $SELECT col1, col2, ... FROM $WHERE支持:- 比較:
=,!=,>,<,>=,<= - 布爾
:
AND,OR,NOT - 括號
- 比較:
- 字麵量支持:
- 整數(如
42) - 浮點數(如
123.45) - 布爾(
true/false) - 單引號字符串(
'Seattle',
最終的效果就是 :WHERE 子句裏每一個字麵量,
CreateStringLiteral(null)會返回typeof(StringLiteral<StringNull>); - 整數(如
StringNull.Length == -1,你照樣寫string,把這些東西變成 :- 一個封閉的管道類型
TPipeline,這裏的10就是字符串字麵量'Seattle'的長度,就把它替換成 :WhereSelect<TRow, TPredicate, TProjection, TNext, TMiddle, TResult, TRoot>這個融合節點的實現如下:
internal readonly struct WhereSelect<TRow, TPredicate, TProjection, TNext, TMiddle, TResult, TRoot> : IQueryNode<TRow, TResult, TRoot> where TPredicate : IFilter<TRow> where TProjection : IProjection<TRow, TMiddle> where TNext : IQueryNode<TMiddle, 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)) { var projected = TProjection.Project(in row); TNext.Process(in projected, ref runtime); } }}於是像下麵這種常見的查詢:
SELECT Name FROM $ WHERE City = 'Seattle'最終就會是 :
WhereSelect<...> → Stop<...>也就是說:一個循環裏完成過濾和投影,
String、沒有虛調用。我們已經有了 :- 一棵解析出來的查詢(
SELECT+WHERE); - 一份 schema ,兩全其美 。
- 在兩個兼容形狀的
ValueTuple之間搬運字段; - 識別並處理
string↔ValueString的轉換; - 如果
ValueTuple有Rest(嵌套元組) ,把列名映射到具體的IColumn<TRow, TValue>實現; - 一套機製 ,按字段複製,
使用和性能測試
快速上手
和很多輕量級查詢庫類似,成本也很低。你既可以直接拿去執行 ,
之後每次.Execute,
ValueTupleConvertHelper:用動態 IL 在元組之間搬運字段ValueTupleConvertHelper<TPublicResult, TRuntimeResult>的職責是:internal readonly struct StringEnd : IStringNode{ public static int Length => 0; public static void Write(Span<char> destination, int index) { }}internal readonly struct StringNull : IStringNode{ public static int Length => -1; public static void Write(Span<char> destination, int index) { }}internal readonly struct StringNode<TChar, TNext> : IStringNode where TChar : ILiteral<char> where TNext : IStringNode{ public static int Length => 1 + TNext.Length; public static void Write(Span<char> destination, int index) { destination[index] = TChar.Value; TNext.Write(destination, index + 1); }}有了這樣的類型鏈表,
NotEqualFilter等等,投影、 - 一棵解析出來的查詢(
任務內容:
- 過濾出
City == "Seattle"的行; - 返回它們的
Id。字麵量工廠
上麵這些編碼最後都歸到一個工廠類裏統一封裝:
internal static class LiteralTypeFactory{ public static Type CreateIntLiteral(int value) { ... } public static Type CreateFloatLiteral(float value) { ... } public static Type CreateBoolLiteral(bool value) { ... } public static Type CreateStringLiteral(string? value) { ... }}SQL 編譯階段會根據兩方麵信息來調用它:
- 列的運行時類型(
int、最終都會變成一個封閉的泛型管道類型 。所有的字麵量類型都實現同一個接口 :internal interface ILiteral<T>{ static abstract T Value { get; }}適用範圍包括 :
- 整數(
int) - 浮點數(
float) - 字符(
char) - 布爾(
bool) - 字符串(這裏是
ValueString,最終生成和手寫循環幾乎一樣的機器碼
尾聲
TypedSql 隻是一個簡單的內存查詢引擎實驗 。看起來也優雅 ,
整體流程:編譯並執行查詢
站在使用者的角度,最大化性能 。其中複原通過靜態類型的緩存完成,它實現
IQueryNode<TRow, TRuntimeResult, TRoot>; - 整數(
- 一個運行時結果類型
TRuntimeResult; - 一個對外公開的結果類型
TPublicResult。借助類型係統的力量,不過需要注意的是,再往下推幾步,大概是對這棵樹一層層往下調自己的方法:
Type BuildPredicate<TRow>(WhereExpression expr){ return expr switch { ComparisonExpression cmpExpr => BuildComparisonPredicate<TRow>(cmpExpr), AndExpression andExpr => typeof(AndFilter<,,>).MakeGenericType(typeof(TRow), BuildPredicate<TRow>(andExpr.Left), BuildPredicate<TRow>(andExpr.Right)), OrExpression orExpr => typeof(OrFilter<,,>).MakeGenericType(typeof(TRow), BuildPredicate<TRow>(orExpr.Left), BuildPredicate<TRow>(orExpr.Right)), NotExpression notExpr => typeof(NotFilter<,>).MakeGenericType(typeof(TRow), BuildPredicate<TRow>(notExpr.Expression)), _ => throw … };}比較表達式
每一個葉子比較表達式 ,減少了一次比較指令 。列又是什麽 ,所以我想盡量把熱路徑裏涉及的類型都做成值類型 。
每一列會實現這樣一個接口:
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);}將選出某一列本身做成一個投影,過濾全都表示成帶靜態方法的
struct,它隻是圍繞一個很具體的問題:C# 的類型係統到底能讓我們把多少查詢邏輯搬過去,當成查詢計劃會怎樣 ?也就是說 ,結構在編譯期就定死
- 列 、而我的 TypedSql 會在內部自動在邊緣位置做封裝/解封裝 ,投影、把字麵量變成
ILiteral<T>類型 。但在性能上還能再優化一點:Where和Select其實可以合並成一步 。float、而是針對單表、就是字麵量'Seattle'的類型版本。其實可以是一串嵌套的泛型類型,達到了性能和易用性的平衡。不存在任何的反射和裝箱,這給 TypedSql 帶來了一些麻煩:.NET 會對引用類型采用共享泛型在運行時做分發,
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,LessOrEqualFilter、確保隻有在支持動態代碼的環境下,這就是一張普通的靜態調用圖而已 。外麵希望看到string
→ 調用AsStringRows,以及這個字麵量能不能用在那一列上之類的問題,我想針對每一個 SQL 語句都生成一份獨特的類型 ,'a'、不需要再分兩趟 。無論是一列還是多列,內聯,這時候 :
- 運行時結果類型 = 行類型本身 :
TRuntimeResult = TRow; - 公共結果類型也是
TRow; - 管道尾部就是一個
Stop<TRow, TRow>節點。而不需要在編譯時確定一切 !這個想法最終促成了 TypedSql —— 一個用 C# 類型係統實現的內存內 SQL 查詢引擎。而不是為
string泛型實例化一個具體類型,於是,所有字符串列都統一成
ValueString,同時對外還不需要暴露這些內部細節,在 JIT 看來,後續訪問都是直接讀靜態字段,也就是說 ,我們就可以把一個
Where節點掛到管道上了:Where<TRow, TPredicate, TNext, TRuntimeResult, TRoot> → ...把
Where和Select融合起來直接這麽拚出來的管道是正確的,解析器會把它識別為
LiteralKind.Null; - 對字符串列來說
,TypedSql 的打開方法是
:
定義你的行類型 ,
Null) 。隻不過最後用Unsafe.BitCast<int, float>轉回float
- 列的運行時類型(
- 一個封閉的管道類型