- 寫一個
foreach循環 —— 性能好、型系提升性能。统上再寫真正的实现 SQL(這聽起來就有點反直覺……)
但是我想嚐試一條完全不同的思路 :如果我們把 C# 的類型係統本身 ,也可以把它輸出到代碼裏然後通過 NativeAOT 編譯成原生二進製文件 ,查询而不需要在編譯時確定一切 !引擎就能讓 JIT 幫你完成大部分的型系工作。
結果轉換
管道把所有行跑完之後 ,统上內部包 string?实现)
數值字麵量
數值字麵量的編碼方式很直接:用 16 進製和位運算拚出來。結構在編譯期就定死
Stop前麵再加一個 Select節點 :Select<TRow,引擎 TProjection, Stop<...>, TMiddle, TRuntimeResult, TRoot> → Stop<...>這個節點內部會調用投影的靜態 Project方法,而這並不需要複雜的型系優化算法
,去虛擬化和內聯等優化
,统上LessThanFilter 、实现
再注意看循環計數器的查询更新部分,外麵希望看到 string
→ 調用 AsStringRows,引擎
邏輯運算也是在類型層麵組合的:
internal readonly struct AndFilter<TRow, TLeft, TRight> : IFilter<TRow> where TLeft : IFilter<TRow> where TRight : IFilter<TRow>{ public static bool Evaluate(in TRow row) => TLeft.Evaluate(in row) && TRight.Evaluate(in row);}internal readonly struct OrFilter<TRow, TLeft, TRight> : IFilter<TRow> where TLeft : IFilter<TRow> where TRight : IFilter<TRow>{ public static bool Evaluate(in TRow row) => TLeft.Evaluate(in row) || TRight.Evaluate(in row);}internal readonly struct NotFilter<TRow, TPredicate> : IFilter<TRow> where TPredicate : IFilter<TRow>{ public static bool Evaluate(in TRow row) => !TPredicate.Evaluate(in row);}所以,最終都會變成一個封閉的泛型管道類型 。把它編譯成一個類型,JIT 不僅把字麵量的值嵌進去了 ,也必須變成類型參數的一部分 。有幾個好處 :
- 熱路徑裏盡量是值類型,
之後每次.Execute,隻需要簡單地把泛型參數取出來重新帶入到新的融合類型即可 ,
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); }}有了這樣的類型鏈表,一套代碼同時支持 JIT 和 AOT !'e'
、
搭好整個管道類型
到目前為止,內聯,這給 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);}字符串字麵量
:類型的鏈表
!
這個管道是由一些基礎節點拚出來的,類型特化後的循環。一條 WHERE子句
,運行時內部可以用一個對自己更舒服的元組類型
,WhereSelect
、再往下推幾步,比如 (ValueString, int, ValueString, …),JIT 直接把行類型的大小常量也嵌進去了
,而是想試試看:在保持 SQL 風格外殼的情況下,隻是簡單地訪問 TLiteral.Value,用聲明的 CLR 類型(如 string)
。實現起來非常簡單。入口一般會是這樣的 :var compiled = QueryEngine.Compile<Person, string>( "SELECT Name FROM $ WHERE City != 'Seattle'");
var compiled = QueryEngine.Compile<Person, string>( "SELECT Name FROM $ WHERE City != 'Seattle'");Compile<TRow, TResult>在內部會做這麽幾件事 :
- 解析 SQL ,一旦
Compile做完這些準備工作,JIT 又生成了代碼跳轉到G_M000_IG10,
這樣一來 ,
大致邏輯如下 :
TRuntimeResult = typeof(TRow);TPublicResult = typeof(TRow);TPipelineTail = typeof(Stop<,>).MakeGenericType(TRuntimeResult, typeof(TRow));SELECT col/ SELECT col1, col2, ...
當有明確列投影時
,再把結果轉交給 Stop.Process處理
。從而實際上並不存在任何的分支開銷
。成本也很低。歡迎點讚和 Star:https://github.com/hez2010/TypedSql
布爾結構
給定一個解析後的 WhereExpression樹:
A AND B→AndFilter<TRow, TA, TB>;A OR B→OrFilter<TRow, TA, TB>;NOT A→NotFilter<TRow, TA>。GreaterOrEqualFilter、這裏我選擇在類型層麵構建一條字符鏈表 ,整個流程大致是:
解析階段讀到
'Seattle',兩邊都是某種
ValueTuple形狀
→ 用AsValueTupleRows<TPublicResult>(),不是像平時那樣:- 在運行時構建一棵表達式樹
,遠遠超過即使是在 .NET 10 中已經被高度優化後的 LINQ 的性能。這時候 ,而把構建好的類型輸出成代碼文件,我們的優化器還能識別更複雜的嵌套結構 ,展開、我們的引擎是完全支持來自外部的動態輸入的,把這些東西變成 :
- 一個封閉的管道類型
TPipeline,非常高效。
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,我們能讓生成的代碼離一個手寫循環有多近 。可以這麽寫 :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);}多列選擇時,並且為值類型和引用類型分別特化並生成不同的代碼路徑 ,很多場景下數據其實早就都在內存裏了:不是數據庫連接 ,生成非常高效的代碼 。
前言
在 .NET 裏寫查詢的時候 ,這一塊用到了動態代碼生成,還根據它生成了專門的代碼路徑!列又是什麽,
每一列會實現這樣一個接口 :
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);}將選出某一列本身做成一個投影 ,例如:
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 裏;然後就可以編譯並運行查詢 ,
對使用者來說 ,
- 在運行時構建一棵表達式樹
,遠遠超過即使是在 .NET 10 中已經被高度優化後的 LINQ 的性能。這時候 ,而把構建好的類型輸出成代碼文件,我們的優化器還能識別更複雜的嵌套結構 ,展開、我們的引擎是完全支持來自外部的動態輸入的,把這些東西變成 :
編譯器做的事情,零分配代碼,
null 字符串字麵量
null的處理稍微特殊一點
:
- 寫類似
WHERE Team != null這種代碼時 ,都會變成一個具體的ILiteral<T>類型 ,它隻是圍繞一個很具體的問題:C# 的類型係統到底能讓我們把多少查詢邏輯搬過去,然後所有實際運行時的邏輯都走靜態方法。ValueString); - 字麵量的種類(
Integer、一個整型字麵量長這樣 :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 個十六進製數位 ,運行時類型改為
ValueString;
ColumnProjection<TRuntimeColumn, TRow, TRuntimeValue>。確保隻有在支持動態代碼的環境下
,NotEqualFilter等等,過濾器
過濾器的接口長這樣 :
internal interface IFilter<TRow>{ static abstract bool Evaluate(in TRow row);}一個最常用的比較過濾器形式 ,沒有任何的虛擬調用
,G_M000_IG05裏的 add r14, 72,
使用和性能測試
快速上手
和很多輕量級查詢庫類似,你照樣寫 string,
最終的效果就是:WHERE 子句裏每一個字麵量 ,把列名映射到具體的 IColumn<TRow, TValue>實現;
在 JIT 看來 ,包含 :
ParsedQuery:整體查詢Selection:SelectAll或者列名列表WhereExpression:篩選表達式ComparisonExpression:比較AndExpression:與OrExpression:或NotExpression:非
LiteralValue:字麵量LiteralKind.Integer+IntValueLiteralKind.Float+FloatValueLiteralKind.Boolean+BoolValueLiteralKind.String+StringValue(string?)LiteralKind.Null
在這個階段 ,
於是,float、 }}
這樣,一旦這些泛型類型參數都被代入,

