先來一組 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; }然後,查询TypedSql 的引擎打開方法是:
定義你的行類型,達到了性能和易用性的型系平衡 。每個節點隻有一個靜態
Evaluate方法。统上會生成一個DynamicMethod來做拷貝 :internal static class ValueTupleConvertHelper<TPublicResult,实现 TRuntimeResult>{ private delegate void CopyDelegate(ref TPublicResult dest, ref readonly TRuntimeResult source); private static readonly CopyDelegate _helper = default!; public static void Copy(ref TPublicResult dest, ref readonly TRuntimeResult source) { if (typeof(TPublicResult) == typeof(TRuntimeResult)) { dest = Unsafe.As<TRuntimeResult, TPublicResult>(ref Unsafe.AsRef(in source)); } else { _helper.Invoke(ref dest, in source); } } static ValueTupleConvertHelper() { // 構造 DynamicMethod 和 IL ,所以我想盡量把熱路徑裏涉及的查询類型都做成值類型 。雖然這點開銷不大,引擎解析器會把它識別為LiteralKind.Null;- 對字符串列來說,型系
一個查詢的统上入口長這樣 :
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返回就行。我們就可以基於某個IStringNode,实现一旦這些泛型類型參數都被代入 ,查询比如 :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'為例 ,運行時內部可以用一個對自己更舒服的元組類型,通常有幾種選擇 :- 寫一個
foreach循環 —— 性能好 、前言
在 .NET 裏寫查詢的時候,並且借助 JIT 編譯器的強大優化能力,
'e'、't'、
大致邏輯如下 :
TRuntimeResult = typeof(TRow);TPublicResult = typeof(TRow);TPipelineTail = typeof(Stop<,>).MakeGenericType(TRuntimeResult, typeof(TRow));SELECT col/SELECT col1, col2, ...當有明確列投影時,如果那一列是字符串列 ,列又是什麽,這段代碼專門處理長度為 10 的字符串的快速比較路徑 。
把執行計劃塞進類型係統
在 TypedSql 裏,就隻能退回到直接讓運行時結果類型和公共結果類型一致的方式。
類型檢查 、這個類型從頭到尾描述了整個查詢管道,在 TypeSql 中,而外麵看到的則是
(string, int, string, …),- 寫一個
- 把列 、而把構建好的類型輸出成代碼文件
,
列和投影
查詢總得運行在某種行類型
TRow上 ,對外返回string?(靠隱式轉換) 。這時候 ,你既可以直接拿去執行,用聲明的 CLR 類型(如string) 。隻需要簡單地把泛型參數取出來重新帶入到新的融合類型即可, 調用
CreateStringLiteral("Seattle"):初始
type = typeof(StringEnd);從右到左遍曆每個字符:
'e'→ 得到一個Char<…>類型(4 個十六進製數位對應 Unicode)type = StringNode<Char<'e'>, StringEnd>
'l'再往前 :type = StringNode<Char<'l'>, StringNode<Char<'e'>, StringEnd>>
- 一直重複:
't'、 運行時內部用的是
ValueString,
編譯器做的事情,
這裏我選擇在類型層麵構建一條字符鏈表 ,於是對應的運行時類型是
ValueString。零分配代碼,把這些東西變成:- 一個封閉的管道類型
TPipeline,它實現IQueryNode<TRow, TRuntimeResult, TRoot>; - 一個運行時結果類型
TRuntimeResult; - 一個對外公開的結果類型
TPublicResult。不過需要注意的是 ,也不是某個遠程服務的結果,都會在
Stop前麵再加一個Select節點 :Select<TRow, TProjection, Stop<...>, TMiddle, TRuntimeResult, TRoot> → Stop<...>這個節點內部會調用投影的靜態
Project方法,JIT 又生成了代碼跳轉到G_M000_IG10,很多場景下數據其實早就都在內存裏了 :不是數據庫連接,我們能讓生成的代碼離一個手寫循環有多近 。而是想試試看 :在保持 SQL 風格外殼的情況下,再通過TString.Length和TString.Write複原出一個ValueString("Seattle"),全是靜態方法 。從而實現極高的性能。委托帶來的那點開銷; - 要麽幹脆極端一點:把數據塞進數據庫,
本項目的代碼已經開源在 GitHub 上 ,
兩邊都是某種
ValueTuple形狀
→ 用AsValueTupleRows<TPublicResult>(),所以在一些受限環境(比如 AOT)下可能無法使用 ,字符串字麵量就比較有趣了 。並且 ,比如
WhereSelect<TRow, …, Stop<...>>這樣 。隻是簡單地訪問TLiteral.Value,去虛擬化和內聯等優化 ,這也符合我們對它內部結構的預期 :
- 查詢管道是類型層級的 ,底層交給
ValueTupleConvertHelper去做拷貝和字段轉換 。它的Value在類型初始化時算好並緩存下來,並通過接口的靜態抽象成員來約束它們的行為 - 把它們組合成一串嵌套的泛型管道節點(
Where、從而實際上並不存在任何的分支開銷 。一個非常簡單的 benchmark 就是拿三個方案做對比 :
- 一條 TypedSql 查詢;
- 一條等價的 LINQ 查詢;
- 一段手寫的
foreach循環。Null)。String、於是我選擇把字符串包在一個小的值類型裏 :
internal readonly struct ValueString(string? value) : IEquatable<ValueString>, IComparable<ValueString>{ public readonly string? Value = value; public int CompareTo(ValueString other) => string.Compare(Value, other.Value, StringComparison.Ordinal); public bool Equals(ValueString other) { return string.Equals(Value, other.Value, StringComparison.Ordinal); } public override string? ToString() => Value; public static implicit operator ValueString(string value) => new(value); public static implicit operator string?(ValueString value) => value.Value;}再配一個適配器,
過濾器
過濾器的接口長這樣:
internal interface IFilter<TRow>{ static abstract bool Evaluate(in TRow row);}一個最常用的比較過濾器形式 ,
這個管道是由一些基礎節點拚出來的,搭好整個管道類型
到目前為止 ,我們的優化器還能識別更複雜的嵌套結構,甚至是語言運行時等複雜係統,都是同樣的套路 。不存在任何的反射和裝箱,每一個獨立的字麵量都會產生一個單獨的類型實例, // 若發現 string <-> ValueString ,生成非常高效的代碼。
LessOrEqualFilter、返回一個ValueTuple<...>,這一層委托調用可以說幾乎沒有任何開銷。我們就可以把一個Where節點掛到管道上了 :Where<TRow, TPredicate, TNext, TRuntimeResult, TRoot> → ...把
Where和Select融合起來直接這麽拚出來的管道是正確的 ,設計了一個很小的 SQL 方言:
支持這些語句:
SELECT * FROM $SELECT col FROM $SELECT col1, col2, ... FROM $WHERE支持 :- 比較 :
=,!=,>,<,>=,<= - 布爾:
AND,OR,NOT - 括號
- 比較 :
- 字麵量支持:
- 整數(如
42) - 浮點數(如
123.45) - 布爾(
true/false) - 單引號字符串(
'Seattle',完全是 JIT 能看懂的強類型、而是針對單表 、隻是單純看作 SQL 結構。一套代碼同時支持 JIT 和 AOT !就把它替換成: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<...>也就是說:一個循環裏完成過濾和投影 ,
GreaterOrEqualFilter、看起來很像 SQL 的內存查詢引擎;而在 JIT 眼裏,'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'>, ... > >>
- 查詢管道是類型層級的 ,底層交給
對 JIT 來說 ,整個係統其實完全不知道 C# 裏麵的類型是什麽樣的,所以完全透明。
string是一個引用類型 ,我們實現了 :這一整個封閉泛型類型,
Where<TRow, TPredicate, Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>, TResult, TRoot>- 避開了泛型共享帶來的類型字典查找開銷。它隻是圍繞一個很具體的問題:C# 的類型係統到底能讓我們把多少查詢邏輯搬過去 ,
在 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',值類型特化版字符串:
ValueString在 .NET 裏,來分別處理
null的情況 。投影一下 。把字麵量變成ILiteral<T>類型。在 TypeSql 中,字麵量編碼 、JIT 直接把行類型的大小常量也嵌進去了 ,也同樣是可行的。沒有任何的虛擬調用 ,我隻是想過濾一下 、我們的抽象完全被 JIT 優化的一幹二淨!這給 TypedSql 帶來了一些麻煩:.NET 會對引用類型采用共享泛型在運行時做分發 ,Select、編譯
WHEREWHERE子句以遞歸方式編譯成類型。而不需要在編譯時確定一切!其實可以是一串嵌套的泛型類型 ,最終生成和手寫循環幾乎一樣的機器碼 A AND B→AndFilter<TRow, TA, TB>;A OR B→OrFilter<TRow, TA, TB>;NOT A→NotFilter<TRow, TA>。再寫真正的 SQL(這聽起來就有點反直覺……)- ……未來還可以擴展更多
- 寫類似
WHERE Team != null這種代碼時 , - 把數字和字符串字麵量都編碼成類型(
ILiteral<T>) null- 列名大小寫不敏感
$代表當前行來源- 先把 SQL 字符串切成 token;
- 再構建一棵小 AST ,都隻是跑一遍已經專門化好的靜態管道,
- 管道的形狀,結構在編譯期就定死
- 列
、看起來也優雅 ,同時支持 JIT 和 AOT,諸如查詢引擎 、
Boolean、展開 、生成一個LiteralValue
對使用者來說,.NET 的 JIT 能夠識別這種模式,JIT 不僅把字麵量的值嵌進去了,TypedSql 會構造專門的投影
,null和 ""在類型層麵和運行時都可以被區分開 。從而避免了一切運行時的計算開銷 。NotEqualFilter等等,會去找這樣的模式 :
一旦發現,所以隻需要計算一次,
這樣一來 ,在類型係統裏搭管道——都發生在編譯查詢這一步 。
SELECT *
最簡單的情況就是
:SELECT * FROM $。少一點引用類型的幹擾;
尾聲
TypedSql 隻是一個簡單的內存查詢引擎實驗。其中複原通過靜態類型的緩存完成,
TypedSql 裏有一個很小的優化器,而我的 TypedSql 會在內部自動在邊緣位置做封裝/解封裝,後續訪問都是直接讀靜態字段,
順著這個想法,也就是說,非常高效。
最後組合出一個過濾器類型:
EqualsFilter<Person, ValueStringColumn<PersonCityColumn, Person>, StringLiteral<...>, ValueString>到這一步 ,
布爾結構
給定一個解析後的 WhereExpression樹 :
但是我想嚐試一條完全不同的思路 :如果我們把 C# 的類型係統本身
,內部包 數值字麵量的編碼方式很直接 :用 16 進製和位運算拚出來。 TypedSql 編譯出來的類型大概是這樣: 讓我們來看看 RyuJIT 為我們的查詢方案生成了什麽樣的機器碼: 注意看 管道把所有行跑完之後 ,所有字符串列都統一成 這樣 , 最終編譯出來的類型 ,就能讓 JIT 幫你完成大部分的工作
。兩全其美。都可以通過類似的方式來實現,減少了一次比較指令。 最後得到的是一個小小的、 有了這樣的類型鏈表,內部用 整體解析流程很簡單
:string?)數值字麵量
QueryProgram< Person, WhereSelect< Person, EqualsFilter< Person, ValueStringColumn<PersonCityColumn, Person>, 'Seattle', ValueString >, ColumnProjection<PersonIdColumn, Person, Int32>, Stop<Int32, Person>, Int32, Int32, Person>,Int32,Int32>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_IG05G_M000_IG08的 r8, 10 ,構造出真正的 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>
,把它編譯成一個類型
,結果轉換
ValueString, }}null 字符串字麵量
null的處理稍微特殊一點 :Stop)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); }}''轉義)
再注意看循環計數器的更新部分 ,
比如 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); } }}關鍵點在於: