C++之手撕引用计数的智能指针

1、实现多个智能指针指向同一个资源,不会造成指针失效,也不会造成资源被多次释放。
2、缺点是多线程环境下,不安全,需要加锁。

#include <iostream>
#include <memory>
using namespace std;

template <typename T>
class RefCnt
{
public:
    RefCnt(T *ptr = nullptr) : mptr(ptr)
    {
        if (mptr != nullptr)
        {
            count = 1;
        }
    }
    ~RefCnt()
    {
        if (count = 0)
        {
            delete mptr;
        }
    }
    int addref()
    {
        return  count++;
    }
    int  cutref()
    {
        return  count--;
    }

private:
    int count;
    T *mptr;
};

template <typename T>
class Smart_ptr
{
public:
    Smart_ptr(T *ptr = nullptr) : Ptr(ptr)
    {
        refcnt = new RefCnt<T>(ptr); //给此处的ptr建立引用计数
    };
    T *operator->()
    {
        return Ptr;
        refcnt = nullptr;
    }
    T &operator*()
    {
        return *Ptr;
    }
    ~Smart_ptr()
    {
         if(0==refcnt->cutref())
        {
            delete Ptr;
        }
        Ptr=nullptr;
    }
    Smart_ptr(const Smart_ptr<T> &src) : Ptr(src), refcnt(src.refcnt)
    {
        if (Ptr != nullptr)
        {
            refcnt->addref();
        }
    }
    Smart_ptr<T> &operator=(const Smart_ptr<T> &src)
    {
        if(this==&src)
        {
            return *this;
        }
        Ptr=src.Ptr;
        refcnt=src.refcnt;
        refcnt->addref();
    }

private:
    T *Ptr;
    RefCnt<T>* refcnt;
};

class test
{

public:
    void fun()
    {
        cout << "fun" << endl;
    }
};

int main()
{
    Smart_ptr<test> ptr(new test);
    Smart_ptr<test> p2;
        Smart_ptr<test> p3;
    p2=ptr;
    p3=p2;
    p2->fun();
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值