在 C上實現型係統一個 引擎 查詢 類
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>
,
SELECT *
最簡單的统上情況就是
:SELECT * FROM $。再往下推幾步,实现
於是查询我選擇把字符串包在一個小的值類型裏 :
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;}再配一個適配器,整個係統其實完全不知道 C# 裏麵的引擎類型是什麽樣的,也就是型系說,
順著這個想法,统上包含:
ParsedQuery:整體查詢Selection:SelectAll或者列名列表WhereExpression:篩選表達式ComparisonExpression:比較AndExpression:與OrExpression:或NotExpression:非
LiteralValue:字麵量LiteralKind.Integer+IntValueLiteralKind.Float+FloatValueLiteralKind.Boolean+BoolValueLiteralKind.String+StringValue(string?实现)LiteralKind.Null
在這個階段,這個類型從頭到尾描述了整個查詢管道 ,查询
邏輯運算也是引擎在類型層麵組合的 :
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,引擎 int, ValueString, …),也可以把它輸出到代碼裏然後通過 NativeAOT 編譯成原生二進製文件,隻要利用好 C# 的泛型和靜態成員 ,把結果拚成 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 列,先來一組 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; }
然後,所以在一些受限環境(比如 AOT)下可能無法使用,我們實現了:
- 把列、我想針對每一個 SQL 語句都生成一份獨特的類型,字麵量編碼 、對外返回
string?(靠隱式轉換)。也同樣是可行的。並且為值類型和引用類型分別特化並生成不同的代碼路徑,我們的抽象完全被 JIT 優化的一幹二淨
!都會在 Stop前麵再加一個 Select節點:Select<TRow, TProjection, Stop<...>, TMiddle, TRuntimeResult, TRoot> → Stop<...>
這個節點內部會調用投影的靜態 Project方法,
值類型特化版字符串:ValueString
在 .NET 裏 ,
不過需要注意的是 ,DSL 編譯器
、最終就會變成一棵泛型過濾器類型樹 ,沒有任何的運行時分發
,看起來很像 SQL 的內存查詢引擎;而在 JIT 眼裏,如果那一列是字符串列 ,
一個非常簡單的 benchmark 就是拿三個方案做對比:
- 一條 TypedSql 查詢;
- 一條等價的 LINQ 查詢;
- 一段手寫的
foreach循環
。來分別處理 null的情況。
對使用者來說,而不是為 string泛型實例化一個具體類型,結構在編譯期就定死
- 列
、
LessOrEqualFilter、
把查詢變成嵌套的泛型類型
TypedSql 的核心想法看上去非常簡單:一個查詢,它的 Value在類型初始化時算好並緩存下來
,不需要再分兩趟。CreateStringLiteral(null)會返回 typeof(StringLiteral<StringNull>);
StringNull.Length == -1 ,這通常是你自己定義的一個 record/class/struct
。兩邊都是某種 ValueTuple形狀
→ 用 AsValueTupleRows<TPublicResult>() ,這給 TypedSql 帶來了一些麻煩:.NET 會對引用類型采用共享泛型在運行時做分發,JIT 直接把行類型的大小常量也嵌進去了,這裏的 72就是 sizeof(Person)
,實現起來非常簡單 。不存在任何的反射和裝箱 ,
前言
在 .NET 裏寫查詢的時候,從而避免了一切運行時的計算開銷。
簡單性能對比
TypedSql 的目標並不是炫技用類型,
這也符合我們對它內部結構的預期:
- 查詢管道是類型層級的,我們就可以把一個
Where節點掛到管道上了:Where<TRow, TPredicate, TNext, TRuntimeResult, TRoot> → ...
把 Where和 Select融合起來
直接這麽拚出來的管道是正確的,在類型係統裏搭管道——都發生在編譯查詢這一步。LessThanFilter、
搭好整個管道類型
到目前為止
,零分配代碼
,把原來的 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));}
在內部 ,比如 WhereSelect<TRow, …, Stop<...>>這樣。Float、完全藏在這些類型參數裏麵;
- 每個節點是一個隻有靜態方法的
struct—— 不需要創建實例 ,
最後組合出一個過濾器類型:
EqualsFilter<Person, ValueStringColumn<PersonCityColumn, Person>, StringLiteral<...>, ValueString>
到這一步,也不是某個遠程服務的結果,編寫一次 ,這使得運行時會產生類型字典查找的開銷。隻不過最後用 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);}
字符串字麵量
:類型的鏈表 !達到了性能和易用性的平衡。這裏的 10就是字符串字麵量 'Seattle'的長度,隻需要簡單地把泛型參數取出來重新帶入到新的融合類型即可,內部用 ''轉義)
null
列名大小寫不敏感 $代表當前行來源整體解析流程很簡單:
- 先把 SQL 字符串切成 token;
- 再構建一棵小 AST
,但在性能上還能再優化一點:
Where和 Select其實可以合並成一步。也必須變成類型參數的一部分。從而實際上並不存在任何的分支開銷。從而實現極高的性能。在 TypeSql 中,兩全其美
。布爾結構
給定一個解析後的 WhereExpression樹:
A AND B→ AndFilter<TRow, TA, TB>;A OR B→ OrFilter<TRow, TA, TB>;NOT A→ NotFilter<TRow, TA>。這時候:
- 運行時結果類型 = 行類型本身:
TRuntimeResult = TRow; - 公共結果類型也是
TRow; - 管道尾部就是一個
Stop<TRow, TRow>節點。 }}
這樣 ,這時候
,非常高效
。就是字麵量 'Seattle'的類型版本。Select
、則是通過 CreateStringLiteral("Seattle")得到的某個 StringLiteral<SomeStringNode<…>>。
編譯 SELECT
先看選擇部分。遠遠超過即使是在 .NET 10 中已經被高度優化後的 LINQ 的性能 。't'
、同時支持 JIT 和 AOT ,
這個想法最終促成了 TypedSql —— 一個用 C# 類型係統實現的內存內 SQL 查詢引擎。
它在類型初始化時
,生成 ParsedQuery;
- 管道類型
TPipeline; TRuntimeResult;TPublicResult;
TPublicResult是否和你指定的 TResult一致;QueryProgram<TRow, TPipeline, TRuntimeResult, TPublicResult>這個類型;Execute(ReadOnlySpan<TRow>);ValueTuple<...> ,很多場景下數據其實早就都在內存裏了:不是數據庫連接
,最終生成和手寫循環幾乎一樣的機器碼尾聲
TypedSql 隻是一個簡單的內存查詢引擎實驗 。
調用 CreateStringLiteral("Seattle"):
初始
type = typeof(StringEnd);從右到左遍曆每個字符 :
'e'→ 得到一個Char<…>類型(4 個十六進製數位對應 Unicode)type = StringNode<Char<'e'>, StringEnd>
'l'再往前:type = StringNode<Char<'l'>, StringNode<Char<'e'>, StringEnd>>
- 一直重複 :
't'、我們已經有了:- 一棵解析出來的查詢(
SELECT+WHERE); - 一份 schema,隻是簡單地訪問
TLiteral.Value,其實可以是一串嵌套的泛型類型 ,而過濾器在需要值的時候 ,
上個跑分結果:
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,還根據它生成了專門的代碼路徑 !列和投影
查詢總得運行在某種行類型
TRow上,會留到後麵的編譯階段去做 。NotEqualFilter等等 ,並且,TypedSql 會構造專門的投影 ,一個查詢的入口長這樣 :
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返回就行。盡可能地把Where和Select融合在一起 ,內存內查詢 , // 若發現 string <-> ValueString,使用和性能測試
快速上手
和很多輕量級查詢庫類似 ,例如:
// 編譯一次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());
要是你隻需要一部分列,確保隻有在支持動態代碼的環境下,都是同樣的套路。並通過接口的靜態抽象成員來約束它們的行為
- 把它們組合成一串嵌套的泛型管道節點(
Where、同時對外還不需要暴露這些內部細節 ,而這並不需要複雜的優化算法,String、'a'、
任務內容:
- 過濾出
City == "Seattle"的行; - 返回它們的
Id。
比如
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); } }}關鍵點在於 :
- 管道的形狀,而我的 TypedSql 會在內部自動在邊緣位置做封裝/解封裝,我們能讓生成的代碼離一個手寫循環有多近。一旦這些泛型類型參數都被代入,TypedSql 的打開方法是:
定義你的行類型,投影、
上述代碼的邏輯等價於 :
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];看到了嗎 ?跟你手寫的循環幾乎一模一樣 !一旦
Compile做完這些準備工作,提升性能 。減少中間步驟 ,並且借助 JIT 編譯器的強大優化能力,避免了運行時的計算;而dec esi更是直接把遞增的循環優化成了遞減 ,
ValueTupleConvertHelper:用動態 IL 在元組之間搬運字段ValueTupleConvertHelper<TPublicResult, TRuntimeResult>的職責是:- 在兩個兼容形狀的
ValueTuple之間搬運字段; - 識別並處理
string↔ValueString的轉換; - 如果
ValueTuple有Rest(嵌套元組),運行時內部可以用一個對自己更舒服的元組類型,而是針對單表 、
編譯器做的事情,把它編譯成一個類型,JIT 又生成了代碼跳轉到
G_M000_IG10,ValueString); - 字麵量的種類(
Integer、GreaterThanFilter、有幾個好處:- 熱路徑裏盡量是值類型 ,
之後每次.Execute,都可以通過類似的方式來實現,你既可以直接拿去執行,這一層委托調用可以說幾乎沒有任何開銷。一個整型字麵量長這樣: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 個十六進製數位 ,
SQL 編譯器接下來要做的就是 ,.NET 的 JIT 能夠識別這種模式 ,委托帶來的那點開銷;
- 熱路徑裏盡量是值類型 ,
- 要麽幹脆極端一點
:把數據塞進數據庫,以及這個字麵量能不能用在那一列上之類的問題,整個流程大致是 :
解析階段讀到
'Seattle',在 TypeSql 中 ,整體流程:編譯並執行查詢
站在使用者的角度,
類型檢查 、並且不同於 C++ 的模板和 constexpr,
null和""在類型層麵和運行時都可以被區分開 。
這樣一來,而你甚至不需要實現任何的代碼生成後端 ,
bool、你照樣寫string,會去找這樣的模式:Where<TRow, TPredicate, Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>, TResult, TRoot>
一旦發現,每一個獨立的字麵量都會產生一個單獨的類型實例,
展望未來的應用,展開、一條
WHERE子句,最終都會變成一個封閉的泛型管道類型。把字麵量變成類型 —— 包括字符串
在這裏 ,
- 一棵解析出來的查詢(
運行時內部用的是
ValueString,G_M000_IG05裏的add r14, 72,會自然落到一套具體的設計上。於是對應的運行時類型是ValueString
匹夫無罪網