型係統在 C一個 引擎 類 查詢上實現
Evaluate方法。统上LessThanFilter、实现把字符串塞進類型
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'
,TypedSql 的引擎打開方法是
:
定義你的行類型,設計了一個很小的型系 SQL 方言:
支持這些語句:
SELECT * FROM $SELECT col FROM $SELECT col1, col2, ... FROM $WHERE支持:- 比較 :
=,!=,>,<,>=,<= - 布爾:
AND,OR,NOT - 括號
- 比較 :
- 字麵量支持
:
- 整數(如
42) - 浮點數(如
123.45) - 布爾(
true/false) - 單引號字符串(
'Seattle',展開 、统上整體流程:編譯並執行查詢
站在使用者的实现角度,每一個獨立的查询字麵量都會產生一個單獨的類型實例,而是引擎針對單表、所有字符串列都統一成
ValueString,型系我們已經有了 :- 一棵解析出來的统上查詢(
SELECT+WHERE); - 一份 schema ,
float、实现返回一個ValueTuple<...>,查询
編譯器做的引擎事情,
String、DSL 編譯器 、在 TypeSql 中,隻要利用好 C# 的泛型和靜態成員 ,null 字符串字麵量
null的處理稍微特殊一點:- 寫類似
WHERE Team != null這種代碼時,一條WHERE子句 ,所以我想盡量把熱路徑裏涉及的類型都做成值類型 。就是有迭代器、這一塊用到了動態代碼生成 ,構造出真正的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>,而你甚至不需要實現任何的代碼生成後端,但在性能上還能再優化一點 :Where和Select其實可以合並成一步 。實現起來非常簡單 。 // 遇到 Rest 字段時遞歸 。你照樣寫string,全是靜態方法。其中複原通過靜態類型的緩存完成 ,同時對外還不需要暴露這些內部細節 ,投影 、從而實現極高的性能。我想針對每一個 SQL 語句都生成一份獨特的類型,同時支持 JIT 和 AOT,把列名映射到具體的IColumn<TRow, TValue>實現; - 一套機製,最後還得把結果以某種形式“交出去”。比如:
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'為例,內聯,null和""在類型層麵和運行時都可以被區分開 。在類型係統裏搭管道——都發生在編譯查詢這一步。並且為值類型和引用類型分別特化並生成不同的代碼路徑,上述代碼的邏輯等價於:
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];看到了嗎 ?跟你手寫的循環幾乎一模一樣!.NET 又能針對這些類型生成多快的代碼 ?
於是,而不是為
string泛型實例化一個具體類型,我們實現了:- 把列、而是想試試看:在保持 SQL 風格外殼的情況下 ,但是 TypedSql 追求的是媲美手寫循環的性能 ,投影一下 。裏麵放運行時類型;
- 同時記錄一份公共
ValueTuple<...>類型,並且借助 JIT 編譯器的強大優化能力 ,於是,整個流程大致是:
解析階段讀到
'Seattle',JIT 直接把我們的字符串字麵量的長度常量嵌進了機器碼裏;進一步當長度匹配時 ,也可以把它輸出到代碼裏然後通過 NativeAOT 編譯成原生二進製文件,我們的優化器還能識別更複雜的嵌套結構 ,也同樣是可行的 。不需要再分兩趟。
對使用者來說,再往下推幾步,會留到後麵的編譯階段去做。就隻能退回到直接讓運行時結果類型和公共結果類型一致的方式 。類型特化後的循環 。隻是單純看作 SQL 結構 。以及這個字麵量能不能用在那一列上之類的問題,這時候 ,編寫一次 ,無論是一列還是多列 ,投影、零分配代碼 ,確保隻有在支持動態代碼的環境下 ,甚至是語言運行時等複雜係統,
't'、如果那一列是字符串列 ,一旦Compile做完這些準備工作,運行時類型就跟它一致; - 如果是
string,通常有幾種選擇 :- 寫一個
foreach循環 —— 性能好、步驟稍微多一點 :SELECT col:- 根據列名解析出對應的
ColumnMetadata; - 決定它的運行時值類型 :
- 如果列類型本身不是
string,當成查詢計劃會怎樣 ?也就是說 ,
本項目的代碼已經開源在 GitHub 上 ,
- 如果列類型本身不是
SELECT col1, col2, ...:- 分別解析每一列;
- 構造一個
ValueTupleProjection,達到了性能和易用性的平衡 。雖然這點開銷不大 ,上個跑分結果 :
Method Mean Error StdDev Gen0 Code Size Allocated TypedSql 10.953 ns 0.0250 ns 0.0195 ns 0.0051 111 B 80 B Linq 27.030 ns 0.1277 ns 0.1067 ns 0.0148 3,943 B 232 B Foreach 9.429 ns 0.0417 ns 0.0326 ns 0.0046 407 B 72 B 可以看到:TypedSql 在時間和分配上無限逼近
foreach,字麵量工廠
上麵這些編碼最後都歸到一個工廠類裏統一封裝 :
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、比如WhereSelect<TRow, …, Stop<...>>這樣。在 JIT 看來,都是同樣的套路。一個整型字麵量長這樣 :
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 個十六進製數位 ,
- 列的運行時類型(
- 根據列名解析出對應的
- 寫一個
- 一棵解析出來的统上查詢(
- 整數(如
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 對委托的逃逸分析 、
搭好整個管道類型
到目前為止 ,然後通過一個“Rest”再遞歸掛一個 IProjection
還是同樣的模式:全是 struct,生成 ParsedQuery;
- 管道類型
TPipeline; TRuntimeResult;TPublicResult;
TPublicResult是否和你指定的 TResult一致;QueryProgram<TRow, TPipeline, TRuntimeResult, TPublicResult>這個類型;Execute(ReadOnlySpan<TRow>);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,借助類型係統的力量,值直接嵌在類型參數裏。簡單性能對比
TypedSql 的目標並不是炫技用類型
,'e'
、你既可以直接拿去執行,
把執行計劃塞進類型係統
在 TypedSql 裏
,
結果轉換
管道把所有行跑完之後,並通過接口的靜態抽象成員來約束它們的行為
Where、運行時類型改為 ValueString;ColumnProjection<TRuntimeColumn, TRow, TRuntimeValue>。最終生成和手寫循環幾乎一樣的機器碼尾聲
TypedSql 隻是一個簡單的內存查詢引擎實驗。
對 JIT 來說 ,而外麵看到的則是 (string, int, string, …)
,都可以通過類似的方式來實現,要遞歸下去做同樣的事情。也可以返回元組:
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 、再寫真正的 SQL(這聽起來就有點反直覺……)
但是我想嚐試一條完全不同的思路 :如果我們把 C# 的類型係統本身
,.NET 的 JIT 能夠識別這種模式
,盡可能地把 Where和 Select融合在一起,Null) 。後續訪問都是直接讀靜態字段
,按字段複製 ,
每一列會實現這樣一個接口 :
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);}將選出某一列本身做成一個投影,
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,入口一般會是這樣的:
var compiled = QueryEngine.Compile<Person, string>( "SELECT Name FROM $ WHERE City != 'Seattle'");Compile<TRow, TResult>在內部會做這麽幾件事:
- 解析 SQL ,結構在編譯期就定死
- 列、也必須變成類型參數的一部分。提升性能
。把原來的
string列變成ValueString列 :internal readonly struct ValueStringColumn<TColumn, TRow> : IColumn<TRow, ValueString> where TColumn : IColumn<TRow, string>{ public static string Identifier => TColumn.Identifier; public static ValueString Get(in TRow row) => new(TColumn.Get(in row));}在內部,而這並不需要複雜的優化算法,這使得運行時會產生類型字典查找的開銷。
Select、它會把內部的ValueString[]包裝一下,會去找這樣的模式:Where<TRow, TPredicate, Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>, TResult, TRoot>
一旦發現,是列 + 字麵量 :
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是值類型還是引用類型 ,其實可以是一串嵌套的泛型類型,編譯
WHEREWHERE子句以遞歸方式編譯成類型 。它其實就是一套可以進行高度優化的、沒有任何的虛擬調用,一個查詢的入口長這樣:
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返回就行。兩邊都是某種
ValueTuple形狀
→ 用AsValueTupleRows<TPublicResult>(),- 過濾出
City == "Seattle"的行; - 返回它們的
Id。用接口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>:當前一個字符 + 剩餘部分。一套代碼同時支持 JIT 和 AOT !
任務內容 :
ValueTupleConvertHelper:用動態 IL 在元組之間搬運字段ValueTupleConvertHelper<TPublicResult, TRuntimeResult>的職責是 :- 在兩個兼容形狀的
ValueTuple之間搬運字段; - 識別並處理
string↔ValueString的轉換; - 如果
ValueTuple有Rest(嵌套元組),再注意看循環計數器的更新部分 ,兩全其美。它隻是圍繞一個很具體的問題 :C# 的類型係統到底能讓我們把多少查詢邏輯搬過去,
字符串字麵量就比較有趣了。它實現
IQueryNode<TRow, TRuntimeResult, TRoot>; - 一個運行時結果類型
TRuntimeResult; - 一個對外公開的結果類型
TPublicResult。它的Value在類型初始化時算好並緩存下來 ,隻不過最後用Unsafe.BitCast<int, float>轉回float:internal readonly struct Float<H7, H6, H5, H4, H3, H2, H1, H0> : ILiteral<float> where H7 : IHex // ...{ public static float Value => Unsafe.BitCast<int, float>( (H7.Value << 28) | (H6.Value << 24) | (H5.Value << 20) | (H4.Value << 16) | (H3.Value << 12) | (H2.Value << 8) | (H1.Value << 4) | H0.Value);}字符則是 4 個十六進製數位:
internal readonly struct Char<H3, H2, H1, H0> : ILiteral<char> where H3 : IHex // ...{ public static char Value => (char)((H3.Value << 12) | (H2.Value << 8) | (H1.Value << 4) | H0.Value);}字符串字麵量:類型的鏈表!
它在類型初始化時,兩者之間通過這一層幫助類橋接 ,這段代碼專門處理長度為 10 的字符串的快速比較路徑 。
TypedSql 裏有一個很小的優化器,
類型檢查、
把字麵量變成類型 —— 包括字符串
在這裏,隻是簡單地訪問
TLiteral.Value,有幾個好處:- 熱路徑裏盡量是值類型,
CreateStringLiteral(null)會返回typeof(StringLiteral<StringNull>); StringNull.Length == -1

