C++ 模拟智能指针

参考:C++智能指针——探究六个常见的智能指针的使用及原理
模仿C++STL中share_ptr智能指针

  • 将普通指针存在一个类中,类随着作用域的结束,在析构函数中释放指针,所以这个类需要是放在栈区,离开作用域后自动执行析构
  • 类中有一个计数器,防止普通指针被多个类管理,多次释放造成程序崩溃
  • 普通指针只有一个,在多个智能指针之间传递地址,计数器也需要是指针,在多个智能指针之间传递地址,所以他们在多个智能指针对象中始终是一个
#include <iostream>

using namespace std;

template <class T>
class SharePointer
{
public:
    SharePointer(T*p)
    {
        m_ptr = p;
        m_cnt = new int(1);
    }
    ~SharePointer()
    {
        (*m_cnt)--;
        cout << "解析后计数器:" << *m_cnt << endl;
        if(*m_cnt == 0)
        {
            delete m_ptr;
            m_ptr = nullptr;
        }
    }
    explicit SharePointer(SharePointer &sp)
    {
        m_ptr = sp.m_ptr;
        m_cnt = sp.m_cnt;
        (*m_cnt)++;
    }
    SharePointer<T>& operator=(SharePointer<T> &sp)
    {
       if(sp != *this)
       {
           if(--(*m_cnt) == 0) //如果该智能指针是最后一个保存另一个指针,则需要释放之前的资源
           {
              delete m_ptr;
              delete m_cnt;
           }
           m_ptr = sp.m_ptr;
           m_cnt = sp.m_cnt;
           (*m_cnt)++;
       }
       return *this;
    }
    int GetCnt()
    {
        return *m_cnt;
    }
    void *GetCntAdd()
    {
        return m_cnt;
    }
    T* operator*()
    {
        return *m_ptr;
    }
    T* operator->()
    {
        return m_ptr;
    }
private:
    T* m_ptr;
    int *m_cnt;
};

class Test
{
public:
    Test()
    {
        a = 0;
    }
    int GetValue()
    {
        return a;
    }
    void SetValue(int val)
    {
        a = val;
    }
private:
    int a;
};

int main()
{
    {
        SharePointer<Test> sp(new Test);
        cout << "first sp:" << sp.GetCnt() << endl;
        cout << "sp cnt address:" << (void*)sp.GetCntAdd() << endl;
        SharePointer<Test> sp1(sp);
        cout << "sp1 cnt address:" << (void*)sp1.GetCntAdd() << endl;
        cout << "second sp:" << sp.GetCnt() << endl;
        cout << "first sp1:" << sp1.GetCnt() << endl;
    }
    return 0;
}

//结果
first sp:1
sp cnt address:0x7411b0
sp1 cnt address:0x7411b0
second sp:2
first sp1:2
解析后计数器:1
解析后计数器:0

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值