标签:inf std 空闲 red 新版 查找 error clu public
1.1 数组的特点
1.2 链表的特点
1.3 基于数组和指针实现的对象管理器
结合了数组和链表的优点,可以O(1)查找元素,O(1)增删元素;
需要预分配内存;
2.代码实现(C++)
/** *@file ObjAllocator *@author jasonxiong *@date 2009-12-09 *@version 1.0 *@brief CObj对象分配类,即新版本的idxobj * * (1)一般用于大型对象的内存分配 */ #ifndef __OBJ_ALLOCATOR_HPP__ #define __OBJ_ALLOCATOR_HPP__ #include <stdio.h> namespace ServerLib { typedef enum enmObjAllocType { EOAT_ALLOC_BY_SELF = 0, //!<对象内存由ObjAllocator自已动态分配 EOAT_ALLOC_BY_SHARED_MEMORY = 1, //!<对象内存由共享内存分配 } ENMOBJALLOCTYPE; class CObj; typedef enum enmIdxUseFlag { EIUF_FREE = 0, //!<该对象未被使用 EIUF_USED = 1, //!<该对象已被使用 } ENMIDXUSEFLAG; //!索引类,仅在CObjAllocator中使用,外层一般不用 class CIdx { public: CIdx(); ~CIdx(); public: //!初始化函数 int Initialize(); //!将对象设置为未使用 inline void SetFree() { m_iUseFlag = EIUF_FREE; } //!将对象设置为已使用 inline void SetUsed() { m_iUseFlag = EIUF_USED; } //!判断对象是否已被使用 inline int IsUsed() const { return m_iUseFlag == EIUF_USED; } //!获取所在链表下一个索引 inline int GetNextIdx() const { return m_iNextIdx; } //!设置所在链表下一个索引 inline void SetNextIdx(int iIdx) { m_iNextIdx = iIdx; } //!获取所在链表上一个索引 inline int GetPrevIdx() const { return m_iPrevIdx; } //!设置所在链表上一个索引 inline void SetPrevIdx(int iIdx) { m_iPrevIdx = iIdx; } //!获取指向的对象 inline CObj *GetAttachedObj() const { return m_pAttachedObj; } //!设置指向的对象 inline void SetAttachedObj(CObj *pObj) { m_pAttachedObj = pObj; } private: int m_iNextIdx; //!<对象索引块链表指针,指向后一个闲/忙索引 int m_iPrevIdx; //!<对象索引块链表指针,指向前一个闲/忙索引 int m_iUseFlag; //!<该对象是否已经被使用标志 CObj *m_pAttachedObj; //!<所指向的对象指针 }; typedef CObj *(*Function_CreateObj)(void *); class CObjAllocator { private: //默认构造函数,不允许上层自行调用 CObjAllocator(); public: CObjAllocator(size_t uiObjSize, int iObjCount, Function_CreateObj pfCreateObj); ~CObjAllocator(); /** *使用共享内存创建ObjAllocator *@param[in] pszKeyFileName 共享内存Attach的文件名 *@param[in] ucKeyPrjID 共享内存的工程ID *@param[in] uiObjSize 对象大小 *@param[in] iObjCount 对象个数 *@param[in] *@return 0 success */ static CObjAllocator *CreateBySharedMemory(const char *pszKeyFileName, const unsigned char ucKeyPrjID, size_t uiObjSize, int iObjCount, Function_CreateObj pfCreateObj); /** *指定内存指针来创建CObjAllocator *@param[in] pszMemoryAddress 指定的内存 *@param[in] uiMemorySize 内存大小 *@param[in] uiObjSize 对象大小 *@param[in] iObjCount 对象个数 *@param[in] pfCreateObj 创建CObj对象的函数,默认可以用DECLARE_DYN中的CreateObject *@return 0 success */ static CObjAllocator *CreateByGivenMemory(char *pszMemoryAddress, size_t uiMemorySize, size_t uiObjSize, int iObjCount, Function_CreateObj pfCreateObj); static size_t CountSize(size_t uiObjSize, int iObjCount); static CObjAllocator *ResumeByGivenMemory(char *pszMemoryAddress, size_t uiMemorySize, size_t uiObjSize, int iObjCount, Function_CreateObj pfCreateObj); public: //!初始化函数,将数据清空 int Initialize(); //!创建一个对象,成功返回对象ID,失败返回小于0的值 int CreateObject(); //!创建一个对象, 并指定其ID,成功返回对象ID,失败返回小于0的值 int CreateObjectByID(int iID); //!根据对象ID销毁一个对象,成功返回0 int DestroyObject(int iObjectID); //!根据ObjectID返回对象,必须保证该对象已被使用 CObj *GetObj(int iObjectID); //!根据ObjectID返回索引 CIdx *GetIdx(int iObjectID); //!获取已用对象链表头索引 inline int GetUsedHead() const { return m_iUsedHeadIdx; } //!获取空闲对象链表头索引 inline int GetFreeHead() const { return m_iFreeHeadIdx; } //获取已用对象个数 inline int GetUsedCount() const { return m_iUsedCount; } //获取空闲对象个数 inline int GetFreeCount() const { return m_iObjCount - m_iUsedCount; } //!在接口返回错误时,调用这个函数获取错误号 inline int GetErrorNO() const { return m_iErrorNO; } //获取对象分配类型 inline int GetObjAllocType() const { return m_shObjAllocType; } //!获取下一个对象 CObj *GetNextObj(int iObjectID); //!设置对象初始化指针 inline void SetCreateObjFunc(Function_CreateObj pfCreateObjFunc) { m_pfCreateObjFunc = pfCreateObjFunc; } private: //!设置错误号 inline void SetErrorNO(int iErrorNO) { m_iErrorNO = iErrorNO; } private: //这几个对象只有在构造函数时确定,后面不会更改 short m_shObjAllocType; //!<对象分配类型 size_t m_uiObjSize; //!<单个对象 int m_iObjCount; //!<最多分配的对象个数 CIdx *m_astIdxs; //!<索引数组,用于管理对象链表 CObj *m_pstObjBuffer; //!<分配的对象内存 Function_CreateObj m_pfCreateObjFunc; //!<在内存上创建CObj对象的函数,每个子类需要自己实现 //以下的对象会更改,调用Initialize初始化 int m_iErrorNO; //!<错误码 int m_iFreeHeadIdx; //!<空闲对象链表头索引 int m_iFreeTailIdx; //!<空闲对象链表尾索引 int m_iUsedHeadIdx; //!<已用对象链表头索引 int m_iUsedCount; //!<已用对象个数 }; //!基本对象类 class CObj { public: CObj() {} virtual ~CObj() {} public: //!对象的初始化函数,在CObjAllocator创建对象时会调用,所以子类一定要实现 virtual int Initialize() = 0; //!显示对象函数,可重载 virtual int Show() const { return 0; } //!获取对象ID inline int GetObjectID() const { return m_iObjectID; } //!设置对象ID inline void SetObjectID(int iObjectID) { m_iObjectID = iObjectID; } virtual int Resume() { return 0; } private: int m_iObjectID; //!对象ID,即在CObjAllocator中的数组下标 friend int CObjAllocator::Initialize(); //!<在这个函数中需要直接赋值m_iObjectID,所以设为友元 }; #define DECLARE_DYN public: void *operator new(size_t uiSize, const void *pThis) throw(); static CObj *CreateObject(void *pMem); #define IMPLEMENT_DYN(class_name) void *class_name::operator new(size_t uiSize, const void *pThis) throw() { if (!pThis) { return NULL; } return (void *)pThis; } CObj *class_name::CreateObject(void *pMem) { return (CObj *)new (pMem) class_name; } } // namespace ServerLib #endif //__OBJ_ALLOCATOR_HPP__ ///:~ /** *@file ObjAllocator.cpp *@author jasonxiong *@date 2009-12-14 *@version 1.0 *@brief 对象分配类的实现文件 * * */ #include <assert.h> #include "ErrorDef.hpp" #include "ObjAllocator.hpp" #include "SharedMemory.hpp" using namespace ServerLib; CIdx::CIdx() { Initialize(); } CIdx::~CIdx() { } int CIdx::Initialize() { m_iNextIdx = -1; m_iPrevIdx = -1; m_iUseFlag = 0; m_pAttachedObj = NULL; return 0; } CObjAllocator::CObjAllocator() { } //void* CObjAllocator::operator new(unsigned int uiSize, const void* pThis) throw() //{ // if(!pThis) // { // return NULL; // } // // return (void*)pThis; //} CObjAllocator::CObjAllocator(size_t uiObjSize, int iObjCount, Function_CreateObj pfCreateObj) { __ASSERT_AND_LOG(uiObjSize > 0 && iObjCount > 0 && pfCreateObj); m_shObjAllocType = EOAT_ALLOC_BY_SELF; m_iObjCount = iObjCount; m_uiObjSize = uiObjSize; m_pfCreateObjFunc = pfCreateObj; m_astIdxs = new CIdx[m_iObjCount]; size_t uiObjMemorySize = uiObjSize * iObjCount; char *pstObjMem = new char[uiObjMemorySize]; m_pstObjBuffer = (CObj *)pstObjMem; __ASSERT_AND_LOG(m_astIdxs && m_pstObjBuffer); Initialize(); } size_t CObjAllocator::CountSize(size_t uiObjSize, int iObjCount) { return sizeof(CObjAllocator) + uiObjSize * iObjCount + iObjCount * sizeof(CIdx); } CObjAllocator *CObjAllocator::CreateByGivenMemory(char *pszMemoryAddress, size_t uiMemorySize, size_t uiObjSize, int iObjCount, Function_CreateObj pfCreateObj) { if (pszMemoryAddress == NULL || uiObjSize <= 0 || iObjCount <= 0 || pfCreateObj == NULL) { TRACESVR("%p, %d, %d, %p.\n", pszMemoryAddress, (int)uiObjSize, iObjCount, pfCreateObj); return NULL; } size_t uiSharedMemorySize = sizeof(CObjAllocator) + uiObjSize * iObjCount + iObjCount * sizeof(CIdx); if (uiSharedMemorySize > uiMemorySize) { TRACESVR("ObjAllocator: alloc size %lu > sh size %lu.\n", (unsigned long)uiSharedMemorySize, (unsigned long)uiMemorySize); return NULL; } //在指定的内存地址上分配CObjAllocator CObjAllocator *pstObjAllocator = (CObjAllocator *)pszMemoryAddress; if (!pstObjAllocator) { TRACESVR("ObjAllocator: pstObjAllocator is NULL.\n"); return NULL; } pstObjAllocator->m_uiObjSize = uiObjSize; pstObjAllocator->m_iObjCount = iObjCount; pstObjAllocator->m_pfCreateObjFunc = pfCreateObj; pstObjAllocator->m_shObjAllocType = EOAT_ALLOC_BY_SHARED_MEMORY; pstObjAllocator->m_astIdxs = (CIdx *)((unsigned char *)pszMemoryAddress + sizeof(CObjAllocator)); pstObjAllocator->m_pstObjBuffer = (CObj *)((unsigned char *)pszMemoryAddress + sizeof(CObjAllocator) + iObjCount * sizeof(CIdx)); pstObjAllocator->Initialize(); return pstObjAllocator; } CObjAllocator *CObjAllocator::ResumeByGivenMemory(char *pszMemoryAddress, size_t uiMemorySize, size_t uiObjSize, int iObjCount, Function_CreateObj pfCreateObj) { if ((NULL == pszMemoryAddress) || (uiObjSize <= 0) || (iObjCount <= 0)) { return NULL; } size_t uiSharedMemorySize = sizeof(CObjAllocator) + uiObjSize * iObjCount + sizeof(CIdx) * iObjCount; if (uiSharedMemorySize > uiMemorySize) { return NULL; } CObjAllocator *pstObjAllocator = (CObjAllocator *)pszMemoryAddress; if ((pstObjAllocator->m_uiObjSize != uiObjSize) || (pstObjAllocator->m_iObjCount != iObjCount)) { return NULL; } pstObjAllocator->m_shObjAllocType = EOAT_ALLOC_BY_SHARED_MEMORY; pstObjAllocator->m_astIdxs = (CIdx *)((unsigned char *)pszMemoryAddress + sizeof(CObjAllocator)); pstObjAllocator->m_pstObjBuffer = (CObj *)((unsigned char *)pszMemoryAddress + sizeof(CObjAllocator) + iObjCount * sizeof(CIdx)); int i; // 重新绑定obj和idx for (i = 0; i < iObjCount; ++i) { // 调用placement-new, 恢复类的虚函数表. CObj *pstObj = (*pfCreateObj)((unsigned char *)pstObjAllocator->m_pstObjBuffer + uiObjSize * i); __ASSERT_AND_LOG(pstObj->GetObjectID() == i); pstObjAllocator->m_astIdxs[i].SetAttachedObj(pstObj); } pstObjAllocator->SetCreateObjFunc(pfCreateObj); return pstObjAllocator; } CObjAllocator *CObjAllocator::CreateBySharedMemory(const char *pszKeyFileName, const unsigned char ucKeyPrjID, size_t uiObjSize, int iObjCount, Function_CreateObj pfCreateObj) { if (pszKeyFileName == NULL || uiObjSize <= 0 || iObjCount <= 0 || pfCreateObj == NULL) { return NULL; } CSharedMemory stSharedMemory; size_t uiSharedMemorySize = sizeof(CObjAllocator) + uiObjSize * iObjCount + iObjCount * sizeof(CIdx); int iRet = stSharedMemory.CreateShmSegment(pszKeyFileName, ucKeyPrjID, (int)uiSharedMemorySize); if (iRet) { return NULL; } //在共享内存的地址上分配CObjAllocator CObjAllocator *pstObjAllocator = (CObjAllocator *)stSharedMemory.GetFreeMemoryAddress(); if (!pstObjAllocator) { return NULL; } pstObjAllocator->m_uiObjSize = uiObjSize; pstObjAllocator->m_iObjCount = iObjCount; pstObjAllocator->m_pfCreateObjFunc = pfCreateObj; pstObjAllocator->m_shObjAllocType = EOAT_ALLOC_BY_SHARED_MEMORY; pstObjAllocator->m_astIdxs = (CIdx *)((unsigned char *)stSharedMemory.GetFreeMemoryAddress() + sizeof(CObjAllocator)); pstObjAllocator->m_pstObjBuffer = (CObj *)((unsigned char *)stSharedMemory.GetFreeMemoryAddress() + sizeof(CObjAllocator) + iObjCount * sizeof(CIdx)); return pstObjAllocator; } CObjAllocator::~CObjAllocator() { if (m_shObjAllocType == EOAT_ALLOC_BY_SELF) { if (m_astIdxs) { delete[] m_astIdxs; m_astIdxs = NULL; } if (m_pstObjBuffer) { char *pstObjMem = (char *)m_pstObjBuffer; delete[] pstObjMem; m_pstObjBuffer = NULL; } } } int CObjAllocator::Initialize() { if (m_pstObjBuffer == NULL || m_astIdxs == NULL) { SetErrorNO(EEN_OBJ_ALLOCATOR__NULL_POINTER); return -1; } if (m_iObjCount <= 0) { SetErrorNO(EEN_OBJ_ALLOCATOR__INVALID_OBJ_COUNT); return -2; } //初始化索引数组 int i; for (i = 0; i < m_iObjCount; ++i) { m_astIdxs[i].Initialize(); m_astIdxs[i].SetPrevIdx(i - 1); m_astIdxs[i].SetNextIdx(i + 1); } m_astIdxs[m_iObjCount - 1].SetNextIdx(-1); //初始化对象数组 for (i = 0; i < m_iObjCount; ++i) { CObj *pstObj = (*m_pfCreateObjFunc)((unsigned char *)m_pstObjBuffer + m_uiObjSize * i); pstObj->m_iObjectID = i; m_astIdxs[i].SetAttachedObj(pstObj); } m_iErrorNO = 0; m_iFreeHeadIdx = 0; m_iFreeTailIdx = m_iObjCount - 1; m_iUsedHeadIdx = -1; m_iUsedCount = 0; return 0; } int CObjAllocator::CreateObject() { if (m_pstObjBuffer == NULL || m_astIdxs == NULL) { SetErrorNO(EEN_OBJ_ALLOCATOR__NULL_POINTER); return -1; } if (m_iUsedCount >= m_iObjCount) { SetErrorNO(EEN_OBJ_ALLOCATOR__OBJ_IS_FULL); return -2; } if (m_iFreeHeadIdx < 0 || m_iFreeHeadIdx >= m_iObjCount) { SetErrorNO(EEN_OBJ_ALLOCATOR__INVALID_OBJ_INDEX); return -3; } //修改空闲链表 int iCurIdx = m_iFreeHeadIdx; m_iFreeHeadIdx = m_astIdxs[m_iFreeHeadIdx].GetNextIdx(); if (m_iFreeHeadIdx >= 0) { m_astIdxs[m_iFreeHeadIdx].SetPrevIdx(-1); } if (iCurIdx == m_iFreeTailIdx) { m_iFreeTailIdx = -1; } //挂到使用链表 m_astIdxs[iCurIdx].SetUsed(); m_astIdxs[iCurIdx].SetNextIdx(m_iUsedHeadIdx); m_astIdxs[iCurIdx].SetPrevIdx(-1); if (m_iUsedHeadIdx >= 0) { m_astIdxs[m_iUsedHeadIdx].SetPrevIdx(iCurIdx); } //初始化对象 m_iUsedHeadIdx = iCurIdx; CObj *pstObj = m_astIdxs[iCurIdx].GetAttachedObj(); if (NULL == pstObj) { SetErrorNO(EEN_OBJ_ALLOCATOR__NULL_POINTER); return -4; } #ifdef OBJ_MEMSET_ON_CREATE memset(pstObj, 0, m_uiObjSize); (*m_pfCreateObjFunc)((unsigned char *)pstObj); pstObj->SetObjectID(iCurIdx); #endif pstObj->Initialize(); ++m_iUsedCount; __ASSERT_AND_LOG(pstObj->GetObjectID() == iCurIdx); return iCurIdx; } int CObjAllocator::CreateObjectByID(int iID) { if (m_pstObjBuffer == NULL || m_astIdxs == NULL) { SetErrorNO(EEN_OBJ_ALLOCATOR__NULL_POINTER); return -1; } if (iID < 0 || iID >= m_iObjCount) { SetErrorNO(EEN_OBJ_ALLOCATOR__INVALID_OBJ_INDEX); return -2; } if (m_astIdxs[iID].IsUsed()) { SetErrorNO(EEN_OBJ_ALLOCATOR__INVALID_OBJ_INDEX); return -3; } //修改空闲链表 int iCurIdx = iID; int iPrevIdx = m_astIdxs[iCurIdx].GetPrevIdx(); int iNextIdx = m_astIdxs[iCurIdx].GetNextIdx(); if (iPrevIdx >= 0) { m_astIdxs[iPrevIdx].SetNextIdx(iNextIdx); } if (iNextIdx >= 0) { m_astIdxs[iNextIdx].SetPrevIdx(iPrevIdx); } if (iCurIdx == m_iFreeHeadIdx) { m_iFreeHeadIdx = iNextIdx; } if (iCurIdx == m_iFreeTailIdx) { m_iFreeTailIdx = -1; } //挂到使用链表 m_astIdxs[iCurIdx].SetUsed(); m_astIdxs[iCurIdx].SetNextIdx(m_iUsedHeadIdx); m_astIdxs[iCurIdx].SetPrevIdx(-1); if (m_iUsedHeadIdx >= 0) { m_astIdxs[m_iUsedHeadIdx].SetPrevIdx(iCurIdx); } m_iUsedHeadIdx = iCurIdx; // add by cary CObj *pstObj = m_astIdxs[iCurIdx].GetAttachedObj(); #ifdef OBJ_MEMSET_ON_CREATE memset(pstObj, 0, m_uiObjSize); (*m_pfCreateObjFunc)((unsigned char *)pstObj); pstObj->SetObjectID(iCurIdx); #endif pstObj->Initialize(); ++m_iUsedCount; __ASSERT_AND_LOG(pstObj->GetObjectID() == iCurIdx); return iCurIdx; } int CObjAllocator::DestroyObject(int iObjectID) { if (m_pstObjBuffer == NULL || m_astIdxs == NULL) { SetErrorNO(EEN_OBJ_ALLOCATOR__NULL_POINTER); return -1; } if (iObjectID >= m_iObjCount || iObjectID < 0 || m_iObjCount <= 0) { SetErrorNO(EEN_OBJ_ALLOCATOR__INVALID_OBJ_INDEX); return -2; } if (!m_astIdxs[iObjectID].IsUsed()) { SetErrorNO(EEN_OBJ_ALLOCATOR__DESTROY_FREE_OBJ); return -3; } //从已用链表中删除 int iCurIdx = iObjectID; int iPrevIdx = m_astIdxs[iCurIdx].GetPrevIdx(); int iNextIdx = m_astIdxs[iCurIdx].GetNextIdx(); if (iPrevIdx >= 0) { m_astIdxs[iPrevIdx].SetNextIdx(iNextIdx); } if (iNextIdx >= 0) { m_astIdxs[iNextIdx].SetPrevIdx(iPrevIdx); } if (iCurIdx == m_iUsedHeadIdx) { m_iUsedHeadIdx = iNextIdx; } //挂入空闲链表尾部 m_astIdxs[iObjectID].SetFree(); m_astIdxs[iObjectID].SetPrevIdx(m_iFreeTailIdx); m_astIdxs[iObjectID].SetNextIdx(-1); if (m_iFreeHeadIdx == -1) { m_iFreeHeadIdx = iCurIdx; } if (m_iFreeTailIdx >= 0) { m_astIdxs[m_iFreeTailIdx].SetNextIdx(iCurIdx); } m_iFreeTailIdx = iCurIdx; --m_iUsedCount; return 0; } CObj *CObjAllocator::GetObj(int iObjectID) { if (m_pstObjBuffer == NULL || m_astIdxs == NULL) { SetErrorNO(EEN_OBJ_ALLOCATOR__NULL_POINTER); return NULL; } if (iObjectID < 0 || iObjectID >= m_iObjCount) { SetErrorNO(EEN_OBJ_ALLOCATOR__INVALID_OBJ_INDEX); return NULL; } if (!m_astIdxs[iObjectID].IsUsed()) { SetErrorNO(EEN_OBJ_ALLOCATOR__GET_FREE_OBJ); return NULL; } return m_astIdxs[iObjectID].GetAttachedObj(); } CIdx *CObjAllocator::GetIdx(int iObjectID) { if (m_pstObjBuffer == NULL || m_astIdxs == NULL) { SetErrorNO(EEN_OBJ_ALLOCATOR__NULL_POINTER); return NULL; } if (iObjectID < 0 || iObjectID >= m_iObjCount) { SetErrorNO(EEN_OBJ_ALLOCATOR__INVALID_OBJ_INDEX); return NULL; } if (!m_astIdxs[iObjectID].IsUsed()) { SetErrorNO(EEN_OBJ_ALLOCATOR__GET_FREE_OBJ); return NULL; } return &m_astIdxs[iObjectID]; } CObj *CObjAllocator::GetNextObj(int iObjectID) { CIdx *pIdx = GetIdx(iObjectID); if (!pIdx) { return NULL; } int iNextObjIdx = pIdx->GetNextIdx(); return GetObj(iNextObjIdx); }
源码地址:https://github.com/dai543103/ServerFramework-1/blob/master/001_ServerLib/BaseLibs/ObjAllocator.hpp
标签:inf std 空闲 red 新版 查找 error clu public
原文地址:https://www.cnblogs.com/yzdai/p/11442172.html