基于我前一篇文章《
基于ACE實現的一個內存池》的代碼，這里是應用，有問題不要打我。。。
// MessageBlockManager.h
/**//**
* @date 2007.10.28
* @author PeakGao <peakgao163@163.com>
*/
#ifndef OM_MESSAGEBLOCKMANAGER_H
#define OM_MESSAGEBLOCKMANAGER_H
#include <ace/thread_mutex.h>
#include <ace/synch_traits.h>
#include <ace/message_block.h>
#include "MemPoolT.h"
namespace om
{
class MessageBlockManager
{
typedef CachePool<ACE_SYNCH_MUTEX> MyCachePool;
MyCachePool mBufPool; /**//// 緩存ACE_Data_Block的緩沖區
MyCachePool mDBPool; /**//// 緩存ACE_Data_Block對象本身，由于不能默認構造，所以沒有用ObjectPool模版
MyCachePool mMBPool; /**//// 緩存ACE_Message_Block對象本身，由于不能默認構造，所以沒有用ObjectPool模版
ACE_SYNCH_MUTEX* mMutex; /**//// 線程同步
public:
static MessageBlockManager* instance();
MessageBlockManager();
~MessageBlockManager();
/**//// 初始化參數
void create(size_t bufSize, size_t objCount);
/**//// 清除緩沖池
void clear();
/**//// 分配一個消息塊
ACE_Message_Block* alloc();
/**//// 釋放指定的消息塊
void free(ACE_Message_Block* mb);
};
} // namespace om
#endif // OM_MESSAGEBLOCKMANAGER_H// MessageBlockManager.cpp
/**//**
* @date 2007.10.28
* @author PeakGao <peakgao163@163.com>
*/
#include "MessageBlockManager.h"
namespace om{
MessageBlockManager* MessageBlockManager::instance()
{
static MessageBlockManager* mbm = 0;
if (mbm == 0)
mbm = new MessageBlockManager();
return mbm;
}
MessageBlockManager::MessageBlockManager()
: mMutex(NULL)
{
}
MessageBlockManager::~MessageBlockManager()
{
clear();
}
void MessageBlockManager::clear()
{
if (mMutex)
{
mBufPool.clear();
mDBPool.clear();
mMBPool.clear();
delete mMutex;
mMutex = NULL;
}
}
void MessageBlockManager::create(size_t bufSize, size_t objCount)
{
if (!mMutex)
mMutex = new ACE_SYNCH_MUTEX();
mBufPool.create(bufSize, objCount, mMutex);
mDBPool.create(sizeof(ACE_Data_Block), objCount, mMutex);
mMBPool.create(sizeof(ACE_Message_Block), objCount, mMutex);
}
ACE_Message_Block* MessageBlockManager::alloc()
{
void* buf = mBufPool.alloc();
ACE_Data_Block* db = ::new (mDBPool.alloc()) ACE_Data_Block(mBufPool.getBlockSize(),
ACE_Message_Block::MB_DATA, static_cast<const char*>(buf), 0, 0, ACE_Message_Block::DONT_DELETE, 0);
ACE_Message_Block* mb = ::new (mMBPool.alloc()) ACE_Message_Block(db, ACE_Message_Block::DONT_DELETE);
return mb;
}
void MessageBlockManager::free(ACE_Message_Block* mb)
{
// 不要調用對象的release方法
mBufPool.free(mb->base());
ACE_Data_Block* db = mb->data_block();
db->~ACE_Data_Block();
mDBPool.free(db);
mb->~ACE_Message_Block();
mMBPool.free(mb);
}
} // namespace om