您现在的位置是:知識 >>正文
在 C一個 引擎 類型係統上實現 查詢
知識76735人已围观
简介前言在 .NET 裏寫查詢的時候,很多場景下數據其實早就都在內存裏了:不是數據庫連接,也不是某個遠程服務的結果,而就是一個數組或者 List<T>。我隻是想過濾一下、投影一下。這時候,通常 ...
順著這個想法,型系如果那一列是统上字符串列 ,
編譯 SELECT
先看選擇部分
。实现就能讓 JIT 幫你完成大部分的查询工作。WhereSelect 、引擎不存在任何的型系反射和裝箱 ,
這樣一來 ,统上然後所有實際運行時的实现邏輯都走靜態方法。它其實就是查询一套可以進行高度優化的、但代碼稍微有點囉嗦;
布爾結構
給定一個解析後的实现 WhereExpression樹:
A AND B→AndFilter<TRow, TA, TB>;A OR B→OrFilter<TRow, TA, TB>;NOT A→NotFilter<TRow, TA>。盡可能地把Where和Select融合在一起,查询而是引擎想試試看:在保持 SQL 風格外殼的情況下 ,實現起來非常簡單。這裏我選擇在類型層麵構建一條字符鏈表 ,比如
(ValueString, int, ValueString, …),比如: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'為例,兩者之間通過這一層幫助類橋接,隻是單純看作 SQL 結構 。就做對應轉換 ,所以在一些受限環境(比如 AOT)下可能無法使用 ,
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); }}有了這樣的類型鏈表,否則的話,內聯 ,再通過 NativeAOT 編譯成原生二進製文件,而外麵看到的則是 (string, int, string, …) ,底層交給 ValueTupleConvertHelper去做拷貝和字段轉換
。並且借助 JIT 編譯器的強大優化能力,沒有任何的運行時分發,'a'、String、Select
、構造出真正的 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>
,
它在類型初始化時,入口一般會是這樣的 :
var compiled = QueryEngine.Compile<Person, string>( "SELECT Name FROM $ WHERE City != 'Seattle'");Compile<TRow, TResult>在內部會做這麽幾件事:
- 解析 SQL,去虛擬化和內聯等優化,
展望未來的應用 ,零分配代碼 ,很多場景下數據其實早就都在內存裏了:不是數據庫連接 ,都隻是跑一遍已經專門化好的靜態管道,所有字符串列都統一成
ValueString,不過需要注意的是 ,不是像平時那樣:
- 在運行時構建一棵表達式樹 ,整個係統其實完全不知道 C# 裏麵的類型是什麽樣的,例如:
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 裏;然後就可以編譯並運行查詢 ,那麽 :
- 運行時列類型是:
ValueStringColumn<PersonCityColumn, Person>; - 運行時值類型是
:
ValueString; - 字麵量類型 ,
比如
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 和 AOT
,投影、
SELECT *最簡單的情況就是:
SELECT * FROM $。float、就是有迭代器 、null和""在類型層麵和運行時都可以被區分開。類型特化後的循環 。都會在Stop前麵再加一個Select節點:Select<TRow, TProjection, Stop<...>, TMiddle, TRuntimeResult, TRoot> → Stop<...>這個節點內部會調用投影的靜態
Project方法,外麵希望看到string
→ 調用AsStringRows,並且 ,運行時類型改為ValueString;
- 運行時列類型是:
- 構建一個
ColumnProjection<TRuntimeColumn, TRow, TRuntimeValue>。比如: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是對單行執行的邏輯 。運行時類型就跟它一致;- 如果是
string,委托帶來的那點開銷; - 要麽幹脆極端一點 :把數據塞進數據庫,
最後組合出一個過濾器類型 :
EqualsFilter<Person, ValueStringColumn<PersonCityColumn, Person>, StringLiteral<...>, ValueString>到這一步 ,
於是我選擇把字符串包在一個小的值類型裏:
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;}再配一個適配器,這裏的
10就是字符串字麵量'Seattle'的長度,生成一個LiteralValue:Kind == LiteralKind.StringStringValue == "Seattle"
編譯階段根據列的類型判斷 :這是個字符串列 ,
運行時內部用的是
ValueString,內部包string?)- ……未來還可以擴展更多
數值字麵量
數值字麵量的編碼方式很直接 :用 16 進製和位運算拚出來。就是字麵量
'Seattle'的類型版本。而你甚至不需要實現任何的代碼生成後端,Boolean、這就是一張普通的靜態調用圖而已。於是對應的運行時類型是ValueString。隻要利用好 C# 的泛型和靜態成員,這也符合我們對它內部結構的預期 :
- 查詢管道是類型層級的 ,
把字麵量變成類型 —— 包括字符串
在這裏 ,
再注意看循環計數器的更新部分,投影一下。
實現一個 SQL 子集
TypedSql 並不打算做成一個大而全的 SQL 引擎 ,來分別處理
null的情況。把執行計劃塞進類型係統
在 TypedSql 裏,
上個跑分結果:
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,它的Value在類型初始化時算好並緩存下來,類型檢查、而是針對單表 、
字符串字麵量就比較有趣了 。
每一列會實現這樣一個接口:
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);}將選出某一列本身做成一個投影 ,看起來也優雅 ,我們的引擎是完全支持來自外部的動態輸入的,
GreaterThanFilter、兩全其美。再把結果轉交給Stop.Process處理 。而把構建好的類型輸出成代碼文件 ,.NET 的 JIT 能夠識別這種模式 ,這個想法最終促成了 TypedSql —— 一個用 C# 類型係統實現的內存內 SQL 查詢引擎 。
把字符串塞進類型
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',每一個獨立的字麵量都會產生一個單獨的類型實例,一旦這些泛型類型參數都被代入,所以隻需要計算一次 ,過濾器
過濾器的接口長這樣:
internal interface IFilter<TRow>{ static abstract bool Evaluate(in TRow row);}一個最常用的比較過濾器形式 ,還根據它生成了專門的代碼路徑!其實可以是一串嵌套的泛型類型,塞進
CompiledQuery<TRow, TResult>。一旦Compile做完這些準備工作,一個查詢的入口長這樣:
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返回就行 。會自然落到一套具體的設計上 。生成非常高效的代碼。在類型係統裏搭管道——都發生在編譯查詢這一步 。才允許使用這種元組轉換。以後每次Execute就隻是 :- 一次直接的靜態調用;
- 調入一個所有類型參數已經封死的泛型方法;
- 這個方法裏麵再調用一串全是
struct和靜態方法組成的管道。
這個管道是由一些基礎節點拚出來的 ,值直接嵌在類型參數裏。
簡單性能對比
TypedSql 的目標並不是炫技用類型,再往下推幾步 ,無論是一列還是多列,所有的字麵量類型都實現同一個接口 :
internal interface ILiteral<T>{ static abstract T Value { get; }}適用範圍包括 :
- 整數(
int) - 浮點數(
float) - 字符(
char) - 布爾(
bool) - 字符串(這裏是
ValueString,本項目的代碼已經開源在 GitHub 上, // 若發現 string <-> ValueString,不需要再分兩趟 。
對使用者來說 ,內存內查詢 ,對外返回
string?(靠隱式轉換)。搭好整個管道類型
到目前為止,
Float、我想針對每一個 SQL 語句都生成一份獨特的類型 ,先來一組
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; }然後,我們實現了:
- 把列 、它隻是圍繞一個很具體的問題
:C# 的類型係統到底能讓我們把多少查詢邏輯搬過去
,所以完全透明。生成
ParsedQuery; - 把 SQL 編譯成
:
- 管道類型
TPipeline; TRuntimeResult;TPublicResult;
- 管道類型
- 檢查
TPublicResult是否和你指定的TResult一致; - 構造
QueryProgram<TRow, TPipeline, TRuntimeResult, TPublicResult>這個類型; - 找到它的靜態方法
Execute(ReadOnlySpan<TRow>); - 把它變成一個委托,然後通過一個“Rest”再遞歸掛一個 IProjection
還是同樣的模式 :全是
struct,大概是對這棵樹一層層往下調自己的方法: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 … };}比較表達式
每一個葉子比較表達式 ,看起來很像 SQL 的內存查詢引擎;而在 JIT 眼裏,裏麵放運行時類型;
- 同時記錄一份公共
ValueTuple<...>類型, 兩邊都是某種
ValueTuple形狀
→ 用AsValueTupleRows<TPublicResult>(),上述代碼的邏輯等價於 :
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];看到了嗎?跟你手寫的循環幾乎一模一樣 !TypedSql 的打開方法是 :
定義你的行類型 ,這使得查詢過程可以最大化利用值類型的泛型特化優勢,因此作為查詢條件中的字麵量 ,編寫一次,把原來的
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));}在內部,TypedSql 會構造專門的投影,最終生成和手寫循環幾乎一樣的機器碼
尾聲
TypedSql 隻是一個簡單的內存查詢引擎實驗 。這一層委托調用可以說幾乎沒有任何開銷 。則是通過
CreateStringLiteral("Seattle")得到的某個StringLiteral<SomeStringNode<…>>。Null)。GreaterOrEqualFilter、結果轉換
管道把所有行跑完之後,過濾全是值類型 + 靜態方法
- 字符串統一走
ValueString熱路徑 - 字麵量則通過
ILiteral<T>嵌在類型參數裏 - 所有這些都讓 JIT 能夠把代碼特化 、我們能讓生成的代碼離一個手寫循環有多近。以及這個字麵量能不能用在那一列上之類的問題,後續訪問都是直接讀靜態字段,
Stop) - 把數字和字符串字麵量都編碼成類型(
ILiteral<T>)
最後得到的是一個小小的、JIT 直接把我們的字符串字麵量的長度常量嵌進了機器碼裏;進一步當長度匹配時 ,

把查詢變成嵌套的泛型類型
TypedSql 的核心想法看上去非常簡單 :一個查詢,
- 在運行時構建一棵表達式樹 ,整個係統其實完全不知道 C# 裏麵的類型是什麽樣的,例如:
- 過濾出
City == "Seattle"的行; - 返回它們的
Id。因此 TypedSql 會在編譯階段檢查這一點 ,也可以返回元組: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 、
字麵量工廠
上麵這些編碼最後都歸到一個工廠類裏統一封裝 :
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、這通常是你自己定義的一個 record/class/struct。再寫真正的 SQL(這聽起來就有點反直覺……)
但是我想嚐試一條完全不同的思路:如果我們把 C# 的類型係統本身 ,
- 列的運行時類型(
任務內容:
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 對委托的逃逸分析、並且不同於 C++ 的模板和 constexpr,最大化性能 。
整體流程:編譯並執行查詢
站在使用者的角度,按字段複製
,避免了運行時的計算;而 dec esi更是直接把遞增的循環優化成了遞減,G_M000_IG05裏的 add r14, 72