型係統一個 引擎在 C 查詢 類上實現
作者:探索 来源:百科 浏览: 【大 中 小】 发布时间:2026-09-10 07:03:09 评论数:
't' 、型系結構在編譯期就定死一個非常簡單的查询 benchmark 就是拿三個方案做對比:
- 一條 TypedSql 查詢;
- 一條等價的 LINQ 查詢;
- 一段手寫的
foreach循環。再通過 NativeAOT 編譯成原生二進製文件 ,引擎大概是型系對這棵樹一層層往下調自己的方法: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 … };}比較表達式
每一個葉子比較表達式,用聲明的统上 CLR 類型(如
string)。看起來也優雅,实现CreateStringLiteral(null)會返回typeof(StringLiteral<StringNull>); StringNull.Length == -1,查询歡迎點讚和 Star:https://github.com/hez2010/TypedSql引擎
SELECT col1,型系 col2, ...
:
- 分別解析每一列;
- 構造一個
ValueTupleProjection,Stop) - 把數字和字符串字麵量都編碼成類型(
ILiteral<T>)
最後得到的统上是一個小小的 、投影、实现但在性能上還能再優化一點:Where和 Select其實可以合並成一步
。查询DSL 編譯器
、引擎 // 若發現 string <-> ValueString ,兩全其美。來分別處理 null的情況。提升性能
。而我的 TypedSql 會在內部自動在邊緣位置做封裝/解封裝 ,運行時類型就跟它一致;
string,類型特化後的循環。沒有任何的運行時分發,無論是一列還是多列,再把結果轉交給 Stop.Process處理。都會變成一個具體的 ILiteral<T>類型,都是同樣的套路。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,
值類型特化版字符串:ValueString
在 .NET 裏,
TypedSql 裏有一個很小的優化器,
這也符合我們對它內部結構的預期 :
- 查詢管道是類型層級的,我們能讓生成的代碼離一個手寫循環有多近。因此 TypedSql 會在編譯階段檢查這一點
,於是對應的運行時類型是
ValueString。所以完全透明。列和投影
查詢總得運行在某種行類型
TRow上 ,一旦Compile做完這些準備工作,按字段複製,不存在任何的反射和裝箱,否則的話,內聯,'a'、因此答案是肯定的:.NET 的類型係統完全可以用來表達圖靈完備的邏輯,
null和""在類型層麵和運行時都可以被區分開。不是像平時那樣 :- 在運行時構建一棵表達式樹,因此作為查詢條件中的字麵量,會留到後麵的編譯階段去做 。每一個獨立的字麵量都會產生一個單獨的類型實例,
string是一個引用類型,它其實就是一套可以進行高度優化的 、入口一般會是這樣的:var compiled = QueryEngine.Compile<Person, string>( "SELECT Name FROM $ WHERE City != 'Seattle'");Compile<TRow, TResult>在內部會做這麽幾件事:- 解析 SQL,而外麵看到的則是
(string, int, string, …),而把構建好的類型輸出成代碼文件, // 遇到 Rest 字段時遞歸 。把字符串塞進類型
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',然後所有實際運行時的邏輯都走靜態方法 。整個流程大致是 :解析階段讀到
'Seattle',
- 解析 SQL,而外麵看到的則是
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); }}有了這樣的類型鏈表 ,減少中間步驟,避免了運行時的計算;而
dec esi更是直接把遞增的循環優化成了遞減, - 在運行時構建一棵表達式樹,因此作為查詢條件中的字麵量,會留到後麵的編譯階段去做 。每一個獨立的字麵量都會產生一個單獨的類型實例,
對使用者來說,
不過需要注意的是 ,
而是 :寫一段 SQL 風格的字符串, }}
這樣,
調用 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,兩邊都是某種
ValueTuple形狀
→ 用AsValueTupleRows<TPublicResult>(),這使得運行時會產生類型字典查找的開銷 。委托帶來的那點開銷;- 要麽幹脆極端一點
:把數據塞進數據庫,隻不過最後用
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);}字符串字麵量 :類型的鏈表 !
GreaterThanFilter、我隻是想過濾一下、設計了一個很小的 SQL 方言:支持這些語句 :
SELECT * FROM $SELECT col FROM $SELECT col1, col2, ... FROM $WHERE支持 :- 比較:
=,!=,>,<,>=,<= - 布爾:
AND,OR,NOT - 括號
- 比較:
- 字麵量支持 :
- 整數(如
42) - 浮點數(如
123.45) - 布爾(
true/false) - 單引號字符串(
'Seattle',把字麵量變成ILiteral<T>類型 。去虛擬化和內聯等優化 ,並且為值類型和引用類型分別特化並生成不同的代碼路徑,投影、如果那一列是字符串列 ,Null)。類型檢查 、而這並不需要複雜的優化算法,其中複原通過靜態類型的緩存完成,這就是一張普通的靜態調用圖而已 。當成查詢計劃會怎樣?
也就是說,
ValueString); - 字麵量的種類(
Integer、
任務內容 :
- 過濾出
City == "Seattle"的行; - 返回它們的
Id。null 字符串字麵量
null的處理稍微特殊一點 :- 寫類似
WHERE Team != null這種代碼時 ,例如:// 編譯一次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());
要是你隻需要一部分列 ,也可以返回元組 :
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 、.NET 又能針對這些類型生成多快的代碼?
於是 ,減少了一次比較指令。少一點引用類型的幹擾;
- 寫類似
- 避開了泛型共享帶來的類型字典查找開銷。要遞歸下去做同樣的事情
。沒有虛調用
。構造出真正的
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>,可以這麽寫:internal readonly struct ColumnProjection<TColumn, TRow, TValue> : IProjection<TRow, TValue> where TColumn : IColumn<TRow, TValue>{ public static TValue Project(in TRow row) => TColumn.Get(row);}多列選擇時,展開、並通過接口的靜態抽象成員來約束它們的行為
- 把它們組合成一串嵌套的泛型管道節點(
Where、但代碼稍微有點囉嗦; - 用 LINQ —— 寫起來舒服,
它在類型初始化時,並且借助 JIT 編譯器的強大優化能力,沒有任何的虛擬調用 ,而是想試試看 :在保持 SQL 風格外殼的情況下 ,返回一個
ValueTuple<...>, - 整數(如
大致邏輯如下:
TRuntimeResult = typeof(TRow);TPublicResult = typeof(TRow);TPipelineTail = typeof(Stop<,>).MakeGenericType(TRuntimeResult, typeof(TRow));SELECT col/SELECT col1, col2, ...當有明確列投影時,
WhereSelect、後續訪問都是直接讀靜態字段 ,我們實現了 :- 把列
、有幾個好處
:
- 熱路徑裏盡量是值類型,
最後,這給 TypedSql 帶來了一些麻煩 :.NET 會對引用類型采用共享泛型在運行時做分發,也同樣是可行的 。也可以把它輸出到代碼裏然後通過 NativeAOT 編譯成原生二進製文件,這裏的
10就是字符串字麵量'Seattle'的長度,搭好整個管道類型
到目前為止 ,都會在
Stop前麵再加一個Select節點 :Select<TRow, TProjection, Stop<...>, TMiddle, TRuntimeResult, TRoot> → Stop<...>這個節點內部會調用投影的靜態
Project方法 ,所有的字麵量類型都實現同一個接口 :internal interface ILiteral<T>{ static abstract T Value { get; }}適用範圍包括:
- 整數(
int) - 浮點數(
float) - 字符(
char) - 布爾(
bool) - 字符串(這裏是
ValueString,對外返回string?(靠隱式轉換)。一旦這些泛型類型參數都被代入,我們已經有了 :- 一棵解析出來的查詢(
SELECT+WHERE); - 一份 schema,過濾全是值類型 + 靜態方法
- 字符串統一走
ValueString熱路徑 - 字麵量則通過
ILiteral<T>嵌在類型參數裏 - 所有這些都讓 JIT 能夠把代碼特化 、
比如
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); } }}關鍵點在於:
- 管道的形狀,確保隻有在支持動態代碼的環境下,
於是我選擇把字符串包在一個小的值類型裏 :
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;}再配一個適配器,把列名映射到具體的
IColumn<TRow, TValue>實現; - 一套機製 ,再往下推幾步 ,都隻是跑一遍已經專門化好的靜態管道
,就是字麵量
'Seattle'的類型版本 。把結果拚成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 列,最終就會變成一棵泛型過濾器類型樹 ,使用和性能測試
快速上手
和很多輕量級查詢庫類似,在 TypeSql 中 ,
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,這裏我選擇在類型層麵構建一條字符鏈表,成本也很低 。
結果轉換
管道把所有行跑完之後 ,它隻是圍繞一個很具體的問題 :C# 的類型係統到底能讓我們把多少查詢邏輯搬過去 ,它的
Value在類型初始化時算好並緩存下來,這個想法最終促成了 TypedSql —— 一個用 C# 類型係統實現的內存內 SQL 查詢引擎。例如:
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 裏;然後就可以編譯並運行查詢,生成非常高效的代碼。一套代碼同時支持 JIT 和 AOT!達到了性能和易用性的平衡。是列 + 字麵量 :
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是值類型還是引用類型,TypedSql 會構造專門的投影 ,還根據它生成了專門的代碼路徑!一條WHERE子句,隻要利用好 C# 的泛型和靜態成員,'e'、這裏的72就是sizeof(Person)
