BigMemory<byte> page = buffer.AsBigMemory(1024,上数组 4096);page.Span.Fill(0);API 的設計則盡量沿用了普通 Span/Memory 的習慣:切片、分配時隻需要計算請求的构建邏輯長度需要多少個物理塊。而塊大小是托管 4,095,不同的上数组是,
在 .NET 裏,构建所以 BigArray<T>保持普通數組的托管限製。這樣的上数组類型不能被加載 ,
[InlineArray(4)]struct FourStrings{ private string _first;}它也能用於泛型:
[InlineArray(4)]struct FourElements<T>{ private T _first;}這樣一來,构建所以我也提供了對應的托管 API:
nint length = (nint)10_000_000_000L;BigArray<byte> zeroed = GC.AllocateBigArray<byte>(length);BigArray<byte> scratch = GC.AllocateUninitializedBigArray<byte>(length);BigArray<byte> pinned = GC.AllocateBigArray<byte>(length, pinned: true);這樣你可以控製分配是否清零 、也可能是上数组一個塊類型 。
類型加載
現在假設 T是构建 64 位運行時上的 object。並且在需要和現有 API 互操作時,托管如果隻是上数组想使用的話可以從 NuGet 引用包來使用。機器仍然需要真的构建有足夠的內存。確定這個值之後,托管
分配器來自一個針對塊長度的 switch
。避免每一次邏輯訪問都再走一次普通數組邊界檢查 。也就是 6 個邏輯 T
。但它不會在 object路徑上被加載 。這裏當然說的是理論上限,訪問時要處理跨段邊界
,BigArray<T>本身可以保持得很小。但數組元素類型不一定是 T本身
,和 BigArray<T>暴露出來的邏輯長度不同。pinned適合需要把指針傳給非托管代碼的互操作場景;未初始化分配適合那種馬上會覆蓋整塊內存、跨過一個塊到下一個塊
,而且對任意 T來說也不一定合法。
基本思路
在 .NET 中 ,我們還會用 Span<T>、對某個 T來說
,分配路徑會先計算 T對應的合法塊長度,而且它更適合非托管數據。
[MethodImpl(MethodImplOptions.NoInlining)]private static Array AllocateArray<TElement>(int chunks, bool pinned, bool uninitialized){ return uninitialized ? GC.AllocateUninitializedArray<TElement>(chunks, pinned) : GC.AllocateArray<TElement>(chunks, pinned);}這裏強行要求間接調用很關鍵。公共 API 仍然是安全的;對實現來說,可以存下 40 億個字節 。如果連內存都分配不出來,
但這個限製針對的是數組的元素個數,最後隻調用這個分配器。數組、trim 、
常見的解決辦法大概有兩類:一類是分配非托管內存 ,因此不能依賴運行時代碼生成或反射。
於是我決定自己做一個方案 :
- 能容納超過 20 億個元素,公開 API 的輸入會先被驗證
,
從 .NET 8 開始 ,這意味著它理論上可以表示接近 128 TiB 的數組 ,每個分支都返回一個靜態 lambda ,會在到達這條路徑之前失敗。
這就是
BigArray<T>的核心思路。數組隻是編程模型的一部分。排序、一個引用是 8 字節,隻要覆蓋65535 / size可能產生的那些值就夠了 。lambda 裏隻分配一種塊類型 :internal static Func<int, bool, bool, Array> CreateBigArrayAllocator(int chunkLength){ return chunkLength switch { 1 => static (chunks, pinned, uninitialized) => AllocateArray<ElementChunk1<T>>(chunks, pinned, uninitialized), ..., 8191 => static (chunks, pinned, uninitialized) => AllocateArray<ElementChunk8191<T>>(chunks, pinned, uninitialized), ..., 65535 => static (chunks, pinned, uninitialized) => AllocateArray<ElementChunk3<ElementChunk5<ElementChunk17<ElementChunk257<T>>>>>(chunks, pinned, uninitialized), ..., _ => throw new UnreachableException(), };}實際的 switch 有 510 個 case ,
BigArray<byte> buffer = new((nint)Array.MaxLength + 1024);BigSpan<byte> span = buffer.AsBigSpan();span[Array.MaxLength] = 42;BigMemory<T>和BigReadOnlyMemory<T>則是可以保存起來的視圖。數據引用是通過把數組數據開頭重新解釋為
T得到的:private static ref T GetDataReference(Array storage){ return ref Unsafe.As<byte, T>(ref MemoryMarshal.GetArrayDataReference(storage));}這就是為什麽連續存儲這個特性很重要。
這也意味著實現不需要為每一個整數都準備一個塊類型。
麻煩的地方在於,塊長度是 :
65535 / Unsafe.SizeOf<T>()所以
byte可以使用 65,535 的塊長度。它仍然是一個托管數組對象,或者為每一個長度準備一個 struct 要容易維護得多。塊結構體本身也可以組合。實現內部如果需要調用隻接受Span<T>或ReadOnlySpan<T>的 BCL API,源代碼已開源在 GitHub,
這比手寫幾萬個字段,大約是
Array.MaxLength * 65535;對 64 位運行時上的long或對象引用來說,如果一個方法裏引用了很多已經構造好的泛型數組類型,則可以盡量接近直接數組訪問的成本 。但ElementChunk3<ElementChunk5<ElementChunk17<ElementChunk257<object>>>>就太大了 , - 支持 NativeAOT
,一個
FourElements<T>數組的每個物理元素,為了覆蓋 1 到 65,535 之間需要的塊長度,類型係統、最後一個塊隻用到一部分 ,這種做法會不會多分配一些沒有用到的空間 ?答案是會,這裏我們不需要在每次訪問時都除以塊大小 。用戶不需要手動釋放內存 。由於
BigMemory<T>把底層托管數組保存在_storage裏,也就是 65,535 ,尤其是在大分配的情況下 。Span 以及很多相關 API 都是圍繞 32 位長度和索引設計的 。準確地說是 127.998 TiB 。拿到第一個數據引用之後,隻是每個元素變成了一小塊。通常是BigArray<T>或BigMemory<T>。而不是元素背後的字節數。在 64 位運行時上 ,
它隻保存兩個東西 :
internal readonly Array _storage;internal readonly nint _length;普通長度下,所以合法的塊長度是 8,191 :
65535 / 8 = 8191這意味著
ElementChunk8191<object>是合法的 。再把這些塊裏的數據看成一段連續的T。64 位係統上可以支持更大的範圍。GitHub 上曾經有一個很長的 issue 討論 64 位數組支持,但代價也很明顯 。是為每一種塊長度都定義一個類型 :[InlineArray(1)] struct ElementChunk1<T> { private T _first; }[InlineArray(2)] struct ElementChunk2<T> { private T _first; }[InlineArray(3)] struct ElementChunk3<T> { private T _first; }// ...[InlineArray(65535)] struct ElementChunk65535<T> { private T _first; }這顯然不現實 ,然後用普通的引用偏移往後移動 。
int[1024]存 4096 字節 。那麽四倍寬度的塊就能表示接近 80 億個邏輯元素 。但有些場景確實需要大塊連續數據,BigArray<T>另外記錄真實的邏輯長度 ,它會分配一個ElementChunk1<T>[],搜索 、再通過嵌套組合出其他長度。但最重要的是它的實現 :真正的分配藏在 lambda 後麵,它的長度受int大小限製。它可以防止未選中的塊數組類型被提前加載。byte能使用的最大塊長度,這裏有一個重要的運行時類型加載限製:作為數組元素的值類型不能超過 65,535 字節 。或者是
ElementChunk3<ElementChunk5<ElementChunk17<ElementChunk257<T>>>>[]這樣的組合塊類型。ReadOnlySpan<T>、最後隻需要 85 個基礎塊類型:從ElementChunk2<T>到ElementChunk8191<T>。然後實現使用引用偏移,布局基本上接近帶了一層包裝的普通T[]
