// 文件名 : Mempool.h
// 作者 : Moonwell
// Msn : archonzhao@hotmail.com
// 将#include "mempool.h"放在#include <windows.h>之前
// 否则会出现"InitializeCriticalSectionAndSpinCount函数没定义"的错误#ifndef _MEM_POOL_H
#define _MEM_POOL_H
#define _WIN32_WINNT 0x0403 //For InitializeCriticalSectionAndSpinCount
#ifdef _DEBUG
#define _DEBUG_POOL
#endif
#include <windows.h>
#include <crtdbg.h>
#include <vector>
#ifdef _DEBUG_POOL
#include <map>
#endif
using namespace std;
template<typename memtype>
class MemPool
{
public:
//init_count = 初始分配的对象个数
MemPool(int init_count);
~MemPool(void);
memtype *alloc();// 读取一个元素
void free(memtype *obj);// 删除一个元素
private:
vector<memtype *> m_free_node; //空闲的对象列表
vector<memtype *> m_new_node; //额外创建的对象列表
#ifdef _DEBUG_POOL
map<memtype *,memtype *> m_map;//用于检测对象是否被释放了多次
#endif
memtype *m_head;//初始创建时候的内存块的指针
CRITICAL_SECTION m_sec;
};
template<typename memtype>
MemPool<memtype>::MemPool(int init_count)
{
m_head = new memtype[init_count];
//先把vector对象的内存增加到原来的两倍大小,可以提高效率
m_free_node.resize(init_count*2);
m_free_node.resize(0);
//将对象添加到空闲列表
for(int i=0;i<init_count;i++)
m_free_node.push_back(&m_head[i]);
InitializeCriticalSectionAndSpinCount(&m_sec,64);
}
template<typename memtype>
MemPool<memtype>::~MemPool(void)
{
vector<memtype *>::iterator it;
DeleteCriticalSection(&m_sec);
delete [] m_head;
// 删除所有额外创建的对象
for(it = m_new_node.begin();it != m_new_node.end();++it)
delete [] *it;
}
template<typename memtype>
memtype *MemPool<memtype>::alloc()
{
memtype *ptmp;
EnterCriticalSection(&m_sec);
if(!m_free_node.empty()) {
ptmp = *(--m_free_node.end()); //取最后一个对象
m_free_node.pop_back();
#ifdef _DEBUG_POOL
m_map.erase(ptmp);
#endif
} else {
ptmp = new memtype; //创建额外的对象
m_new_node.push_back(ptmp); //添加到额外列表
}
LeaveCriticalSection(&m_sec);
return ptmp;
}
template<typename memtype>
void MemPool<memtype>::free(memtype *obj)
{
EnterCriticalSection(&m_sec);
m_free_node.push_back(obj);
#ifdef _DEBUG_POOL //obj已经被释放了吗?
_ASSERTE(m_map.find(obj) == m_map.end());
m_map[obj] = obj;
#endif
LeaveCriticalSection(&m_sec);
}
#endif
在原来基础上增加了多次free()检测的功能,对同一个对象释放两次将会引发一个断言.
