HCppBox/hcppbox/HCPPObjectPool.cpp

#include "HCPPObjectPool.h"
#include "map"
#include <mutex>

namespace  HCPPObjectPoolGlobal
{
extern  std::mutex HCPPObjectPoolLock;
extern  std::map<std::string,HCPPObject *> HCPPObjectPool;
extern  std::map<void *,std::string> HCPPObjectPoolDeleteHelper;
}


/** \brief 删除对象池的对象(一般用于自动删除)
 *
 * \param ptr void* 待删除的指针
 *
 */
void HCPPObjectPool_ObjectDelete(void *ptr)
{
    std::lock_guard<std::mutex> lock( HCPPObjectPoolGlobal::HCPPObjectPoolLock);
    if( HCPPObjectPoolGlobal::HCPPObjectPoolDeleteHelper.find(ptr)!= HCPPObjectPoolGlobal::HCPPObjectPoolDeleteHelper.end())
    {
        std::string id= HCPPObjectPoolGlobal::HCPPObjectPoolDeleteHelper[ptr];
        if( HCPPObjectPoolGlobal::HCPPObjectPool.find(id)!= HCPPObjectPoolGlobal::HCPPObjectPool.end())
        {
            HCPPObjectPoolGlobal::HCPPObjectPool.erase(HCPPObjectPoolGlobal::HCPPObjectPool.find(id));
        }
        HCPPObjectPoolGlobal::HCPPObjectPoolDeleteHelper.erase( HCPPObjectPoolGlobal::HCPPObjectPoolDeleteHelper.find(ptr));
    }
}
void HCPPObjectPool_Set(std::string id,HCPPObject *obj)
{
    if(id.empty() || obj == NULL)
    {
        //不接受id为空或obj为NULL
        return;
    }
    std::lock_guard<std::mutex> lock( HCPPObjectPoolGlobal::HCPPObjectPoolLock);
    HCPPObjectPoolGlobal::HCPPObjectPool[id]=obj;
    HCPPObjectPoolGlobal::HCPPObjectPoolDeleteHelper[obj->GetVoidPtr()]=id;

}

HCPPObject *HCPPObjectPool_Get(std::string id)
{
    std::lock_guard<std::mutex> lock( HCPPObjectPoolGlobal::HCPPObjectPoolLock);
    if( HCPPObjectPoolGlobal::HCPPObjectPool.find(id)!= HCPPObjectPoolGlobal::HCPPObjectPool.end())
    {
        return  HCPPObjectPoolGlobal::HCPPObjectPool[id];
    }
    return NULL;
}

std::string HCPPObjectPool_Id_Helper(std::vector<std::string> id_list)
{
    std::string ret;
    for(size_t i=0; i<id_list.size(); i++)
    {
        if(id_list[i].empty())
        {
            //跳过空子字符串
            continue;
        }
        ret+=id_list[i];
        if((i+1) != id_list.size())
        {
            ret+= ".";
        }
    }
    return ret;
}

std::vector<std::string> HCPPObjectPool_Id_Helper(std::string id)
{
    std::vector<std::string>  ret;
    while(!id.empty())
    {
        std::string::size_type pos=id.find('.');
        if(pos == 0)
        {
            //跳过空子字符串
            id=id.substr(1);
            continue;
        }
        if(pos==std::string::npos)
        {
            ret.push_back(id);
            break;
        }
        else
        {
            ret.push_back(id.substr(0,pos));
            id=id.substr(pos+1);
        }
    }
    return ret;
}

void HCPPObjectPool_Enum(std::function<void(std::string,HCPPObject *)> on_enum)
{
    if(on_enum==NULL)
    {
        return;
    }
    std::lock_guard<std::mutex> lock( HCPPObjectPoolGlobal::HCPPObjectPoolLock);
    for(std::map<std::string,HCPPObject *>::iterator it= HCPPObjectPoolGlobal::HCPPObjectPool.begin(); it!= HCPPObjectPoolGlobal::HCPPObjectPool.end(); it++)
    {
        on_enum(it->first,it->second);
    }
}