实现智能指针

原创

mb63887cf57331d 2022-12-01 19:19:07 ©著作权

©著作权归作者所有：来自51CTO博客作者mb63887cf57331d的原创作品，请联系作者获取转载授权，否则将追究法律责任

#include<iostream>
using namespace std;

class point{
public:
    point(int val) :x(val){}
    ~point(){
        cout << "deconstrutor" << endl;
    }
    void set(int val){
        x = val;
    }
    int get(){
        return x;
    }
private:
    int x;
};

//该类成员访问权限全部为private，因为不想让用户直接使用该类
class refPtr{
private:
    //定义智能指针类为友元，因为智能指针类需要直接操纵辅助类
    friend class smartPtr;

    refPtr(point* ptr) :p(ptr),count(1){}

    ~refPtr(){
        delete p;
    }

    int count;
    point* p;
};

class smartPtr{
public:
    smartPtr(point* p):rp(new refPtr(p)){}
    smartPtr(const smartPtr& sp) :rp(sp.rp){
        rp->count++;
    }
    smartPtr& operator=(const smartPtr& sp){
        sp.rp->count++;
        if (--rp->count == 0){
            delete rp;
        }
        rp = sp.rp;
        return *this;
    }
    ~smartPtr(){
        //当引用计数减为0时，删除辅助类对象指针，从而删除基础对象
        if (--rp->count == 0)
            delete rp;
    }

private:
    refPtr* rp;
};

int main(){

    smartPtr sp(new point(3));
    //smartPtr sp2(sp);
    smartPtr p3(new point(4));
    sp = p3;

    return 0;
}