tars源码分析之20

涛歌依旧

于 2022-07-10 23:19:25 发布

阅读量1k

点赞数 1

分类专栏： S1: C/C++ s2: tarsC++ 文章标签： java 前端大数据

本文链接：https://blog.csdn.net/stpeace/article/details/125712806

版权

S1: C/C++ 同时被 2 个专栏收录

972 篇文章 329 订阅

订阅专栏

s2: tarsC++

78 篇文章 28 订阅

订阅专栏

该博客详细介绍了C++中共享内存库TC_Shm的使用方法，包括构造函数、析构函数以及初始化、分离和删除共享内存的实现。通过示例代码展示了如何创建、连接已有共享内存以及处理可能出现的错误情况，强调了多线程环境下的安全性。

摘要由CSDN通过智能技术生成

共享内存到底怎么用，没搞清楚的话，去面试很尴尬的，直接看源码：

#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;
}

}