共享内存到底怎么用,没搞清楚的话,去面试很尴尬的,直接看源码:
#include "util/tc_shm.h"
#include <cassert>
#include <errno.h>
namespace tars
{
TC_Shm::TC_Shm(size_t iShmSize, key_t iKey, bool bOwner)
{
init(iShmSize, iKey, bOwner);
}
TC_Shm::~TC_Shm()
{
if(_bOwner)
{
detach();
}
}
void TC_Shm::init(size_t iShmSize, key_t iKey, bool bOwner)
{
assert(_pshm == NULL);
_bOwner = bOwner;
//注意_bCreate的赋值位置:保证多线程用一个对象的时候也不会有问题
//试图创建
if ((_shemID = shmget(iKey, iShmSize, IPC_CREAT | IPC_EXCL | 0666)) < 0)
{
_bCreate = false;
//有可能是已经存在同样的key_shm,则试图连接
if ((_shemID = shmget(iKey, iShmSize, 0666)) < 0)
{
throw TC_Shm_Exception("[TC_Shm::init()] shmget error", errno);
}
}
else
{
_bCreate = true;
}
//try to access shm
if ((_pshm = shmat(_shemID, NULL, 0)) == (char *) -1)
{
throw TC_Shm_Exception("[TC_Shm::init()] shmat error", errno);
}
_shmSize = iShmSize;
_shmKey = iKey;
}
int TC_Shm::detach()
{
int iRetCode = 0;
if(_pshm != NULL)
{
iRetCode = shmdt(_pshm);
_pshm = NULL;
}
return iRetCode;
}
int TC_Shm::del()
{
int iRetCode = 0;
if(_pshm != NULL)
{
iRetCode = shmctl(_shemID, IPC_RMID, 0);
_pshm = NULL;
}
return iRetCode;
}
}