pool.h

/* Copyright (c) 2010, Stefan Eilemann <eile@eyescale.ch> 
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License version 2.1 as published
 * by the Free Software Foundation.
 *  
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#ifndef COBASE_POOL_H
#define COBASE_POOL_H

#include <co/base/nonCopyable.h>  // base class
#include <co/base/scopedMutex.h>  // member
#include <co/base/spinLock.h>     // member
#include <co/base/thread.h>       // thread-safety checks

namespace co
{
namespace base
{
    template< typename T, bool locked = false > class Pool : public NonCopyable
    {
    public:
        Pool() : _lock( locked ? new SpinLock : 0 ) {}

        virtual ~Pool() { flush(); delete _lock; }

        T* alloc()
            {
                ScopedMutex< SpinLock > mutex( _lock );
                EQ_TS_SCOPED( _thread );
                if( _cache.empty( ))
                    return new T;

                T* item = _cache.back();
                _cache.pop_back();
                return item;
            }

        void release( T* item )
            {
                ScopedMutex< SpinLock > mutex( _lock );
                EQ_TS_SCOPED( _thread );
                _cache.push_back( item );
            }

        void flush()
            {
                ScopedMutex< SpinLock > mutex( _lock );
                EQ_TS_SCOPED( _thread );
                while( !_cache.empty( ))
                {
                    delete _cache.back();
                    _cache.pop_back();
                }
            }

    private:
        SpinLock* const _lock;
        std::vector< T* > _cache;
        EQ_TS_VAR( _thread );
    };
}
}
#endif // COBASE_POOL_H