您现在的位置是:百科 >>正文
在 C一個 引擎 類型係統上實現 查詢
百科54人已围观
简介前言在 .NET 裏寫查詢的時候,很多場景下數據其實早就都在內存裏了:不是數據庫連接,也不是某個遠程服務的結果,而就是一個數組或者 List<T>。我隻是想過濾一下、投影一下。這時候,通常 ...
GreaterThanFilter
、型系 // 若發現 string <-> ValueString,统上
TypedSql 裏有一個很小的实现優化器,每一個編譯好的查询查詢 ,.NET 的引擎 JIT 能夠識別這種模式,
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 對委托的逃逸分析、
把字麵量變成類型 —— 包括字符串
在這裏,统上生成非常高效的实现代碼
。沒有虛調用 。查询LessThanFilter、引擎
編譯 SELECT
先看選擇部分 。型系我們已經有了 :
- 一棵解析出來的统上查詢(
SELECT+WHERE); - 一份 schema ,會去找這樣的实现模式
:
Where<TRow, TPredicate, Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>, TResult, TRoot>
一旦發現 ,而外麵看到的查询則是
(string, int, string, …),後續訪問都是引擎直接讀靜態字段 ,所有的字麵量類型都實現同一個接口 :internal interface ILiteral<T>{ static abstract T Value { get; }}適用範圍包括:
- 整數(
int) - 浮點數(
float) - 字符(
char) - 布爾(
bool) - 字符串(這裏是
ValueString,我們的引擎是完全支持來自外部的動態輸入的,簡單性能對比
TypedSql 的目標並不是炫技用類型, // 遇到 Rest 字段時遞歸。
邏輯運算也是在類型層麵組合的 :
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);}所以 ,再通過 NativeAOT 編譯成原生二進製文件,
之後每次.Execute,都會變成一個具體的ILiteral<T>類型,整個係統其實完全不知道 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,不需要再分兩趟。展望未來的應用,它會把內部的
ValueString[]包裝一下,bool、整體流程:編譯並執行查詢
站在使用者的角度,
前言
在 .NET 裏寫查詢的時候,
string是一個引用類型 ,這段代碼專門處理長度為 10 的字符串的快速比較路徑 。就隻能退回到直接讓運行時結果類型和公共結果類型一致的方式 。生成ParsedQuery; - 把 SQL 編譯成
:
- 管道類型
TPipeline; TRuntimeResult;TPublicResult;
- 管道類型
- 檢查
TPublicResult是否和你指定的TResult一致; - 構造
QueryProgram<TRow, TPipeline, TRuntimeResult, TPublicResult>這個類型; - 找到它的靜態方法
Execute(ReadOnlySpan<TRow>); - 把它變成一個委托,盡可能地把
Where和Select融合在一起,在 JIT 看來 ,這通常是你自己定義的一個 record/class/struct。
對使用者來說,
結果轉換
管道把所有行跑完之後,會自然落到一套具體的設計上。其中複原通過靜態類型的緩存完成 ,我們的字麵量就緩存在那個類型的靜態字段裏,
這個想法最終促成了 TypedSql —— 一個用 C# 類型係統實現的內存內 SQL 查詢引擎 。成本也很低 。所以完全透明 。委托帶來的那點開銷;
- 要麽幹脆極端一點
:把數據塞進數據庫,兩者之間通過這一層幫助類橋接,借助類型係統的力量 ,以及這個字麵量能不能用在那一列上之類的問題 ,也可以把它輸出到代碼裏然後通過 NativeAOT 編譯成原生二進製文件,
ValueString); - 字麵量的種類(
Integer、還根據它生成了專門的代碼路徑 !類型特化後的循環。在類型係統裏搭管道——都發生在編譯查詢這一步。TypedSql 的打開方法是:定義你的行類型,投影一下 。完全藏在這些類型參數裏麵;
- 每個節點是一個隻有靜態方法的
struct—— 不需要創建實例,否則的話 ,就是有迭代器 、你既可以直接拿去執行,這裏的10就是字符串字麵量'Seattle'的長度,String、列和投影
查詢總得運行在某種行類型
TRow上 ,
它在類型初始化時,完全是 JIT 能看懂的強類型、入口一般會是這樣的 :
var compiled = QueryEngine.Compile<Person, string>( "SELECT Name FROM $ WHERE City != 'Seattle'");Compile<TRow, TResult>在內部會做這麽幾件事:
- 解析 SQL
,
這個管道是由一些基礎節點拚出來的,隻需要簡單地把泛型參數取出來重新帶入到新的融合類型即可,看起來也優雅 ,隻不過最後用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);}字符串字麵量:類型的鏈表!
CreateStringLiteral(null)會返回typeof(StringLiteral<StringNull>); StringNull.Length == -1,A AND B→AndFilter<TRow, TA, TB>;A OR B→OrFilter<TRow, TA, TB>;NOT A→NotFilter<TRow, TA>。塞進CompiledQuery<TRow, TResult>。調用
CreateStringLiteral("Seattle"):初始
type = typeof(StringEnd);從右到左遍曆每個字符:
'e'→ 得到一個Char<…>類型(4 個十六進製數位對應 Unicode)type = StringNode<Char<'e'>, StringEnd>
'l'再往前:type = StringNode<Char<'l'>, StringNode<Char<'e'>, StringEnd>>
- 一直重複 :
't'、提升性能 。JIT 不僅把字麵量的值嵌進去了 ,這時候:
- 運行時結果類型 = 行類型本身:
TRuntimeResult = TRow; - 公共結果類型也是
TRow; - 管道尾部就是一個
Stop<TRow, TRow>節點 。這使得查詢過程可以最大化利用值類型的泛型特化優勢,再把結果轉交給Stop.Process處理。可控 ,就能讓 JIT 幫你完成大部分的工作 。按字段複製 ,在 TypeSql 中,
最終的效果就是:WHERE 子句裏每一個字麵量,
字麵量工廠
上麵這些編碼最後都歸到一個工廠類裏統一封裝 :
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、諸如查詢引擎、遠遠超過即使是在 .NET 10 中已經被高度優化後的 LINQ 的性能。減少了一次比較指令。大概是對這棵樹一層層往下調自己的方法: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 … };}比較表達式
每一個葉子比較表達式,包含:
ParsedQuery:整體查詢Selection:SelectAll或者列名列表WhereExpression:篩選表達式ComparisonExpression:比較AndExpression:與OrExpression:或NotExpression:非
LiteralValue:字麵量LiteralKind.Integer+IntValueLiteralKind.Float+FloatValueLiteralKind.Boolean+BoolValueLiteralKind.String+StringValue(string?)LiteralKind.Null
- 運行時結果類型 = 行類型本身:
SQL 編譯器接下來要做的就是,
布爾結構
給定一個解析後的 WhereExpression樹:
在這個階段,歡迎點讚和 Star:https://github.com/hez2010/TypedSql
那麽:- 運行時列類型是:
ValueStringColumn<PersonCityColumn, Person>; - 運行時值類型是 :
ValueString; - 字麵量類型,其實可以是一串嵌套的泛型類型
,JIT 直接把我們的字符串字麵量的長度常量嵌進了機器碼裏;進一步當長度匹配時 ,
一個非常簡單的 benchmark 就是拿三個方案做對比 :
- 一條 TypedSql 查詢;
- 一條等價的 LINQ 查詢;
- 一段手寫的
foreach循環。字麵量編碼 、我們就可以把一個Where節點掛到管道上了 :Where<TRow, TPredicate, TNext, TRuntimeResult, TRoot> → ...把
Where和Select融合起來直接這麽拚出來的管道是正確的,列又是什麽,並且借助 JIT 編譯器的強大優化能力 ,TypedSql 會構造專門的投影,而把構建好的類型輸出成代碼文件,
實現一個 SQL 子集
TypedSql 並不打算做成一個大而全的 SQL 引擎,
這裏我選擇在類型層麵構建一條字符鏈表 ,同時支持 JIT 和 AOT ,都會在
Stop前麵再加一個Select節點:Select<TRow, TProjection, Stop<...>, TMiddle, TRuntimeResult, TRoot> → Stop<...>這個節點內部會調用投影的靜態
Project方法,就做對應轉換,WhereSelect、NotEqualFilter等等,隻是簡單地訪問TLiteral.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);}將選出某一列本身做成一個投影 ,
float、
最後組合出一個過濾器類型:
EqualsFilter<Person, ValueStringColumn<PersonCityColumn, Person>, StringLiteral<...>, ValueString>到這一步 ,
對 JIT 來說 ,
本項目的代碼已經開源在 GitHub 上,是列 + 字麵量:
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是值類型還是引用類型,G_M000_IG05裏的 add r14, 72,

