C#?ArrayPool的實(shí)現(xiàn)示例

更新時(shí)間：2025年07月01日 10:53:24 作者：Xioa.

ArrayPool是.NET中一個(gè)高性能數(shù)組池,用于減少內(nèi)存分配和垃圾回收的開銷,下面就來介紹一下ArrayPool的具體使用,感興趣的可以了解一下

基本用法

// 基本使用方式
public void BasicUsage()
{
    // 從共享池獲取數(shù)組
    char[] array = ArrayPool<char>.Shared.Rent(100);
    
    try
    {
        // 使用數(shù)組
        // ... 業(yè)務(wù)邏輯 ...
    }
    finally
    {
        // 歸還數(shù)組到池中
        ArrayPool<char>.Shared.Return(array);
    }
}

高級用法

// 創(chuàng)建自定義數(shù)組池
public class CustomArrayPoolExample
{
    private readonly ArrayPool<byte> _customPool;
    
    public CustomArrayPoolExample()
    {
        // 創(chuàng)建最大數(shù)組長度為1024，最多保留10個(gè)數(shù)組的池
        _customPool = ArrayPool<byte>.Create(maxArrayLength: 1024, maxArraysPerBucket: 10);
    }
    
    public void UseCustomPool()
    {
        // 從自定義池租用數(shù)組
        byte[] buffer = _customPool.Rent(256);
        
        try
        {
            // 使用數(shù)組
            // ... 業(yè)務(wù)邏輯 ...
        }
        finally
        {
            // 清除敏感數(shù)據(jù)
            Array.Clear(buffer, 0, buffer.Length);
            // 歸還數(shù)組
            _customPool.Return(buffer);
        }
    }
}

性能優(yōu)化示例

public class HighPerformanceExample
{
    private static readonly ArrayPool<char> _charPool = ArrayPool<char>.Shared;
    
    public string ProcessLargeString(string input)
    {
        char[] buffer = _charPool.Rent(input.Length * 2);
        
        try
        {
            // 處理字符串
            int resultLength = ProcessBuffer(input, buffer);
            return new string(buffer, 0, resultLength);
        }
        finally
        {
            _charPool.Return(buffer);
        }
    }
    
    private int ProcessBuffer(string input, char[] buffer)
    {
        // 處理邏輯
        return input.Length;
    }
}

最佳實(shí)踐

使用 using 語句

public string SafeArrayPoolUsage(int size)
{
    using var owner = MemoryPool<char>.Shared.Rent(size);
    var memory = owner.Memory;
    // 使用 memory
    return memory.ToString();
}

/**

需要直接數(shù)組操作，選擇 ArrayPool
需要安全性保證，選擇 MemoryPool
異步操作推薦 MemoryPool
高性能場景推薦 ArrayPool
*/

避免常見錯(cuò)誤

// 錯(cuò)誤示例 - 不要這樣做
var array = ArrayPool<int>.Shared.Rent(100);
// 忘記返回?cái)?shù)組 - 內(nèi)存泄漏!

// 正確示例
var array = ArrayPool<int>.Shared.Rent(100);
try
{
    // 使用數(shù)組
}
finally
{
    ArrayPool<int>.Shared.Return(array);
}

使用場景

大數(shù)據(jù)處理場景

public class BigDataProcessor
{
    private readonly ArrayPool<byte> _pool = ArrayPool<byte>.Shared;

    public void ProcessLargeData(Stream stream)
    {
        byte[] buffer = _pool.Rent(81920); // 80KB 緩沖區(qū)
        try
        {
            while (stream.Read(buffer, 0, buffer.Length) > 0)
            {
                // 處理數(shù)據(jù)
            }
        }
        finally
        {
            _pool.Return(buffer);
        }
    }
}

圖像處理

public class ImageProcessor
{
    private readonly ArrayPool<byte> _pixelPool = ArrayPool<byte>.Shared;

    public void ProcessImage(Bitmap bitmap)
    {
        byte[] pixels = _pixelPool.Rent(bitmap.Width * bitmap.Height * 4);
        try
        {
            // 處理圖像像素
        }
        finally
        {
            _pixelPool.Return(pixels);
        }
    }
}

網(wǎng)絡(luò)通信

public class NetworkHandler
{
    private readonly ArrayPool<byte> _networkPool = ArrayPool<byte>.Shared;

    public async Task HandleNetworkData(NetworkStream stream)
    {
        byte[] buffer = _networkPool.Rent(4096);
        try
        {
            await stream.ReadAsync(buffer, 0, buffer.Length);
            // 處理網(wǎng)絡(luò)數(shù)據(jù)
        }
        finally
        {
            _networkPool.Return(buffer);
        }
    }
}

字符串處理

public class StringProcessor
{
    private readonly ArrayPool<char> _charPool = ArrayPool<char>.Shared;

    public string ProcessLargeString(string input)
    {
        char[] buffer = _charPool.Rent(input.Length * 2);
        try
        {
            // 字符串處理邏輯
            return new string(buffer, 0, input.Length);
        }
        finally
        {
            _charPool.Return(buffer);
        }
    }
}

高頻臨時(shí)數(shù)組

public class HighFrequencyProcessor
{
    private readonly ArrayPool<int> _intPool = ArrayPool<int>.Shared;

    public void ProcessData()
    {
        int[] tempArray = _intPool.Rent(1000);
        try
        {
            // 高頻臨時(shí)計(jì)算
        }
        finally
        {
            _intPool.Return(tempArray, clearArray: true); // 清除敏感數(shù)據(jù)
        }
    }
}