實現一個簡單的shared_ptr

翻看以前的代碼的時候發現一個shared_ptr的簡單實現。

我記得是網上的一篇例子(好像改了一點)，但是又懶得找出處了 ╮(╯▽╰)╭。

覺得這份代碼足以用來初步了解shared_ptr的實現了。

 

一般來說，智能指針的實現需要以下步驟：

1.一個模板指針T* ptr，指向實際的對象。

2.一個引用次數(必須new出來的，不然會多個shared_ptr里面會有不同的引用次數而導致多次delete)。

3.重載operator*和operator->，使得能像指針一樣使用shared_ptr。

4.重載copy constructor，使其引用次數加一。

5.重載operator=，如果原來的shared_ptr已經有對象，則讓其引用次數減一並判斷引用是否為零(是否調用delete)。

　然后將新的對象引用次數加一。

6.重載析構函數，使引用次數減一並判斷引用是否為零(是否調用delete)。

 

源碼如下:

#ifndef __SHARED_PTR_
#define __SHARED_PTR_

template <typename T>
class shared_ptr {
public:
    shared_ptr(T* p) : count(new int(1)), _ptr(p) {}
    shared_ptr(shared_ptr<T>& other) : count(&(++*other.count)), _ptr(other._ptr) {}
    T* operator->() { return _ptr; }
    T& operator*() { return *_ptr; }
    shared_ptr<T>& operator=(shared_ptr<T>& other)
    {
        ++*other.count;
        if (this->_ptr && 0 == --*this->count)
        {
            delete count;
            delete _ptr;
        }
        this->_ptr = other._ptr;
        this->count = other.count;
        return *this;
    }
    ~shared_ptr()
    {
        if (--*count == 0)
        {
            delete count;
            delete _ptr;
        }
    }
    int getRef() { return *count; }
private:
    int* count;
    T* _ptr;
};


#endif