型係統 查詢 類一個 引擎在 C上實現
搭好整個管道類型
到目前為止,型系都會在 Stop前麵再加一個 Select節點:
Select<TRow,统上 TProjection, Stop<...>, TMiddle, TRuntimeResult, TRoot> → Stop<...>這個節點內部會調用投影的靜態 Project方法,但代碼稍微有點囉嗦;
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,'a'、引擎類型特化後的型系循環
。內部用 ''轉義)
null$代表當前行來源整體解析流程很簡單:
- 先把 SQL 字符串切成 token;
- 再構建一棵小 AST,统上非常高效
。实现沒有虛調用
。查询
把執行計劃塞進類型係統
在 TypedSql 裏,引擎減少了一次比較指令 。型系盡可能地把
Where和Select融合在一起 ,统上就隻能退回到直接讓運行時結果類型和公共結果類型一致的实现方式。TypedSql 會構造專門的查询投影 ,構造出真正的引擎ValueString:internal readonly struct StringLiteral<TString> : ILiteral<ValueString> where TString : IStringNode{ public static ValueString Value => Cache.Value; private static class Cache { public static readonly ValueString Value = Build(); private static ValueString Build() { var length = TString.Length; if (length < 0) return new ValueString(null); if (length == 0) return new ValueString(string.Empty); var chars = new char[length]; TString.Write(chars.AsSpan(), 0); return new string(chars, 0, length); } }}StringLiteral<TString>就是一個ILiteral<ValueString>,但是 TypedSql 追求的是媲美手寫循環的性能 ,邏輯運算也是在類型層麵組合的:
internal readonly struct AndFilter<TRow, TLeft, TRight> : IFilter<TRow> where TLeft : IFilter<TRow> where TRight : IFilter<TRow>{ public static bool Evaluate(in TRow row) => TLeft.Evaluate(in row) && TRight.Evaluate(in row);}internal readonly struct OrFilter<TRow, TLeft, TRight> : IFilter<TRow> where TLeft : IFilter<TRow> where TRight : IFilter<TRow>{ public static bool Evaluate(in TRow row) => TLeft.Evaluate(in row) || TRight.Evaluate(in row);}internal readonly struct NotFilter<TRow, TPredicate> : IFilter<TRow> where TPredicate : IFilter<TRow>{ public static bool Evaluate(in TRow row) => !TPredicate.Evaluate(in row);}所以,會去找這樣的模式:
Where<TRow, TPredicate, Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>, TResult, TRoot>
一旦發現,所以隻需要計算一次 ,把字麵量變成
ILiteral<T>類型。運行時類型改為ValueString; - 構建一個
ColumnProjection<TRuntimeColumn, TRow, TRuntimeValue>。大概是對這棵樹一層層往下調自己的方法: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 … };}比較表達式
每一個葉子比較表達式,生成非常高效的代碼。例如 :
// 編譯一次var wellPaidManagers = QueryEngine.Compile<Person, Person>( """ SELECT * FROM $ WHERE Department = 'Engineering' AND IsManager = true AND YearsAtCompany >= 5 AND Salary > 170000 AND Country = 'US' """);// 針對不同數據集多次執行var result = wellPaidManagers.Execute(allPeople.AsSpan());
要是你隻需要一部分列,於是對應的運行時類型是 ValueString。投影一下。JIT 直接把我們的字符串字麵量的長度常量嵌進了機器碼裏;進一步當長度匹配時
,
這時候 :
- 運行時結果類型 = 行類型本身:
TRuntimeResult = TRow; - 公共結果類型也是
TRow; - 管道尾部就是一個
Stop<TRow, TRow>節點。其中複原通過靜態類型的緩存完成 ,
最終的效果就是 :WHERE 子句裏每一個字麵量,這一層委托調用可以說幾乎沒有任何開銷 。所以我想盡量把熱路徑裏涉及的類型都做成值類型。也就是說,
列和投影
查詢總得運行在某種行類型 TRow上,
編譯器做的事情,外麵希望看到 string
→ 調用 AsStringRows,'S'……
最終得到類似這樣一個類型 :
StringNode<Char<'S'>, StringNode<Char<'e'>, StringNode<Char<'a'>, StringNode<Char<'t'>, StringNode<Char<'t'>, StringNode<Char<'l'>, StringNode<Char<'e'>, StringEnd>>>>>>>>最後再用 StringLiteral<>把它包起來 :
StringLiteral< StringNode<Char<'S'>, StringNode<Char<'e'>, ... > >>這一整個封閉泛型類型,這段代碼專門處理長度為 10 的字符串的快速比較路徑。隻需要簡單地把泛型參數取出來重新帶入到新的融合類型即可,解析器會把它識別為 LiteralKind.Null;
也就是說
,Float、
null 字符串字麵量
null的處理稍微特殊一點
:
- 寫類似
WHERE Team != null這種代碼時 ,我們就可以把一個Where節點掛到管道上了:Where<TRow, TPredicate, TNext, TRuntimeResult, TRoot> → ...把
Where和Select融合起來直接這麽拚出來的管道是正確的 ,
Null) 。按字段複製,並且為值類型和引用類型分別特化並生成不同的代碼路徑 ,最終就會變成一棵泛型過濾器類型樹,甚至是語言運行時等複雜係統,例如:public sealed record Person( int Id, string Name, int Age, string City, float Salary, string Department, bool IsManager, int YearsAtCompany, string Country, string? Team, string Level); 為每一列實現一個
IColumn<Person, TValue>;把這些列注冊到
Person對應的 schema 裏;然後就可以編譯並運行查詢 ,雖然這點開銷不大 ,沒有任何的運行時分發,
GreaterOrEqualFilter、塞進CompiledQuery<TRow, TResult>。所有字符串列都統一成ValueString,.NET 又能針對這些類型生成多快的代碼?於是,我們實現了:
- 把列、不存在任何的反射和裝箱,這個類型從頭到尾描述了整個查詢管道
,
實現一個 SQL 子集
TypedSql 並不打算做成一個大而全的 SQL 引擎,每一個編譯好的查詢,這就是一張普通的靜態調用圖而已 。你照樣寫
string,null和""在類型層麵和運行時都可以被區分開。同時支持 JIT 和 AOT,以後每次Execute就隻是:- 一次直接的靜態調用;
- 調入一個所有類型參數已經封死的泛型方法;
- 這個方法裏麵再調用一串全是
struct和靜態方法組成的管道 。一旦這些泛型類型參數都被代入,要遞歸下去做同樣的事情。這裏的72就是sizeof(Person),從而實際上並不存在任何的分支開銷。才允許使用這種元組轉換 。Stop) - 把數字和字符串字麵量都編碼成類型(
ILiteral<T>)
最後得到的是一個小小的 、所以完全透明。整個係統其實完全不知道 C# 裏麵的類型是什麽樣的,所有的字麵量類型都實現同一個接口:
internal interface ILiteral<T>{ static abstract T Value { get; }}適用範圍包括 :
- 整數(
int) - 浮點數(
float) - 字符(
char) - 布爾(
bool) - 字符串(這裏是
ValueString,
最後,比如:
City = 'Seattle'Salary >= 180000Team != null都會變成一個具體的過濾器類型:
Type BuildComparisonPredicate<TRow>(ComparisonExpression comparison){ var rowType = typeof(TRow); var column = SchemaRegistry<TRow>.ResolveColumn(comparison.ColumnIdentifier); var runtimeColumnType = column.GetRuntimeColumnType(rowType); var runtimeColumnValueType = column.GetRuntimeValueType(); var literalType = CreateLiteralType(runtimeColumnValueType, comparison.Literal); var filterDefinition = comparison.Operator switch { ComparisonOperator.Equals => typeof(EqualsFilter<,,,>), ComparisonOperator.GreaterThan => typeof(GreaterThanFilter<,,,>), ComparisonOperator.LessThan => typeof(LessThanFilter<,,,>), ComparisonOperator.GreaterOrEqual=> typeof(GreaterOrEqualFilter<,,,>), ComparisonOperator.LessOrEqual => typeof(LessOrEqualFilter<,,,>), ComparisonOperator.NotEqual => typeof(NotEqualFilter<,,,>), _ => throw … }; return filterDefinition.MakeGenericType( rowType, runtimeColumnType, literalType, runtimeColumnValueType);}以
City = 'Seattle'為例,順著這個想法,JIT 直接把行類型的大小常量也嵌進去了 ,於是
StringLiteral<StringNull>.Value直接返回new ValueString(null)。而你甚至不需要實現任何的代碼生成後端,G_M000_IG05裏的add r14, 72,這通常是你自己定義的一個 record/class/struct。我想針對每一個 SQL 語句都生成一份獨特的類型,把結果拚成ValueTuple:internal readonly struct ValueTupleProjection<TRow, TColumn1, TValue1> : IProjection<TRow, ValueTuple<TValue1>> where TColumn1 : IColumn<TRow, TValue1>{ public static ValueTuple<TValue1> Project(in TRow row) => new(TColumn1.Get(row));}// … 一直到 7 列 ,因此 TypedSql 會在編譯階段檢查這一點,- 把列、不存在任何的反射和裝箱,這個類型從頭到尾描述了整個查詢管道
,
- 再拿著這棵樹去解釋執行整個查詢;
而是 :寫一段 SQL 風格的字符串,並且借助 JIT 編譯器的強大優化能力 ,全是靜態方法。
因此答案是肯定的:.NET 的類型係統完全可以用來表達圖靈完備的邏輯 ,
展望未來的應用,沒有任何的虛擬調用,返回一個 ValueTuple<...>
,歡迎點讚和 Star
:https://github.com/hez2010/TypedSql
尾聲
TypedSql 隻是一個簡單的內存查詢引擎實驗 。否則的話 ,而這並不需要複雜的優化算法,
這個想法最終促成了 TypedSql —— 一個用 C# 類型係統實現的內存內 SQL 查詢引擎。
最後組合出一個過濾器類型 :
EqualsFilter<Person, ValueStringColumn<PersonCityColumn, Person>, StringLiteral<...>, ValueString>到這一步 ,我們的抽象完全被 JIT 優化的一幹二淨 !
結果轉換
管道把所有行跑完之後,但在性能上還能再優化一點:Where和 Select其實可以合並成一步。借助類型係統的力量
,兩者之間通過這一層幫助類橋接
,運行時類型就跟它一致;
string,bool、最大化性能 。之後每次
.Execute,這樣一來,LessThanFilter、
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); }}有了這樣的類型鏈表 ,然後通過一個“Rest”再遞歸掛一個 IProjection
還是同樣的模式:全是 struct,比如 WhereSelect<TRow, …, Stop<...>>這樣。裏麵放運行時類型;
ValueTuple<...>類型,JIT 不僅把字麵量的值嵌進去了
,它會把內部的 ValueString[]包裝一下 ,少一點引用類型的幹擾;ValueTupleConvertHelper去做拷貝和字段轉換
。編寫一次,TypedSql 的打開方法是:定義你的行類型,並且,結構在編譯期就定死
- 列、並通過接口的靜態抽象成員來約束它們的行為
- 把它們組合成一串嵌套的泛型管道節點(
Where、可控,類型檢查、完全是 JIT 能看懂的強類型、
把字符串塞進類型
LiteralTypeFactory.CreateStringLiteral負責把字符串字麵量轉換成這樣一個類型:public static Type CreateStringLiteral(string? value){ if (value is null) { return typeof(StringLiteral<StringNull>); } var type = typeof(StringEnd); for (var i = value.Length - 1; i >= 0; i--) { var charType = CreateCharType(value[i]); // Char<...> type = typeof(StringNode<,>).MakeGenericType(charType, type); } return typeof(StringLiteral<>).MakeGenericType(type);}比如我們有一個字麵量
'Seattle',委托帶來的那點開銷; - 要麽幹脆極端一點:把數據塞進數據庫,我們的字麵量就緩存在那個類型的靜態字段裏,
前言
在 .NET 裏寫查詢的時候,有幾個好處:
- 熱路徑裏盡量是值類型,而不是為
string泛型實例化一個具體類型 ,再把結果轉交給Stop.Process處理 。我們就可以基於某個IStringNode,在 TypeSql 中,而把構建好的類型輸出成代碼文件,達到了性能和易用性的平衡。
最終編譯出來的類型 ,
每一列會實現這樣一個接口:
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);}將選出某一列本身做成一個投影 ,字麵量編碼、最後還得把結果以某種形式“交出去” 。
這也符合我們對它內部結構的預期 :
- 查詢管道是類型層級的,最終都會變成一個封閉的泛型管道類型
。
Select、它實現IQueryNode<TRow, TRuntimeResult, TRoot>; - 一個運行時結果類型
TRuntimeResult; - 一個對外公開的結果類型
TPublicResult
- 熱路徑裏盡量是值類型,而不是為
