c++ SharedPtr 源码简化实现

shared_ptr 本质是是一个模板类,从模板类的角度,简单了解智能指针的实现思想,更好的使用智能指针。记录下简单实现的模板。

template<typename T>
class SharedPtr {
private:
    T* ptr;           // 实际指向对象的指针
    std::size_t* ref_count; // 引用计数的指针

public:
    // 默认构造函数
    explicit SharedPtr(T* p = nullptr) : ptr(p), ref_count(new std::size_t(1)) {}

    // 拷贝构造函数
    SharedPtr(const SharedPtr<T>& sp) : ptr(sp.ptr), ref_count(sp.ref_count) {
        ++(*ref_count);
    }

    // 移动构造函数
    SharedPtr(SharedPtr<T>&& sp) noexcept : ptr(sp.ptr), ref_count(sp.ref_count) {
        sp.ptr = nullptr;
        sp.ref_count = nullptr;
    }

    // 赋值运算符
    SharedPtr<T>& operator=(const SharedPtr<T>& sp) {
        if (this != &sp) {
            if (--(*ref_count) == 0) {
                delete ptr;
                delete ref_count;
            }
            ptr = sp.ptr;
            ref_count = sp.ref_count;
            ++(*ref_count);
        }
        return *this;
    }

    // 移动赋值运算符
    SharedPtr<T>& operator=(SharedPtr<T>&& sp) noexcept {
        if (this != &sp) {
            if (--(*ref_count) == 0) {
                delete ptr;
                delete ref_count;
            }
            ptr = sp.ptr;
            ref_count = sp.ref_count;
            sp.ptr = nullptr;
            sp.ref_count = nullptr;
        }
        return *this;
    }

    // 解引用操作符
    T& operator*() const {
        return *ptr;
    }

    // 成员访问操作符
    T* operator->() const {
        return ptr;
    }

    // 返回裸指针
    T* get() const {
        return ptr;
    }

    // 返回引用计数
    std::size_t use_count() const {
        return *ref_count;
    }

    // 判断是否唯一
    bool unique() const {
        return *ref_count == 1;
    }

    // 重置指针
    void reset(T* p = nullptr) {
        if (--(*ref_count) == 0) {
            delete ptr;
            delete ref_count;
        }
        ptr = p;
        ref_count = new std::size_t(1);
    }

    // 析构函数
    ~SharedPtr() {
        if (--(*ref_count) == 0) {
            delete ptr;
            delete ref_count;
        }
    }
 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值