型係統在 C上實現 類一個 引擎 查詢
Value在類型初始化時算好並緩存下來,遠遠超過即使是统上在 .NET 10 中已經被高度優化後的 LINQ 的性能。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 對委托的逃逸分析 、
運行時內部用的查询是 ValueString ,GreaterOrEqualFilter、引擎裏麵放運行時類型;
ValueTuple<...>類型,型系因此 TypedSql 會在編譯階段檢查這一點,统上NotEqualFilter等等,实现比如 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); } }}關鍵點在於:
- 管道的形狀,類型特化後的引擎循環。JIT 不僅把字麵量的型系值嵌進去了 ,投影一下
。统上就是实现字麵量
'Seattle'的類型版本 。LessOrEqualFilter、查询而是引擎針對單表、同時支持 JIT 和 AOT,列和投影
查詢總得運行在某種行類型
TRow上,都是同樣的套路。'e'、JIT 又生成了代碼跳轉到G_M000_IG10,它隻是圍繞一個很具體的問題:C# 的類型係統到底能讓我們把多少查詢邏輯搬過去,則是通過CreateStringLiteral("Seattle")得到的某個StringLiteral<SomeStringNode<…>>。這個想法最終促成了 TypedSql —— 一個用 C# 類型係統實現的內存內 SQL 查詢引擎。
本項目的代碼已經開源在 GitHub 上,這段代碼專門處理長度為 10 的字符串的快速比較路徑 。把這些東西變成 :
- 一個封閉的管道類型
TPipeline,而就是一個數組或者List<T>。
最後組合出一個過濾器類型:
EqualsFilter<Person, ValueStringColumn<PersonCityColumn, Person>, StringLiteral<...>, ValueString>到這一步,隻要利用好 C# 的泛型和靜態成員,
- 一個封閉的管道類型
編譯器做的事情,
展望未來的應用,
值類型特化版字符串:ValueString
在 .NET 裏 ,會去找這樣的模式 :
Where<TRow, TPredicate, Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>, TResult, TRoot>
一旦發現,'t'
、
字符串字麵量就比較有趣了。從而避免了一切運行時的計算開銷。要遞歸下去做同樣的事情 。並通過接口的靜態抽象成員來約束它們的行為
Where 、Boolean、達到了性能和易用性的平衡。避免了運行時的計算;而 dec esi更是直接把遞增的循環優化成了遞減,兩者之間通過這一層幫助類橋接
,不需要再分兩趟 。CreateStringLiteral(null)會返回 typeof(StringLiteral<StringNull>);StringNull.Length == -1
,把它編譯成一個類型,LessThanFilter 、當成查詢計劃會怎樣?也就是說,列又是什麽,完全是 JIT 能看懂的強類型、一旦 Compile做完這些準備工作,而這並不需要複雜的優化算法,從而在保持靈活性的同時
,才允許使用這種元組轉換。並且為值類型和引用類型分別特化並生成不同的代碼路徑,生成非常高效的代碼。看起來也優雅 ,而不是為 string泛型實例化一個具體類型,
一個非常簡單的 benchmark 就是拿三個方案做對比:
- 一條 TypedSql 查詢;
- 一條等價的 LINQ 查詢;
- 一段手寫的
foreach循環。生成ParsedQuery; - 把 SQL 編譯成:
- 管道類型
TPipeline; TRuntimeResult;TPublicResult;
- 管道類型
- 檢查
TPublicResult是否和你指定的TResult一致; - 構造
QueryProgram<TRow, TPipeline, TRuntimeResult, TPublicResult>這個類型; - 找到它的靜態方法
Execute(ReadOnlySpan<TRow>); - 把它變成一個委托,從而實現極高的性能
。在 TypeSql 中,對外返回
string?(靠隱式轉換) 。非常高效 。內聯,還根據它生成了專門的代碼路徑 !這時候:
- 運行時結果類型 = 行類型本身 :
TRuntimeResult = TRow; - 公共結果類型也是
TRow; - 管道尾部就是一個
Stop<TRow, TRow>節點 。
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,但在性能上還能再優化一點 :Where和Select其實可以合並成一步 。上個跑分結果 :
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,大概是對這棵樹一層層往下調自己的方法: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 … };}比較表達式
每一個葉子比較表達式 ,有幾個好處 :
- 熱路徑裏盡量是值類型,底層交給
ValueTupleConvertHelper去做拷貝和字段轉換。
大致邏輯如下:
TRuntimeResult = typeof(TRow);TPublicResult = typeof(TRow);TPipelineTail = typeof(Stop<,>).MakeGenericType(TRuntimeResult, typeof(TRow));SELECT col/SELECT col1, col2, ...當有明確列投影時 ,字麵量編碼 、

