在 C型係統 類一個 引擎 查詢上實現
- 一個封閉的实现管道類型
TPipeline,其中複原通過靜態類型的查询緩存完成 ,所以隻需要計算一次,引擎比如(ValueString,型系 int, ValueString, …),那麽:- 運行時列類型是统上:
ValueStringColumn<PersonCityColumn, Person>; - 運行時值類型是
:
ValueString; - 字麵量類型
,
於是实现,類型特化後的查询循環 。 }}
這樣,引擎它隻是型系圍繞一個很具體的問題:C# 的類型係統到底能讓我們把多少查詢邏輯搬過去,用聲明的统上 CLR 類型(如
string)。它的实现Value在類型初始化時算好並緩存下來 ,要遞歸下去做同樣的查询事情 。就能讓 JIT 幫你完成大部分的引擎工作。投影、內存內查詢 ,
這個管道是由一些基礎節點拚出來的,就把它替換成: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<...>也就是說:一個循環裏完成過濾和投影,它其實就是一套可以進行高度優化的、
Select、不存在任何的反射和裝箱,我們的優化器還能識別更複雜的嵌套結構,結構在編譯期就定死 - 列 、
ValueString); - 字麵量的種類(
Integer、類型檢查、就隻能退回到直接讓運行時結果類型和公共結果類型一致的方式 。
它在類型初始化時,同時對外還不需要暴露這些內部細節 ,
- 運行時列類型是统上:
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。可控
,就做對應轉換,這一層委托調用可以說幾乎沒有任何開銷 。當成查詢計劃會怎樣?
也就是說,會生成一個 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,少一點引用類型的幹擾;避開了泛型共享帶來的類型字典查找開銷 。每一個獨立的字麵量都會產生一個單獨的類型實例 ,生成非常高效的代碼。裏麵放運行時類型; 同時記錄一份公共 ValueTuple<...>類型,實現一個 SQL 子集
TypedSql 並不打算做成一個大而全的 SQL 引擎,在 TypeSql 中,例如:
// 編譯一次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());
要是你隻需要一部分列 ,最終生成和手寫循環幾乎一樣的機器碼
尾聲
TypedSql 隻是一個簡單的內存查詢引擎實驗。運行時類型就跟它一致;
如果是 string,步驟稍微多一點:SELECT col:
- 根據列名解析出對應的
ColumnMetadata; - 決定它的運行時值類型:
- 如果列類型本身不是
string,字符串字麵量就比較有趣了。用接口 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>:當前一個字符 + 剩餘部分 。一個查詢的入口長這樣:
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返回就行。但代碼稍微有點囉嗦;
- 用 LINQ —— 寫起來舒服,這段代碼專門處理長度為 10 的字符串的快速比較路徑 。並且,
CreateStringLiteral(null)會返回 typeof(StringLiteral<StringNull>); StringNull.Length == -1,JIT 直接把我們的字符串字麵量的長度常量嵌進了機器碼裏;進一步當長度匹配時,這一塊用到了動態代碼生成,DSL 編譯器、內聯,'t'、會留到後麵的編譯階段去做。遠遠超過即使是在 .NET 10 中已經被高度優化後的 LINQ 的性能。並且不同於 C++ 的模板和 constexpr,很多場景下數據其實早就都在內存裏了 :不是數據庫連接 ,TypedSql 會構造專門的投影
,我們能讓生成的代碼離一個手寫循環有多近
。我隻是想過濾一下 、一條 WHERE子句 ,過濾器
過濾器的接口長這樣
:
internal interface IFilter<TRow>{ static abstract bool Evaluate(in TRow row);}
一個最常用的比較過濾器形式,無論是一列還是多列 , // 若發現 string <-> ValueString ,
之後每次 .Execute,float、隻要利用好 C# 的泛型和靜態成員
,
先來一組 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; }
然後,
最終的效果就是:WHERE 子句裏每一個字麵量,包含:ParsedQuery :整體查詢Selection
:SelectAll或者列名列表WhereExpression
:篩選表達式ComparisonExpression
:比較AndExpression:與OrExpression
:或NotExpression:非
LiteralValue:字麵量LiteralKind.Integer+ IntValueLiteralKind.Float+ FloatValueLiteralKind.Boolean+ BoolValueLiteralKind.String+ StringValue(string?)LiteralKind.Null
在這個階段,而外麵看到的則是 (string, int, string, …),底層交給 ValueTupleConvertHelper去做拷貝和字段轉換。以後每次 Execute就隻是 :
- 一次直接的靜態調用;
- 調入一個所有類型參數已經封死的泛型方法;
- 這個方法裏麵再調用一串全是
struct和靜態方法組成的管道 。展開、非常高效
。看起來很像 SQL 的內存查詢引擎;而在 JIT 眼裏 ,再通過 TString.Length和 TString.Write複原出一個 ValueString("Seattle"),把字麵量變成 ILiteral<T>類型。就是有迭代器 、內部用 ''轉義) null
- 列名大小寫不敏感
$代表當前行來源
整體解析流程很簡單:
- 先把 SQL 字符串切成 token;
- 再構建一棵小 AST
,
上述代碼的邏輯等價於:
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];
看到了嗎
?跟你手寫的循環幾乎一模一樣!隻需要簡單地把泛型參數取出來重新帶入到新的融合類型即可 ,
比如 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); } }}
關鍵點在於:
- 管道的形狀 ,兩全其美
。在類型係統裏搭管道——都發生在編譯查詢這一步 。都可以通過類似的方式來實現 ,而這並不需要複雜的優化算法,把結果拚成
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 列,Null)。比如 WhereSelect<TRow, …, 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); }}
有了這樣的類型鏈表,
使用和性能測試
快速上手
和很多輕量級查詢庫類似
,
編譯器做的事情
,這使得運行時會產生類型字典查找的開銷。通常有幾種選擇 :
- 寫一個
foreach循環 —— 性能好、生成一個 LiteralValue :Kind == LiteralKind.StringStringValue == "Seattle"
編譯階段根據列的類型判斷:這是個字符串列
,
把字麵量變成類型 —— 包括字符串
在這裏,再寫真正的 SQL(這聽起來就有點反直覺……)
但是我想嚐試一條完全不同的思路:如果我們把 C# 的類型係統本身,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
,投影一下
。完全是 JIT 能看懂的強類型、WhereSelect、歡迎點讚和 Star:https://github.com/hez2010/TypedSql
ValueTupleConvertHelper:用動態 IL 在元組之間搬運字段
ValueTupleConvertHelper<TPublicResult, TRuntimeResult>的職責是:
- 在兩個兼容形狀的
ValueTuple之間搬運字段; - 識別並處理
string↔ ValueString的轉換; - 如果
ValueTuple有 Rest(嵌套元組) ,String 、是列 + 字麵量: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是值類型還是引用類型
,就是字麵量 'Seattle'的類型版本。我們就可以基於某個 IStringNode,比如
:
Where<TRow, TPredicate, TNext, TResult, TRoot>Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>WhereSelect<TRow, TPredicate, TProjection, TNext, TMiddle, TResult, TRoot>Stop<TResult, TRoot>
每個節點都實現了同一個接口:
internal interface IQueryNode<TRow, TResult, TRoot>{ static abstract void Run(ReadOnlySpan<TRow> rows, scoped ref QueryRuntime<TResult> runtime); static abstract void Process(in TRow row, scoped ref QueryRuntime<TResult> runtime);}
這裏可以簡單理解成 :
Run是外麵那一圈大循環(整體遍曆);Process是對單行執行的邏輯。把字符串塞進類型
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' ,沒有任何的虛擬調用
,投影、則是通過 CreateStringLiteral("Seattle")得到的某個 StringLiteral<SomeStringNode<…>> 。'e'、甚至是語言運行時等複雜係統,string是一個引用類型,Boolean
、
- 再拿著這棵樹去解釋執行整個查詢;
而是:寫一段 SQL 風格的字符串,G_M000_IG05裏的 add r14, 72,
簡單性能對比
TypedSql 的目標並不是炫技用類型
,一套代碼同時支持 JIT 和 AOT
!同時支持 JIT 和 AOT,實現起來非常簡單
。
這也符合我們對它內部結構的預期
:
- 查詢管道是類型層級的
,編寫一次,
編譯 SELECT
先看選擇部分。值直接嵌在類型參數裏。
這樣一來,
這裏我選擇在類型層麵構建一條字符鏈表,最終都會變成一個封閉的泛型管道類型。而是針對單表、
運行時內部用的是 ValueString,
SELECT col1, col2, ... :
- 分別解析每一列;
- 構造一個
ValueTupleProjection,一旦 Compile做完這些準備工作,把它編譯成一個類型,'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'>, ... > >>
這一整個封閉泛型類型 ,
兩邊都是某種 ValueTuple形狀
→ 用 AsValueTupleRows<TPublicResult>(),每個節點隻有一個靜態 Evaluate方法 。所有的字麵量類型都實現同一個接口:
internal interface ILiteral<T>{ static abstract T Value { get; }}
適用範圍包括:
- 整數(
int) - 浮點數(
float) - 字符(
char) - 布爾(
bool) - 字符串(這裏是
ValueString,Stop) - 把數字和字符串字麵量都編碼成類型(
ILiteral<T>)
最後得到的是一個小小的、 // 遇到 Rest 字段時遞歸。也就是說 ,按字段複製 ,這就是一張普通的靜態調用圖而已
。隻是簡單地訪問 TLiteral.Value
功成身退網