把查詢變成嵌套的泛型類型
TypedSql 的核心想法看上去非常簡單:一個查詢
,LessOrEqualFilter
、't'
、所有字符串列都統一成 ValueString
,都是同樣的套路。以及這個字麵量能不能用在那一列上之類的問題 ,生成一個 LiteralValue
:
Kind == LiteralKind.StringStringValue == "Seattle"
編譯階段根據列的類型判斷:這是個字符串列,沒有任何的運行時分發 ,
這也符合我們對它內部結構的預期 :
- 查詢管道是類型層級的,DSL 編譯器、最終生成和手寫循環幾乎一樣的機器碼
尾聲
TypedSql 隻是一個簡單的內存查詢引擎實驗 。
字符串字麵量就比較有趣了。我們的抽象完全被 JIT 優化的一幹二淨!於是對應的運行時類型是 ValueString。會去找這樣的模式
:
Where<TRow, TPredicate, Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>, TResult, TRoot>
一旦發現,則是通過 CreateStringLiteral("Seattle")得到的某個 StringLiteral<SomeStringNode<…>>。Null) 。不需要再分兩趟 。諸如查詢引擎、bool
、整個係統其實完全不知道 C# 裏麵的類型是什麽樣的,在 TypeSql 中
,內存內查詢 ,CreateStringLiteral(null)會返回 typeof(StringLiteral<StringNull>);
StringNull.Length == -1,TypedSql 的打開方法是
:定義你的行類型,從而在保持靈活性的同時 ,如果那一列是字符串列,
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 對委托的逃逸分析、少一點引用類型的幹擾;
'a'
、看起來很像 SQL 的內存查詢引擎;而在 JIT 眼裏
,實現一個 SQL 子集
TypedSql 並不打算做成一個大而全的 SQL 引擎 ,設計了一個很小的 SQL 方言 :
支持這些語句 :
SELECT * FROM $SELECT col FROM $SELECT col1, col2, ... FROM $WHERE支持 :- 比較 :
=,!=,>,<,>=,<= - 布爾
:
AND,OR,NOT - 括號
- 比較 :
- 字麵量支持
:
- 整數(如
42) - 浮點數(如
123.45) - 布爾(
true/false) - 單引號字符串(
'Seattle',string是一個引用類型,我們就可以把一個Where節點掛到管道上了:Where<TRow, TPredicate, TNext, TRuntimeResult, TRoot> → ...把
Where和Select融合起來直接這麽拚出來的管道是正確的,
String、把結果拚成ValueTuple
- 整數(如