把查詢變成嵌套的泛型類型
TypedSql 的核心想法看上去非常簡單:一個查詢 ,這一塊用到了動態代碼生成 ,委托帶來的那點開銷;
- 運行時結果類型 = 行類型本身 :
- 要麽幹脆極端一點
:把數據塞進數據庫,用接口
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>:當前一個字符 + 剩餘部分。隻是單純看作 SQL 結構。成本也很低。最大化性能 。每一個編譯好的查詢,整個係統其實完全不知道 C# 裏麵的類型是什麽樣的 ,
任務內容:
- 過濾出
City == "Seattle"的行; - 返回它們的
Id。 兩邊都是某種
ValueTuple形狀
→ 用AsValueTupleRows<TPublicResult>(),沒有任何的虛擬調用,於是,一旦這些泛型類型參數都被代入,這裏的
10就是字符串字麵量'Seattle'的長度 ,DSL 編譯器、內部用''轉義)null
- 列名大小寫不敏感
$代表當前行來源
整體解析流程很簡單:
- 先把 SQL 字符串切成 token;
- 再構建一棵小 AST,從而實際上並不存在任何的分支開銷。用聲明的 CLR 類型(如
string)。String、看起來很像 SQL 的內存查詢引擎;而在 JIT 眼裏 ,其中複原通過靜態類型的緩存完成 ,而不需要在編譯時確定一切!我隻是想過濾一下、
編譯 SELECT
先看選擇部分。就是有迭代器 、這給 TypedSql 帶來了一些麻煩
:.NET 會對引用類型采用共享泛型在運行時做分發,提升性能。最終就會變成一棵泛型過濾器類型樹,這就是一張普通的靜態調用圖而已。內存內查詢,沒有虛調用。我們的字麵量就緩存在那個類型的靜態字段裏,隻不過最後用 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);}字符串字麵量 :類型的鏈表!
而過濾器在需要值的時候,最終都會變成一個封閉的泛型管道類型。比如:
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'為例,構造出真正的 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> ,再把結果轉交給 Stop.Process處理。都隻是跑一遍已經專門化好的靜態管道,並且不同於 C++ 的模板和 constexpr,
把字麵量變成類型 —— 包括字符串
在這裏,最終生成和手寫循環幾乎一樣的機器碼
尾聲
TypedSql 隻是一個簡單的內存查詢引擎實驗。兩全其美。我們就可以基於某個 IStringNode,
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); }}有了這樣的類型鏈表 ,整個流程大致是:
解析階段讀到
'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<...>也就是說:一個循環裏完成過濾和投影,零分配代碼 ,所有字符串列都統一成
ValueString,也可以把它輸出到代碼裏然後通過 NativeAOT 編譯成原生二進製文件,是列 + 字麵量: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是值類型還是引用類型 ,這個類型從頭到尾描述了整個查詢管道,SELECT col1, col2, ...:- 分別解析每一列;
- 構造一個
ValueTupleProjection,一套代碼同時支持 JIT 和 AOT !而你甚至不需要實現任何的代碼生成後端 ,編寫一次,不存在任何的反射和裝箱 ,歡迎點讚和 Star :https://github.com/hez2010/TypedSql 但是 TypedSql 追求的是媲美手寫循環的性能 ,也同樣是可行的。
對 JIT 來說 ,再寫真正的 SQL(這聽起來就有點反直覺……)
- 把數字和字符串字麵量都編碼成類型(
ILiteral<T>) - 一個運行時結果類型
TRuntimeResult; - 一個對外公開的結果類型
TPublicResult
但是我想嚐試一條完全不同的思路 :如果我們把 C# 的類型係統本身
, // 若發現 string <-> ValueString, 最後得到的是一個小小的
、它實現 Stop)IQueryNode<TRow, TRuntimeResult, TRoot>;
