从造轮子做起: shared_ptr

由于 auto_ptr 与 scoped_ptr 都有一些弊端,所以 shared_ptr 出现了。它采用了引用计数的方式来处理拷贝构造与赋值运算。

template <typename T>
class SharedPtr {
public:
    SharedPtr(T* ptr)
        :_ptr(ptr) {
        _count = new size_t(0);
        ++*_count;
    }

    SharedPtr(SharedPtr& s) {
        _ptr = s._ptr;
        _count = s._count;
        ++*_count;
    }

    SharedPtr& operator=(const SharedPtr& s) {
        if (this == &s)
            return;

        if (_ptr != NULL) {
            if (--*_count) {
                delete _count;
                delete _ptr;
                _count = nullptr;
                _ptr = nullptr;
            }
        }
        _ptr = s._ptr;
        _count = s._count;
        ++*_count;
    }

    T* operator->() {
        return _ptr;
    }

    T& operator*() {
        return *_ptr;
    }

    ~SharedPtr() {
        if (_ptr == nullptr)
            return;
        if (!--*count) {
            delete _count;
            delete _ptr;
            _count = nullptr;
            _ptr = nullptr;
        }
    }

private:
    T* _ptr;
    size_t* _count;
};

template <typename T>
class SharedArray {
public:
    SharedArray(T* ptr)
        :_ptr(ptr) {
        _count = new size(0);
        ++*_count;
    }

    SharedArray(const SharedArray& s) {
        _ptr = s._ptr;
        _count = s._count;
        ++*_count;
    }

    SharedArray& operator=(const SharedArray& s) {
        if (this == &s)
            return;

        if (_ptr != nullptr) {
            if (!--*_count) {
                delete _ptr;
                delete _count;
                _ptr = nullptr;
                _count = nullptr;
            }
        }
        _ptr = s._ptr;
        _count = s._count;
        ++*_count;
    }

    T* operator->() {
        return _ptr;
    }

    T& operator*() {
        return *_ptr;
    }

    T& operator[](size_t pos) {
        return _ptr[pos];
    }

    ~SharedArray() {
        if (_ptr == nullptr)
            return;

        if (!--*_count) {
            delete _ptr;
            delete _count;
            _ptr == nullptr;
            _count = nullptr;
        }
    }

private:
    T* _ptr;
    size_t* _count;
};

这里要提一下,由于 shared_ptr 采用引用计数来处理的赋值运算。在某些情况下会出现一些问题。比如:

struct ListNode {
    ListNode() = default;
    shared_ptr<ListNode> _prev;
    shared_ptr<ListNode> _next;
    // weak_ptr<ListNode> _prev;
    // weak_ptr<ListNode> _next;
    ~ListNode() = default;
};

void fun() {
    shared_ptr<ListNode> p1(new ListNode());
    shared_ptr<ListNode> p2(new ListNode());

    p1->_next = p2;
    p2->_prev = p1;
}

这种情况下,就出现了 shared_ptr 的重大问题~~循环引用问题。导致了 shared_ptr 所指向的空间无法被销毁, 那么这个时候就要引入 weak_ptr 来处理这个问题。

weak_ptr 的实质就是只指向,而不增加引用计数。


欢迎大家共同讨论,如有错误及时联系作者指出,并改正。谢谢大家!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值