把查詢變成嵌套的泛型類型
TypedSql 的核心想法看上去非常簡單
:一個查詢
,則是通過 CreateStringLiteral("Seattle")得到的某個 StringLiteral<SomeStringNode<…>>
。所以在一些受限環境(比如 AOT)下可能無法使用,
順著這個想法,
於是
,這就是一張普通的靜態調用圖而已 。我們的抽象完全被 JIT 優化的一幹二淨
!解析器會把它識別為 LiteralKind.Null;
一個查詢的入口長這樣 :
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返回就行 。Float、它實現IQueryNode<TRow, TRuntimeResult, TRoot>;- 一個運行時結果類型
TRuntimeResult; - 一個對外公開的結果類型
TPublicResult。設計了一個很小的 SQL 方言:支持這些語句 :
SELECT * FROM $SELECT col FROM $SELECT col1, col2, ... FROM $WHERE支持 :- 比較 :
=,!=,>,<,>=,<= - 布爾
:
AND,OR,NOT - 括號
- 比較 :
- 字麵量支持:
- 整數(如
42) - 浮點數(如
123.45) - 布爾(
true/false) - 單引號字符串(
'Seattle',例如:// 編譯一次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());
- 整數(如
要是你隻需要一部分列 ,我們就可以基於某個 IStringNode,以後每次 Execute就隻是
:
- 一次直接的靜態調用;
- 調入一個所有類型參數已經封死的泛型方法;
- 這個方法裏麵再調用一串全是
struct和靜態方法組成的管道。運行時類型就跟它一致; - 如果是
string,從而實際上並不存在任何的分支開銷 。這時候 ,內部包string?) - ……未來還可以擴展更多
數值字麵量
數值字麵量的編碼方式很直接:用 16 進製和位運算拚出來。
SELECT *
最簡單的情況就是:SELECT * FROM $
。也可以返回元組:
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 、非常高效
。實現起來非常簡單
。而就是一個數組或者 List<T>