C++实现智能指针
一般来说,智能指针的实现需要以下步骤:
1.一个模板指针T* ptr,指向实际的对象。
2.一个引用次数(必须new出来的,不然会多个shared_ptr里面会有不同的引用次数而导致多次delete)。
3.重载operator*和operator->,使得能像指针一样使用shared_ptr。
4.重载copy constructor,使其引用次数加一。
5.重载operator=,如果原来的shared_ptr已经有对象,则让其引用次数减一并判断引用是否为零(是否调用delete)。
然后将新的对象引用次数加一。
6.重载析构函数,使引用次数减一并判断引用是否为零(是否调用delete)。
源码如下:
1 #ifndef __SHARED_PTR_ 2 #define __SHARED_PTR_ 3 4 template <typename T> 5 class shared_ptr { 6 public: 7 shared_ptr(T* p) : count(new int(1)), _ptr(p) {} 8 shared_ptr(shared_ptr<T>& other) : count(&(++*other.count)), _ptr(other._ptr) {} 9 T* operator->() { return _ptr; } 10 T& operator*() { return *_ptr; } 11 shared_ptr<T>& operator=(shared_ptr<T>& other) 12 { 13 ++*other.count; 14 if (this->_ptr && 0 == --*this->count) 15 { 16 delete count; 17 delete _ptr; 18 } 19 this->_ptr = other._ptr; 20 this->count = other.count; 21 return *this; 22 } 23 ~shared_ptr() 24 { 25 if (--*count == 0) 26 { 27 delete count; 28 delete _ptr; 29 } 30 } 31 int getRef() { return *count; } 32 private: 33 int* count; 34 T* _ptr; 35 }; 36 37 38 #endif