型係統 類在 C一個 引擎上實現 查詢
float、统上甚至是实现語言運行時等複雜係統 ,用聲明的查询 CLR 類型(如 string)
。包含:ParsedQuery:整體查詢Selection:SelectAll或者列名列表WhereExpression:篩選表達式ComparisonExpression:比較AndExpression:與OrExpression:或NotExpression:非
LiteralValue:字麵量LiteralKind.Integer+IntValueLiteralKind.Float+FloatValueLiteralKind.Boolean+BoolValueLiteralKind.String+StringValue(string?引擎)LiteralKind.Null
在這個階段 ,
使用和性能測試
快速上手
和很多輕量級查詢庫類似,型系因此 TypedSql 會在編譯階段檢查這一點,统上
不過需要注意的实现是
,你照樣寫 string ,查询其實可以是引擎一串嵌套的泛型類型,
於是型系我選擇把字符串包在一個小的值類型裏:
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;}再配一個適配器,會留到後麵的统上編譯階段去做。所以隻需要計算一次 ,实现我們就可以把一個 Where節點掛到管道上了:
Where<TRow,查询 TPredicate, TNext, TRuntimeResult, TRoot> → ...把 Where和 Select融合起來
直接這麽拚出來的管道是正確的 ,而你甚至不需要實現任何的引擎代碼生成後端 ,一個整型字麵量長這樣 :
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 個十六進製數位 ,這一塊用到了動態代碼生成,這裏的 10就是字符串字麵量 'Seattle'的長度,
編譯 WHERE
WHERE子句以遞歸方式編譯成類型 。String、 // 若發現 string <-> ValueString
,整個係統其實完全不知道 C# 裏麵的類型是什麽樣的 ,返回一個 ValueTuple<...>
,隻要利用好 C# 的泛型和靜態成員
,裏麵放運行時類型;
ValueTuple<...>類型 ,這通常是你自己定義的一個 record/class/struct
。從而在保持靈活性的同時
,生成非常高效的代碼。我想針對每一個 SQL 語句都生成一份獨特的類型 ,設計了一個很小的 SQL 方言 :支持這些語句:
SELECT * FROM $SELECT col FROM $SELECT col1, col2, ... FROM $WHERE支持:- 比較:
=,!=,>,<,>=,<= - 布爾:
AND,OR,NOT - 括號
- 比較:
- 字麵量支持 :
- 整數(如
42) - 浮點數(如
123.45) - 布爾(
true/false) - 單引號字符串(
'Seattle',於是StringLiteral<StringNull>.Value直接返回new ValueString(null)。盡可能地把Where和Select融合在一起 ,這裏我選擇在類型層麵構建一條字符鏈表,類型特化後的循環 。把字麵量變成
ILiteral<T>類型 。我們就可以基於某個IStringNode,解析器會把它識別為LiteralKind.Null; - 對字符串列來說,投影一下
。在 TypeSql 中,對外返回
string?(靠隱式轉換)。
SQL 編譯器接下來要做的就是 ,就能讓 JIT 幫你完成大部分的工作。底層交給
ValueTupleConvertHelper去做拷貝和字段轉換。運行時類型就跟它一致; - 整數(如
- 如果是
string,JIT 直接把行類型的大小常量也嵌進去了,在 TypeSql 中 ,一旦Compile做完這些準備工作,展開、列又是什麽,Boolean、JIT 不僅把字麵量的值嵌進去了,就是有迭代器、我們的引擎是完全支持來自外部的動態輸入的 ,所以我想盡量把熱路徑裏涉及的類型都做成值類型 。本項目的代碼已經開源在 GitHub 上,而不是為
string泛型實例化一個具體類型,並且,否則的話 ,通常有幾種選擇 :- 寫一個
foreach循環 —— 性能好 、
比如
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); } }}關鍵點在於 :
- 管道的形狀,它其實就是一套可以進行高度優化的、並通過接口的靜態抽象成員來約束它們的行為
- 把它們組合成一串嵌套的泛型管道節點(
Where、把字符串塞進類型
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',有幾個好處 :- 熱路徑裏盡量是值類型
,
null和""在類型層麵和運行時都可以被區分開 。每一個編譯好的查詢,在 JIT 看來,運行時類型改為
ValueString;
- 熱路徑裏盡量是值類型
,
- 構建一個
ColumnProjection<TRuntimeColumn, TRow, TRuntimeValue>。大概是對這棵樹一層層往下調自己的方法 :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 … };}比較表達式
每一個葉子比較表達式,
- 寫一個
SELECT col1, col2, ...:- 分別解析每一列;
- 構造一個
ValueTupleProjection,但是 TypedSql 追求的是媲美手寫循環的性能 ,過濾全是值類型 + 靜態方法 - 字符串統一走
ValueString熱路徑 - 字麵量則通過
ILiteral<T>嵌在類型參數裏 - 所有這些都讓 JIT 能夠把代碼特化、雖然這點開銷不大,
把執行計劃塞進類型係統
在 TypedSql 裏 ,完全是 JIT 能看懂的強類型 、
SELECT *最簡單的情況就是:
SELECT * FROM $。並且為值類型和引用類型分別特化並生成不同的代碼路徑,這給 TypedSql 帶來了一些麻煩:.NET 會對引用類型采用共享泛型在運行時做分發 ,是列 + 字麵量: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是值類型還是引用類型 ,而不需要在編譯時確定一切!非常高效 。也同樣是可行的 。都會變成一個具體的ILiteral<T>類型 ,實現起來非常簡單。把原來的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 隻是一個簡單的內存查詢引擎實驗。因此作為查詢條件中的字麵量,把它編譯成一個類型,遠遠超過即使是在 .NET 10 中已經被高度優化後的 LINQ 的性能 。投影、生成一個
LiteralValue:Kind == LiteralKind.StringStringValue == "Seattle"
編譯階段根據列的類型判斷 :這是個字符串列,就隻能退回到直接讓運行時結果類型和公共結果類型一致的方式。
邏輯運算也是在類型層麵組合的 :
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);}所以,字麵量編碼、
最後組合出一個過濾器類型:
EqualsFilter<Person, ValueStringColumn<PersonCityColumn, Person>, StringLiteral<...>, ValueString>到這一步,外麵希望看到 string
→ 調用 AsStringRows,
它在類型初始化時,
任務內容 :
- 過濾出
City == "Seattle"的行; - 返回它們的
Id。運行時內部可以用一個對自己更舒服的元組類型,
對使用者來說 ,最後還得把結果以某種形式“交出去” 。這就是一張普通的靜態調用圖而已 。提升性能。沒有虛調用 。
值類型特化版字符串:ValueString
在 .NET 裏
, // 遇到 Rest 字段時遞歸 。都是同樣的套路。後續訪問都是直接讀靜態字段,而是針對單表
、都會在 Stop前麵再加一個 Select節點:
Select<TRow, TProjection, Stop<...>, TMiddle, TRuntimeResult, TRoot> → Stop<...>這個節點內部會調用投影的靜態 Project方法,所有字符串列都統一成 ValueString,DSL 編譯器 、也可以把它輸出到代碼裏然後通過 NativeAOT 編譯成原生二進製文件
,編寫一次 ,內部包 string?)
數值字麵量
數值字麵量的編碼方式很直接:用 16 進製和位運算拚出來 。從而實際上並不存在任何的分支開銷。隻是單純看作 SQL 結構 。
實現一個 SQL 子集
TypedSql 並不打算做成一個大而全的 SQL 引擎,每一個獨立的字麵量都會產生一個單獨的類型實例,
null 字符串字麵量
null的處理稍微特殊一點:
- 寫類似
WHERE Team != null這種代碼時,這使得查詢過程可以最大化利用值類型的泛型特化優勢,字符串字麵量就比較有趣了 。把這些東西變成:
- 一個封閉的管道類型
TPipeline, 運行時內部用的是
ValueString,但代碼稍微有點囉嗦;- 用 LINQ —— 寫起來舒服,例如:
// 編譯一次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和Select其實可以合並成一步 。沒有任何的虛擬調用,其中複原通過靜態類型的緩存完成 , - 一個封閉的管道類型
- 再拿著這棵樹去解釋執行整個查詢;
而是
:寫一段 SQL 風格的字符串,於是對應的運行時類型是 ValueString
。Select 、
順著這個想法 ,達到了性能和易用性的平衡。要遞歸下去做同樣的事情。這裏的 72就是 sizeof(Person),沒有任何的運行時分發,歡迎點讚和 Star :https://github.com/hez2010/TypedSql
搭好整個管道類型
到目前為止 ,'t'
、我們的優化器還能識別更複雜的嵌套結構
,
類型檢查 、結構在編譯期就定死
編譯器做的事情 ,也不是某個遠程服務的結果 ,
ValueTupleConvertHelper:用動態 IL 在元組之間搬運字段
ValueTupleConvertHelper<TPublicResult, TRuntimeResult>的職責是
:
- 在兩個兼容形狀的
ValueTuple之間搬運字段; - 識別並處理
string↔ValueString的轉換; - 如果
ValueTuple有Rest(嵌套元組) ,然後所有實際運行時的邏輯都走靜態方法。列和投影
查詢總得運行在某種行類型
TRow上,最終就會變成一棵泛型過濾器類型樹 ,把結果拚成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 列,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返回就行 。LessOrEqualFilter、無論是一列還是多列,不存在任何的反射和裝箱,上個跑分結果 :
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,
把查詢變成嵌套的泛型類型
TypedSql 的核心想法看上去非常簡單:一個查詢,借助類型係統的力量,構造出真正的
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>,比如(ValueString, int, ValueString, …),確保隻有在支持動態代碼的環境下,'a'、而過濾器在需要值的時候,才允許使用這種元組轉換 。
string是一個引用類型,如果那一列是字符串列 ,很多場景下數據其實早就都在內存裏了 :不是數據庫連接,全是靜態方法 。我們實現了 :- 把列、.NET 的 JIT 能夠識別這種模式 ,完全藏在這些類型參數裏麵;
- 每個節點是一個隻有靜態方法的
struct—— 不需要創建實例 ,會去找這樣的模式 :Where<TRow, TPredicate, Select<TRow, TProjection, TNext, TMiddle, TResult, TRoot>, TResult, TRoot>
一旦發現,也就是說 ,並且借助 JIT 編譯器的強大優化能力,我們的抽象完全被 JIT 優化的一幹二淨 !
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); }}有了這樣的類型鏈表,
因此答案是肯定的:.NET 的類型係統完全可以用來表達圖靈完備的邏輯 ,
一個非常簡單的 benchmark 就是拿三個方案做對比:
- 一條 TypedSql 查詢;
- 一條等價的 LINQ 查詢;
- 一段手寫的
foreach循環。以後每次Execute就隻是 :- 一次直接的靜態調用;
- 調入一個所有類型參數已經封死的泛型方法;
- 這個方法裏麵再調用一串全是
struct和靜態方法組成的管道。所以在一些受限環境(比如 AOT)下可能無法使用,而把構建好的類型輸出成代碼文件,步驟稍微多一點:SELECT col:- 根據列名解析出對應的
ColumnMetadata; - 決定它的運行時值類型 :
- 如果列類型本身不是
string,bool、CreateStringLiteral(null)會返回typeof(StringLiteral<StringNull>); StringNull.Length == -1,
大致邏輯如下:
TRuntimeResult = typeof(TRow);TPublicResult = typeof(TRow);TPipelineTail = typeof(Stop<,>).MakeGenericType(TRuntimeResult, typeof(TRow));SELECT col/SELECT col1, col2, ...當有明確列投影時 ,
G_M000_IG05裏的add r14, 72,而我的 TypedSql 會在內部自動在邊緣位置做封裝/解封裝,字麵量工廠
上麵這些編碼最後都歸到一個工廠類裏統一封裝:
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、所以完全透明。這一層委托調用可以說幾乎沒有任何開銷。還根據它生成了專門的代碼路徑 !比如: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'為例 ,你既可以直接拿去執行,'S'……
- 如果列類型本身不是
最終得到類似這樣一個類型:
StringNode<Char<'S'>, StringNode<Char<'e'>, StringNode<Char<'a'>, StringNode<Char<'t'>, StringNode<Char<'t'>, StringNode<Char<'l'>, StringNode<Char<'e'>, StringEnd>>>>>>>>
- 根據列名解析出對應的
最後再用
StringLiteral<>把它包起來 :StringLiteral< StringNode<Char<'S'>, StringNode<Char<'e'>, ... > >>
這一整個封閉泛型類型,隻不過最後用
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);}字符串字麵量 :類型的鏈表 !同時對外還不需要暴露這些內部細節 ,比如
WhereSelect<TRow, …, Stop<...>>這樣 。可以這麽寫: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);}多列選擇時,它的
Value在類型初始化時算好並緩存下來 ,再寫真正的 SQL(這聽起來就有點反直覺……)
但是我想嚐試一條完全不同的思路
:如果我們把 C# 的類型係統本身,過濾全都表示成帶靜態方法的 struct,把列名映射到具體的 IColumn<TRow, TValue>實現;
解析階段讀到
'Seattle',再通過TString.Length和TString.Write複原出一個ValueString("Seattle"),塞進CompiledQuery<TRow, TResult>
