windows平台上的一个内存池的实现
.h文件
/**********************说明*************************
* 这是MPool内存池的实现,他具有例如以下特性:
* 1. 池中的内存块是大小是同样的
* 2. 由宏定义_MP_NO_SERIALIZE决定是否须要多线程同步
* 3. 他利用windows的堆内存API进行内存分配
* 4. 他不能替换crt的malloc和free
* 5. 他不是一个通用型的内存池
* 6. 适用于特定的应用环境(高频率的申请释放内存,如网络server),应用环境影响池中内存块的大小。以及是否须要处理多线程同步
*
* 编写这样一个内存池基于这几点原因
* 1. 降低内存碎片(每次分配等长的内存块有利于降低内存碎片。内存块的大小还要考虑内存对齐)
* 2. 提高内存分配效率(非序列化的内存分配能够显著提高效率)
* 3. 网络上非常多通用型内存池的做法,本人觉得通用型的内存池就应该是操作系统本身的内存管理API,
在ring3层做通用内存池没有意义。甚至可能起到反效果。这样的情况最简单的解决这个问题的办法是添加物理内存的容量。
仅仅有针对特定的应用环境编写的内存池才是有意义的。
*/
#ifndef MPOOL_H
#define MPOOL_H
#include <Windows.h>
#include <vector>
using std::vector;
class MPool
{
public:
MPool(size_t block_size);
~MPool();
void *MPAlloc();
void MPFree(void *p);
private:
MPool(const MPool&);
MPool& operator=(const MPool&);
private:
RTL_CRITICAL_SECTION *m_pool_lock;
vector<void*> m_block_list;
size_t m_block_size;
size_t m_total_size;
size_t m_block_count;
size_t m_reserve_count;
HANDLE m_heap;
DWORD m_heap_option;
};
#endif.cpp文件
#include "mpool.h"
#ifdef _MP_NO_SERIALIZE
#define MPLOCK()
#define MPUNLOCK()
#else
#define MP_LOCK() EnterCriticalSection(m_pool_lock);
#define MP_UNLOCK() LeaveCriticalSection(m_pool_lock);
#endif
MPool::MPool( size_t block_size )
{
m_block_size = block_size;
size_t m = m_block_size % 8;
if(m != 0)
{
m_block_size += (8-m);
}
_SYSTEM_INFO info;
memset(&info,0,sizeof(_SYSTEM_INFO));
GetSystemInfo(&info);
m_heap_option = 0;
#ifdef _MP_NO_SERIALIZE
m_heap_option = HEAP_NO_SERIALIZE;
#else
m_pool_lock = new RTL_CRITICAL_SECTION;
InitializeCriticalSectionAndSpinCount(m_pool_lock,4000);
#endif
m_heap = HeapCreate(m_heap_option,info.dwPageSize,0);
m_block_list.reserve(1024);
}
MPool::~MPool()
{
while(!m_block_list.empty())
{
HeapFree(m_heap,m_heap_option,m_block_list.back());
m_block_list.pop_back();
}
#ifndef _MP_NO_SERIALIZE
DeleteCriticalSection(m_pool_lock);
delete m_pool_lock;
#endif
}
_inline void * MPool::MPAlloc()
{
void *ret;
MP_LOCK();
if(!m_block_list.empty())
{
ret = m_block_list.back();
m_block_list.pop_back();
}
else
{
ret = HeapAlloc(m_heap,m_heap_option,m_block_size);
}
MP_UNLOCK();
return ret;
}
_inline void MPool::MPFree( void *p )
{
if(p)
{
MP_LOCK();
m_block_list.push_back(p);
MP_UNLOCK();
}
}
浙公网安备 33010602011771号