在C#中實現簡單的對象池

當我們頻繁創建刪除大量對象的時候，對象的創建刪除所造成的開銷就不容小覷了。為了提高性能，我們往往需要實現一個對象池作為Cache：使用對象時，它從池中提取。用完對象時，它放回池中。從而減少創建對象的開銷。

由於.net BCL庫中並沒有對象池的標准實現，因此需要我們自己去實現。好在實現功能簡單的對象池並不麻煩，一般幾十行代碼就能實現。需要注意的一點是，對象池大多是需要支持多線程訪問的，因此需要考慮線程安全問題。

在.Net 4.0后，BCL在System.Collections.Concurrent名字空間下引入了一系列線程安全的對象，微軟甚至在MSDN上給了一個通過實現對象池的簡單示例：How to: Create an Object Pool by Using a ConcurrentBag.

這個例子本身沒有什么問題，但如果放在實際生產中就覺得有點簡單過頭了，一般還需要加上容量限制和重用時進行reset操作。這里我就稍微將其改了下： 

public class ObjectPool<T>
    {
        ConcurrentBag<T> buffer;
        Func<T> createFunc;
        Action<T> resetFunc;

        public ObjectPool(Func<T> createFunc, Action<T> resetFunc, int capacity)
        {
            Contract.Assume(createFunc != null);
            Contract.Assume(capacity > 0);

            this.buffer = new ConcurrentBag<T>();
            this.createFunc = createFunc;
            this.resetFunc = resetFunc;

            this.Capacity = capacity;
        }

        public int Capacity { get; private set; }
        public int Count { get { return buffer.Count; } }

        /// <summary>
        /// 申請對象
        /// </summary>
        public T GetObject()
        {
            var obj = default(T);

            if (!buffer.TryTake(out obj))
                return createFunc();
            else
                return obj;
        }

        /// <summary>
        /// 釋放對象
        /// </summary>
        public void PutObject(T obj)
        {
            Contract.Assume(obj != null);

            if (Count >= Capacity)        //超過容量了，不再需要
                return;

            if (resetFunc != null)
                resetFunc(obj);

            buffer.Add(obj);
        }
    }

需要注意的是，我這里的實現並沒有完全確保Capacity的絕對性：當兩個線程同時往一個即將到達上限的對象池中放置對象時，可能都會成功。因為我覺得這個是沒有太大必要的，感興趣的朋友可以把它改下。