【C++11】实现对象池

对象池

对象池顾名思义，就是存放一堆对象的池，目前一般的实现方式大多都是如下思路：

1. 初始化创建一些系列（或者从外部加载）
2. 从对象池取出一个对象使用
3. 用完之后返回对象池

对象池有很多场景都可以使用，例如：一个对象构造很消耗时间，又或者一些对象被很频繁的使用，都可以使用对象池。它的思想类似于数据库连接池、网络句柄连接池等等。

在C++中借助智能指针，可以很方便的实现一个对象池，智能指针可以自定义删除器，在智能指针需要释放的时候调用删除器，将对象重新放入对象池。
这里关键要注意：

1. 自定义删除器
2. shared_ptr或者unique_ptr

细节

• 自定义删除器只做一件事，就是将对象重新放入对象池。如果对象池初始化的时候就自定义删除器的话，删除器中的逻辑是将对象放回对象池，放回的时候无法再定义一个这样的删除器，所以这种做法行不通。

• 需要注意，回收的对象只能是默认删除器的。除了前述原因之外，另外一个原因是对象池释放的时候需要释放所有的智能指针，释放的时候如果存在自定义删除器将会导致对象无法删除。

• 只有在get的时候定义删除器才行，但是初始创建或加入的智能指针是默认删除器，所以我们需要把智能指针的默认删除器改为自定义删除器。

因为我们需要把智能指针的默认删除器改为自定义删除器，用shared_ptr会很不方便，因为你无法直接将shared_ptr的删除器修改为自定义删除器，虽然你可以通过重新创建一个新对象，把原对象拷贝过来的做法来实现，但是这样做效率比较低。而unique_ptr由于是独占语义，提供了一种简便的方法方法可以实现修改删除器，所以用unique_ptr是最合适的。

代码实现

template<class T>
class SimpleObjectPool
{
public:
    using DelType = std::function<void(T*)>;

    void add(std::unique_ptr<T> t)
    {
        m_pool.push_back(std::move(t));
    }

    std::unique_ptr<T, DelType> get()
    {
        if (m_pool.empty())
        {
            throw std::logic_error("no object");
        }

        //every time add custom deleter for default unique_ptr
        std::unique_ptr<T, DelType> ptr(m_pool.back().release(), [this](T* t)
        {
            m_pool.push_back(std::unique_ptr<T>(t));    //删除函数执行的操作
        });

        m_pool.pop_back();  //删除最后一个被释放的unique_ptr
        return std::move(ptr);
    }

    bool empty() const
    {
        return m_pool.empty();
    }

    int size() const
    {
        return m_pool.size();
    }
private:
    std::vector<std::unique_ptr<T>> m_pool;
};

//测试对象
struct STA
{
    STA()
    {
        cout << "create" << endl;
    }
    STA(int i)
    {
        a = i;
        cout << "create int" << endl;
    }
    STA(const STA& other)
    {
        *this = other;
        cout << "copy" << endl;
    }
    ~STA()
    {
        cout << "delete" << endl;
    }
    int a = 0;
};

//测试代码
void testPool()
{
    SimpleObjectPool<STA> pool;
    pool.add(std::unique_ptr<STA>(new STA()));
    pool.add(std::unique_ptr<STA>(new STA()));
    {
        auto ptr = pool.get();
        pool.get();
        cout <<"pool size:"<< pool.size() << endl;
    }
    cout << "pool size:" << pool.size() << endl;
    {
        pool.get();
        pool.get();
        cout << "pool size:" << pool.size() << endl;
    }
    cout << "pool size:" << pool.size() << endl;
}

输出如下：