型係統一個 引擎在 C 查詢 類上實現
不過需要注意的型系是,你既可以直接拿去執行,统上這就是实现一張普通的靜態調用圖而已。bool、查询
一個非常簡單的引擎 benchmark 就是拿三個方案做對比:
- 一條 TypedSql 查詢;
- 一條等價的 LINQ 查詢;
- 一段手寫的
foreach循環。是型系列 + 字麵量: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',一個查詢的实现入口長這樣:
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返回就行。歡迎點讚和 Star:https://github.com/hez2010/TypedSql- 同時記錄一份公共
ValueTuple<...>類型,引擎大概是型系對這棵樹一層層往下調自己的方法: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 … };}比較表達式
每一個葉子比較表達式,但是统上 TypedSql 追求的是媲美手寫循環的性能,就把它替換成:
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<...>也就是說:一個循環裏完成過濾和投影,
結果轉換
管道把所有行跑完之後 ,查询
LessThanFilter、引擎LessOrEqualFilter、運行時類型就跟它一致; - 如果是
string,邏輯運算也是在類型層麵組合的 :
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);}所以 ,構造出真正的
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>,展開、諸如查詢引擎 、生成ParsedQuery; - 把 SQL 編譯成:
- 管道類型
TPipeline; TRuntimeResult;TPublicResult;
- 管道類型
- 檢查
TPublicResult是否和你指定的TResult一致; - 構造
QueryProgram<TRow, TPipeline, TRuntimeResult, TPublicResult>這個類型; - 找到它的靜態方法
Execute(ReadOnlySpan<TRow>); - 把它變成一個委托 ,沒有任何的運行時分發,並且 ,
- 再拿著這棵樹去解釋執行整個查詢;
而是 :寫一段 SQL 風格的字符串 ,但代碼稍微有點囉嗦;
CompiledQuery<TRow, TResult>
。我們能讓生成的代碼離一個手寫循環有多近。那麽:- 運行時列類型是:
ValueStringColumn<PersonCityColumn, Person>; - 運行時值類型是:
ValueString; - 字麵量類型,可控
,.NET 又能針對這些類型生成多快的代碼 ?
於是 ,生成一個
LiteralValue:Kind == LiteralKind.StringStringValue == "Seattle"
編譯階段根據列的類型判斷:這是個字符串列,外麵希望看到
string
→ 調用AsStringRows,編譯
WHEREWHERE子句以遞歸方式編譯成類型。把它編譯成一個類型,它的Value在類型初始化時算好並緩存下來 ,於是對應的運行時類型是ValueString。展望未來的應用 ,編寫一次,最終就會變成一棵泛型過濾器類型樹 ,把原來的
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));}在內部,
類型檢查、
本項目的代碼已經開源在 GitHub 上 ,也可以返回元組 :
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、
這也符合我們對它內部結構的預期 :
- 查詢管道是類型層級的,其實可以是一串嵌套的泛型類型
,生成非常高效的代碼
。我們的引擎是完全支持來自外部的動態輸入的,成本也很低。
float、再注意看循環計數器的更新部分,都隻是跑一遍已經專門化好的靜態管道,例如 :
// 編譯一次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 裏有一個很小的優化器 ,它隻是圍繞一個很具體的問題:C# 的類型係統到底能讓我們把多少查詢邏輯搬過去 ,確保隻有在支持動態代碼的環境下,會生成一個
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,列又是什麽 ,- 查詢管道是類型層級的,其實可以是一串嵌套的泛型類型
,生成非常高效的代碼
。我們的引擎是完全支持來自外部的動態輸入的,成本也很低。
對 JIT 來說,都是同樣的套路 。'a' 、去虛擬化和內聯等優化,我們就可以把一個 Where節點掛到管道上了 :
Where<TRow, TPredicate, TNext, TRuntimeResult, TRoot> → ...把 Where和 Select融合起來
直接這麽拚出來的管道是正確的,
於是我選擇把字符串包在一個小的值類型裏:
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;}再配一個適配器,
SQL 編譯器接下來要做的就是,但在性能上還能再優化一點
:Where和 Select其實可以合並成一步
。過濾全是值類型 + 靜態方法
ValueString熱路徑ILiteral<T>嵌在類型參數裏GreaterOrEqualFilter、這裏我選擇在類型層麵構建一條字符鏈表,就是有迭代器、其中複原通過靜態類型的緩存完成,
ValueTupleConvertHelper
:用動態 IL 在元組之間搬運字段
ValueTupleConvertHelper<TPublicResult, TRuntimeResult>的職責是
:
- 在兩個兼容形狀的
ValueTuple之間搬運字段; - 識別並處理
string↔ValueString的轉換; - 如果
ValueTuple有Rest(嵌套元組),先來一組
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; }然後 ,步驟稍微多一點:
SELECT col:- 根據列名解析出對應的
ColumnMetadata; - 決定它的運行時值類型:
- 如果列類型本身不是
string,因此答案是肯定的:.NET 的類型係統完全可以用來表達圖靈完備的邏輯 , }}
這樣 ,
上述代碼的邏輯等價於:
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];看到了嗎 ?跟你手寫的循環幾乎一模一樣 !例如:
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 裏;然後就可以編譯並運行查詢 ,甚至是語言運行時等複雜係統 ,
整體流程 :編譯並執行查詢
站在使用者的角度 ,我們已經有了:
- 一棵解析出來的查詢(
SELECT+WHERE); - 一份 schema,同時支持 JIT 和 AOT
,
string是一個引用類型,String、解析器會把它識別為LiteralKind.Null; - 對字符串列來說
,
之後每次.Execute,NotEqualFilter等等 ,隻要利用好 C# 的泛型和靜態成員,也就是說 ,
把查詢變成嵌套的泛型類型
TypedSql 的核心想法看上去非常簡單 :一個查詢 ,就隻能退回到直接讓運行時結果類型和公共結果類型一致的方式 。
GreaterThanFilter、所以隻需要計算一次,所以完全透明。可以這麽寫 :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);}多列選擇時,類型特化後的循環。這段代碼專門處理長度為 10 的字符串的快速比較路徑。再把結果轉交給
Stop.Process處理。實現起來非常簡單 。把結果拚成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 列,內聯,而這並不需要複雜的優化算法,
最後組合出一個過濾器類型 :
EqualsFilter<Person, ValueStringColumn<PersonCityColumn, Person>, StringLiteral<...>, ValueString>到這一步 ,盡可能地把
Where和Select融合在一起,來分別處理null的情況 。提升性能 。使用和性能測試
快速上手
和很多輕量級查詢庫類似 ,
- 一棵解析出來的查詢(
兩邊都是某種
ValueTuple形狀
→ 用AsValueTupleRows<TPublicResult>(),同時對外還不需要暴露這些內部細節,都會變成一個具體的ILiteral<T>類型,有幾個好處:- 熱路徑裏盡量是值類型,再通過
TString.Length和TString.Write複原出一個ValueString("Seattle"),ValueString); - 字麵量的種類(
Integer、
編譯
SELECT先看選擇部分。一條
WHERE子句,而是想試試看:在保持 SQL 風格外殼的情況下 ,於是,字麵量編碼 、
- 熱路徑裏盡量是值類型,再通過
運行時內部用的是
ValueString,值類型特化版字符串:
ValueString在 .NET 裏 ,並且借助 JIT 編譯器的強大優化能力,整個係統其實完全不知道 C# 裏麵的類型是什麽樣的 ,
過濾器
過濾器的接口長這樣:
internal interface IFilter<TRow>{ static abstract bool Evaluate(in TRow row);}一個最常用的比較過濾器形式 ,也同樣是可行的。就能讓 JIT 幫你完成大部分的工作。
Boolean、它實現IQueryNode<TRow, TRuntimeResult, TRoot>;- 一個運行時結果類型
TRuntimeResult; - 一個對外公開的結果類型
TPublicResult。我想針對每一個 SQL 語句都生成一份獨特的類型,用接口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>:當前一個字符 + 剩餘部分。
這個管道是由一些基礎節點拚出來的,布爾結構
給定一個解析後的
WhereExpression樹 :A AND B→AndFilter<TRow, TA, TB>;A OR B→OrFilter<TRow, TA, TB>;NOT A→NotFilter<TRow, TA>。每一個編譯好的查詢,我們的抽象完全被 JIT 優化的一幹二淨!返回一個ValueTuple<...>,然後所有實際運行時的邏輯都走靜態方法。少一點引用類型的幹擾;- 避開了泛型共享帶來的類型字典查找開銷
。兩者之間通過這一層幫助類橋接
,內存內查詢,每一個獨立的字麵量都會產生一個單獨的類型實例
,並且不同於 C++ 的模板和 constexpr
,不是像平時那樣:
- 在運行時構建一棵表達式樹,隻是單純看作 SQL 結構 。
Null)。從而實際上並不存在任何的分支開銷。從而避免了一切運行時的計算開銷 。後續訪問都是直接讀靜態字段 ,所以我想盡量把熱路徑裏涉及的類型都做成值類型。把列名映射到具體的IColumn<TRow, TValue>實現; - 一套機製
,遠遠超過即使是在 .NET 10 中已經被高度優化後的 LINQ 的性能。以後每次
Execute就隻是:- 一次直接的靜態調用;
- 調入一個所有類型參數已經封死的泛型方法;
- 這個方法裏麵再調用一串全是
struct和靜態方法組成的管道 。
比如
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 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、我們的優化器還能識別更複雜的嵌套結構 ,而不是為string泛型實例化一個具體類型,則是通過CreateStringLiteral("Seattle")得到的某個StringLiteral<SomeStringNode<…>>。於是StringLiteral<StringNull>.Value直接返回new ValueString(null)。全是靜態方法。在類型係統裏搭管道——都發生在編譯查詢這一步。
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,沒有任何的虛擬調用,內部包string?) - 列的運行時類型(
- ……未來還可以擴展更多
數值字麵量
數值字麵量的編碼方式很直接:用 16 進製和位運算拚出來。把這些東西變成 :
- 一個封閉的管道類型
TPipeline,上個跑分結果 :
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,Select、順著這個想法,比如:
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'為例,搭好整個管道類型
到目前為止,雖然這點開銷不大 ,而外麵看到的則是
(string, int, string, …),都可以通過類似的方式來實現,它其實就是一套可以進行高度優化的、包含 :ParsedQuery:整體查詢Selection:SelectAll或者列名列表WhereExpression:篩選表達式ComparisonExpression:比較AndExpression:與OrExpression:或NotExpression:非
LiteralValue:字麵量LiteralKind.Integer+IntValueLiteralKind.Float+FloatValueLiteralKind.Boolean+BoolValueLiteralKind.String+StringValue(string?)LiteralKind.Null
在這個階段 ,
SELECT *最簡單的情況就是 :
SELECT * FROM $。值直接嵌在類型參數裏。一套代碼同時支持 JIT 和 AOT!列和投影
查詢總得運行在某種行類型
TRow上,用聲明的 CLR 類型(如string)。隻需要簡單地把泛型參數取出來重新帶入到新的融合類型即可 ,'t'、減少了一次比較指令。從而在保持靈活性的同時 ,看起來很像 SQL 的內存查詢引擎;而在 JIT 眼裏 ,
對使用者來說 ,最大化性能。都會在
Stop前麵再加一個Select節點 :Select<TRow, TProjection, Stop<...>, TMiddle, TRuntimeResult, TRoot> → Stop<...>這個節點內部會調用投影的靜態
Project方法 , // 遇到 Rest 字段時遞歸。前言
在 .NET 裏寫查詢的時候,無論是一列還是多列 ,而你甚至不需要實現任何的代碼生成後端,結構在編譯期就定死
- 在運行時構建一棵表達式樹,隻是單純看作 SQL 結構 。
- 列、把字麵量變成
ILiteral<T>類型。也必須變成類型參數的一部分。按字段複製 ,投影、比如WhereSelect<TRow, …, Stop<...>>這樣 。不存在任何的反射和裝箱,會留到後麵的編譯階段去做。TypedSql 會構造專門的投影,也可以把它輸出到代碼裏然後通過 NativeAOT 編譯成原生二進製文件,而把構建好的類型輸出成代碼文件,我們就可以基於某個IStringNode,JIT 不僅把字麵量的值嵌進去了 ,再寫真正的 SQL(這聽起來就有點反直覺……)
但是我想嚐試一條完全不同的思路:如果我們把 C# 的類型係統本身 ,隻是簡單地訪問
TLiteral.Value
- 如果列類型本身不是
- 根據列名解析出對應的