如何在 C# 中实现可重用的对象池

本文关键字:对象 实现 | 更新日期: 2023-09-27 17:56:26

我正在处理流套接字中的大量数据。数据被使用并留给 GC 清理。我想预先分配一个可重用的池,并重用它以防止大量 GC。

谁能帮我?

如何在 C# 中实现可重用的对象池

恕我直言,这是一个有效的问题。特别是在使用频繁分配缓冲区的套接字服务器时。这就是所谓的蝇量级模式。

但我不会轻易决定使用它。

class BufferPool<T>
{
    private readonly Func<T> _factoryMethod;
    private ConcurrentQueue<T> _queue = new ConcurrentQueue<T>();
    public BufferPool(Func<T> factoryMethod)
    {
        _factoryMethod = factoryMethod;
    }
    public void Allocate(int count)
    {
        for (int i = 0; i < count; i++)
            _queue.Enqueue(_factoryMethod());
    }
    public T Dequeue()
    {
        T buffer;
        return !_queue.TryDequeue(out buffer) ? _factoryMethod() : buffer;
    }
    public void Enqueue(T buffer)
    {
        _queue.Enqueue(buffer);
    }
}

用法:

var myPool = new BufferPool<byte[]>(() => new byte[65535]);
myPool.Allocate(1000);
var buffer= myPool.Dequeue();
// .. do something here ..
myPool.Enqueue(buffer);