模拟实现智能指针 scope_ptr、shared_ptr

转自
http://blog.51cto.com/10740184/1753183

scope_ptr

注意，将拷贝构造函数和拷贝复制运算符设为私有，表示所有权不能转让。

#include<iostream>
using namespace std;

template<class T>
class ScopedPtr
{
public:
    ScopedPtr(T* ptr)
        :_ptr(ptr)
    {}

    Scoped()
        :_ptr(NULL)
    {}

    ~ScopedPtr()
    {
        if (_ptr)
        {
            delete _ptr;
            _ptr = NULL;
        }
    }

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

    T* GetPtr()
    {
        return _ptr;
    }

protected:
    ScopedPtr<T>(const ScopedPtr<T>& sp);
    ScopedPtr<T>& operator = (const ScopedPtr<T>& sp);

private:
    T* _ptr;
};

void Test()
{
    ScopedPtr<int> sp1(new int(2));
    ScopedPtr<int> sp2 = sp1;
    ScopedPtr<int> sp3(new int(3));
    sp3 = sp1;
}

int main()
{
    Test();
    system("pause");
    return 0;
}

shared_ptr

拷贝构造函数中引用计数增加，拷贝复制运算符中，先减去左边的引用计数，然后增加右边的引用计数。

#include<iostream>
using namespace std;

template<class T>
class SharedPtr
{
public:
    SharedPtr(T* ptr)
        :_ptr(ptr)
        , _pCount(new long(1))
    {}

    SharedPtr()
        :_ptr(NULL)
        , _pCount(new long(1))
    {}

    SharedPtr<T>(const SharedPtr<T>& sp)
        : _ptr(sp._ptr)
        , _pCount(sp._pCount)
    {
        ++(*_pCount);
    }

    SharedPtr<T>& operator=(const SharedPtr<T>& sp)
    {
        if (&sp != this)
        {
            if (--(*_pCount) == 0)
            {
                delete _ptr;
                delete _pCount;
            }
            _ptr = sp._ptr;
            _pCount = sp._pCount;
            ++(*_pCount);
        }
        return *this;
    }

    ~SharedPtr()
    {
        if (_ptr)
        {
            if (--(*_pCount) == 0)
            {
                delete _ptr;
                delete _pCount;
            }
        }
    }

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

    long GetCount()
    {
        return *(_pCount);
    }

    T* GetPtr()
    {
        return _ptr;
    }

private:
    T* _ptr;
    long* _pCount;
};


void Test()
{
    SharedPtr<int> sp1 = new int(1);
    SharedPtr<int> sp2 = sp1;
    SharedPtr<int> sp3 = new int(2);
    sp3 = sp1;
}


int main()
{
    Test();
    system("pause");
    return 0;
}