转自
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;
}