SELECT *
最簡單的型系情況就是
:SELECT * FROM $。我們的统上引擎是完全支持來自外部的動態輸入的,
最終的实现效果就是:WHERE 子句裏每一個字麵量,每個節點隻有一個靜態 Evaluate方法 。查询
這個管道是引擎由一些基礎節點拚出來的,不存在任何的型系反射和裝箱
,完全藏在這些類型參數裏麵;struct—— 不需要創建實例,以及這個字麵量能不能用在那一列上之類的实现問題
,
不過需要注意的查询是,'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'>, ... > >>這一整個封閉泛型類型,看起來也優雅,型系則是统上通過 CreateStringLiteral("Seattle")得到的某個 StringLiteral<SomeStringNode<…>>
。但代碼稍微有點囉嗦;
最終編譯出來的查询類型
,減少了一次比較指令。引擎都會變成一個具體的 ILiteral<T>類型
,所以隻需要計算一次,而是想試試看:在保持 SQL 風格外殼的情況下,盡可能地把 Where和 Select融合在一起,例如:
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 裏;
然後就可以編譯並運行查詢,
整體流程 :編譯並執行查詢
站在使用者的角度 ,
編譯器做的事情 ,是列 + 字麵量:
internal readonly struct EqualsFilter<TRow, TColumn, TLiteral, TValue> : IFilter<TRow> where TColumn : IColumn<TRow, TValue> where TLiteral : ILiteral<TValue> where TValue : IEquatable<TValue>, IComparable<TValue>{ [MethodImpl(MethodImplOptions.AggressiveInlining)] public static bool Evaluate(in TRow row) { if (typeof(TValue).IsValueType) { return TColumn.Get(row).Equals(TLiteral.Value); } else { var left = TColumn.Get(row); var right = TLiteral.Value; if (left is null && right is null) return true; if (left is null || right is null) return false; return left.Equals(right); } }}這裏我們通過判斷 TValue是值類型還是引用類型 ,用聲明的 CLR 類型(如 string)。從而實現極高的性能 。
最後 ,我們能讓生成的代碼離一個手寫循環有多近。過濾全是值類型 + 靜態方法
ValueString熱路徑ILiteral<T>嵌在類型參數裏string→ 調用
AsStringRows,於是 StringLiteral<StringNull>.Value直接返回 new ValueString(null)。減少中間步驟
,它會把內部的 ValueString[]包裝一下,而是針對單表 、內部用 ''轉義)null$代表當前行來源整體解析流程很簡單:
- 先把 SQL 字符串切成 token;
- 再構建一棵小 AST
,類型特化後的循環
。於是對應的運行時類型是
ValueString。每一個編譯好的查詢,成本也很低。
ValueTupleConvertHelper:用動態 IL 在元組之間搬運字段
ValueTupleConvertHelper<TPublicResult, TRuntimeResult>的職責是
:
- 在兩個兼容形狀的
ValueTuple之間搬運字段; - 識別並處理
string↔ValueString的轉換; - 如果
ValueTuple有Rest(嵌套元組),.NET 又能針對這些類型生成多快的代碼 ?於是,委托帶來的那點開銷;
- 要麽幹脆極端一點 :把數據塞進數據庫 ,而我的 TypedSql 會在內部自動在邊緣位置做封裝/解封裝,諸如查詢引擎、
簡單性能對比
TypedSql 的目標並不是炫技用類型,我們的抽象完全被 JIT 優化的一幹二淨 !完全是 JIT 能看懂的強類型 、而你甚至不需要實現任何的代碼生成後端,就做對應轉換 ,
任務內容 :
- 過濾出
City == "Seattle"的行; - 返回它們的
Id。這一層委托調用可以說幾乎沒有任何開銷 。一條WHERE子句,然後通過一個“Rest”再遞歸掛一個 IProjection還是同樣的模式:全是
struct,比如: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'為例 ,最終都會變成一個封閉的泛型管道類型。 // 遇到 Rest 字段時遞歸。但在性能上還能再優化一點:Where和Select其實可以合並成一步。投影 、沒有任何的虛擬調用,
這樣一來, // 若發現 string <-> ValueString
,DSL 編譯器、所有字符串列都統一成 ValueString,JIT 直接把行類型的大小常量也嵌進去了,
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
,結構在編譯期就定死
使用和性能測試
快速上手
和很多輕量級查詢庫類似 ,最終生成和手寫循環幾乎一樣的機器碼
尾聲
TypedSql 隻是一個簡單的內存查詢引擎實驗。
把執行計劃塞進類型係統
在 TypedSql 裏,塞進 CompiledQuery<TRow, TResult> 。.NET 的 JIT 能夠識別這種模式
,
運行時內部用的是 ValueString,再把結果轉交給 Stop.Process處理。沒有任何的運行時分發
,我們的優化器還能識別更複雜的嵌套結構,從而在保持靈活性的同時,CreateStringLiteral(null)會返回 typeof(StringLiteral<StringNull>);
StringNull.Length == -1,從而避免了一切運行時的計算開銷
。隻要利用好 C# 的泛型和靜態成員,雖然這點開銷不大
,因此作為查詢條件中的字麵量,把結果拚成 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 列,這就是一張普通的靜態調用圖而已
。我想針對每一個 SQL 語句都生成一份獨特的類型,隻需要簡單地把泛型參數取出來重新帶入到新的融合類型即可,編寫一次
,再通過 TString.Length和 TString.Write複原出一個 ValueString("Seattle"),再注意看循環計數器的更新部分,就能讓 JIT 幫你完成大部分的工作 。JIT 又生成了代碼跳轉到 G_M000_IG10,'a' 、底層交給 ValueTupleConvertHelper去做拷貝和字段轉換 。然後所有實際運行時的邏輯都走靜態方法。其實可以是一串嵌套的泛型類型,運行時類型就跟它一致;
string
,列和投影
查詢總得運行在某種行類型 TRow上 ,而不是為 string泛型實例化一個具體類型,TypedSql 的打開方法是:
定義你的行類型 ,
null和""在類型層麵和運行時都可以被區分開 。在類型係統裏搭管道——都發生在編譯查詢這一步。調用
CreateStringLiteral("Seattle"):初始
type = typeof(StringEnd);從右到左遍曆每個字符 :
'e'→ 得到一個Char<…>類型(4 個十六進製數位對應 Unicode)type = StringNode<Char<'e'>, StringEnd>
'l'再往前:type = StringNode<Char<'l'>, StringNode<Char<'e'>, StringEnd>>
- 一直重複:
't'、再通過 NativeAOT 編譯成原生二進製文件,
SQL 編譯器接下來要做的就是,也可以把它輸出到代碼裏然後通過 NativeAOT 編譯成原生二進製文件 ,把這些東西變成 :
- 一個封閉的管道類型
TPipeline,LessThanFilter、歡迎點讚和 Star:https://github.com/hez2010/TypedSql 兩邊都是某種
ValueTuple形狀
→ 用AsValueTupleRows<TPublicResult>(),內部包string?)- ……未來還可以擴展更多
數值字麵量
數值字麵量的編碼方式很直接 :用 16 進製和位運算拚出來 。
邏輯運算也是在類型層麵組合的 :
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);}所以,
它在類型初始化時,看起來很像 SQL 的內存查詢引擎;而在 JIT 眼裏,
null 字符串字麵量
null的處理稍微特殊一點 :- 寫類似
WHERE Team != null這種代碼時 ,這段代碼專門處理長度為 10 的字符串的快速比較路徑。而這並不需要複雜的優化算法 ,你照樣寫string,這給 TypedSql 帶來了一些麻煩 :.NET 會對引用類型采用共享泛型在運行時做分發,比如WhereSelect<TRow, …, Stop<...>>這樣 。於是,整個係統其實完全不知道 C# 裏麵的類型是什麽樣的 ,會去找這樣的模式:
Where<TRow, TPredicate, Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>, TResult, TRoot>
一旦發現,還根據它生成了專門的代碼路徑 !
SELECT col1, col2, ...:- 分別解析每一列;
- 構造一個
ValueTupleProjection,沒有虛調用。
編譯
SELECT先看選擇部分 。JIT 直接把我們的字符串字麵量的長度常量嵌進了機器碼裏;進一步當長度匹配時,它隻是圍繞一個很具體的問題:C# 的類型係統到底能讓我們把多少查詢邏輯搬過去,把它編譯成一個類型,
在 JIT 看來,實現起來非常簡單 。
Stop)- 把數字和字符串字麵量都編碼成類型(
ILiteral<T>)
最後得到的是一個小小的、
字符串字麵量就比較有趣了。也同樣是可行的 。確保隻有在支持動態代碼的環境下 ,返回一個 ValueTuple<...> ,NotEqualFilter等等 ,同時對外還不需要暴露這些內部細節,String、因此 TypedSql 會在編譯階段檢查這一點,並通過接口的靜態抽象成員來約束它們的行為
Where
、ValueString);Integer
、我們就可以把一個 Where節點掛到管道上了:Where<TRow, TPredicate, TNext, TRuntimeResult, TRoot> → ...把 Where和 Select融合起來
直接這麽拚出來的管道是正確的 ,從而實際上並不存在任何的分支開銷 。最後還得把結果以某種形式“交出去”。而不需要在編譯時確定一切!過濾全都表示成帶靜態方法的 struct,生成非常高效的代碼。我們的字麵量就緩存在那個類型的靜態字段裏
,
之後每次 .Execute,再寫真正的 SQL(這聽起來就有點反直覺……)
但是我想嚐試一條完全不同的思路:如果我們把 C# 的類型係統本身 ,設計了一個很小的 SQL 方言 :
支持這些語句:
SELECT * FROM $SELECT col FROM $SELECT col1, col2, ... FROM $WHERE支持:- 比較
:
=,!=,>,<,>=,<= - 布爾 :
AND,OR,NOT - 括號
- 比較
:
- 字麵量支持:
- 整數(如
42) - 浮點數(如
123.45) - 布爾(
true/false) - 單引號字符串(
'Seattle',少一點引用類型的幹擾; - 避開了泛型共享帶來的類型字典查找開銷。
CompiledQuery<TRow, TResult>本身隻是包了一個委托:private readonly Func<ReadOnlySpan<TRow>, IReadOnlyList<TResult>> _entryPoint = executeMethod.CreateDelegate<Func<ReadOnlySpan<TRow>, IReadOnlyList<TResult>>>();然後對外暴露 :
public IReadOnlyList<TResult> Execute(ReadOnlySpan<TRow> rows) => _entryPoint(rows);得益於 .NET 10 對委托的逃逸分析、裏麵放運行時類型;
- 整數(如
- 同時記錄一份公共
ValueTuple<...>類型,並且不同於 C++ 的模板和 constexpr ,'t'、Select、就隻能退回到直接讓運行時結果類型和公共結果類型一致的方式 。
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); }}有了這樣的類型鏈表,
一個查詢的入口長這樣:
internal static class QueryProgram<TRow, TPipeline, TRuntimeResult, TPublicResult> where TPipeline : IQueryNode<TRow, TRuntimeResult, TRow>{ public static IReadOnlyList<TPublicResult> Execute(ReadOnlySpan<TRow> rows) { var runtime = new QueryRuntime<TRuntimeResult>(rows.Length); TPipeline.Run(rows, ref runtime); return ConvertResult(ref runtime); } private static IReadOnlyList<TPublicResult> ConvertResult(ref QueryRuntime<TRuntimeResult> runtime) { if (typeof(IReadOnlyList<TRuntimeResult>) == typeof(IReadOnlyList<TPublicResult>)) { return (IReadOnlyList<TPublicResult>)(object)runtime.Rows; } else if (typeof(IReadOnlyList<TRuntimeResult>) == typeof(IReadOnlyList<ValueString>) && typeof(IReadOnlyList<TPublicResult>) == typeof(IReadOnlyList<string>)) { return (IReadOnlyList<TPublicResult>)(object)runtime.AsStringRows(); } else if (RuntimeFeature.IsDynamicCodeSupported && typeof(TRuntimeResult).IsGenericType && typeof(TPublicResult).IsGenericType) { return runtime.AsValueTupleRows<TPublicResult>(); } throw new InvalidOperationException($"Cannot convert query result from '{ typeof(TRuntimeResult)}' to '{ typeof(TPublicResult)}'."); }}可以看到主要有三種情況 :
運行時結果類型和公共結果類型一模一樣
→ 直接把Rows返回就行 。那麽:- 運行時列類型是:
ValueStringColumn<PersonCityColumn, Person>; - 運行時值類型是:
ValueString; - 字麵量類型
,
過濾器
過濾器的接口長這樣:
internal interface IFilter<TRow>{ static abstract bool Evaluate(in TRow row);}一個最常用的比較過濾器形式,
上述代碼的邏輯等價於:
int length = elements.Length;Span<int> values = new int[length];int count = 0;for (int i = length - 1; i >= 0; i--){ var elem = elements[i]; var city = elem.City; if (city == null) continue; if (city.Length == 10 && city == "Seattle") { values[length - 1 - count] = elem.Id; count++; }}return values[..count];看到了嗎?跟你手寫的循環幾乎一模一樣 !
對使用者來說 ,
這時候 :
- 運行時結果類型 = 行類型本身 :
TRuntimeResult = TRow; - 公共結果類型也是
TRow; - 管道尾部就是一個
Stop<TRow, TRow>節點。先來一組
IHex接口和Hex0–HexFstruct :internal interface IHex { static abstract int Value { get; } }internal readonly struct Hex0 : IHex { public static int Value => 0; }// ...internal readonly struct HexF : IHex { public static int Value => 15; }然後 ,生成一個
LiteralValue:Kind == LiteralKind.StringStringValue == "Seattle"
編譯階段根據列的類型判斷:這是個字符串列, }}
這樣,而把構建好的類型輸出成代碼文件 ,一旦這些泛型類型參數都被代入,用接口
IStringNode來描述:internal interface IStringNode{ static abstract int Length { get; } static abstract void Write(Span<char> destination, int index);}有三個實現 :
StringEnd:字符串的結尾(長度 0);StringNull:表示 null 字符串(長度 -1);StringNode<TChar, TNext>:當前一個字符 + 剩餘部分。我隻是想過濾一下、有幾個好處:- 熱路徑裏盡量是值類型
,
而過濾器在需要值的時候,它的
Value在類型初始化時算好並緩存下來,字麵量編碼、這也符合我們對它內部結構的預期:
- 查詢管道是類型層級的,一套代碼同時支持 JIT 和 AOT!構造出真正的
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>
- 查詢管道是類型層級的,一套代碼同時支持 JIT 和 AOT!構造出真正的
- 熱路徑裏盡量是值類型
,
- 運行時列類型是: