模拟share_ptr的实现-01-简单实现

真正的boost::shared_ptr代码很多，这里只是模拟shared_ptr实现类的引用计数。

至于其他的获取指针，判断指针等都很简单了，就不再描述了。

#pragma once

#include <iostream>

#include <assert.h>

using namespace std;

class Test

{

public:

    Test()

    {

        cout<<__FUNCTION__<<endl;

    }

    ~Test()

    {

        cout<<__FUNCTION__<<endl;

    }

};

class test_share_ptr

{

public:

    test_share_ptr(Test* test)

    {

        m_ref_count = new int[1]; // 申请一个内存空间，用于保存test的引用次数

        (*m_ref_count) = 1;

        m_data = test;

        cout<<__FUNCTION__<<"construct--->"<<"m_ref_count = "<< (*m_ref_count) <<endl;

    }

    //拷贝构造函数

    test_share_ptr(const test_share_ptr& src)

    {

        assert( ( *(src.m_ref_count) ) >= 1); // src是有效的

        m_data = src.m_data;

        m_ref_count = src.m_ref_count; // 记录保存的次数

        (*m_ref_count) ++;

        cout<<__FUNCTION__<<"copy construct--->"<<"m_ref_count = "<< (*m_ref_count) <<endl;

    }

    test_share_ptr& operator = (const test_share_ptr& rightvalue)

    {

        assert( ( *(rightvalue.m_ref_count) ) >= 1); // src是有效的

        // 如果是别人的指针传给自己，那么就的先释放自己

        if(rightvalue.m_data != m_data)

        {

            free_one_ref(); // 先释放一次现在的

            // 更新引用并加引用即可

            m_data = rightvalue.m_data;

            m_ref_count = rightvalue.m_ref_count;

            (*m_ref_count) ++;

        }

        cout<<__FUNCTION__<<"operater override--->"<<"m_ref_count = "<< (*m_ref_count) <<endl;

        return *this; // 自己传自己就直接返回

    }

    ~test_share_ptr()

    {

        free_one_ref();

    }

private:

    Test* m_data;

    int* m_ref_count;

    void free_one_ref()

    {

        cout<<__FUNCTION__<<"--->"<<"m_ref_count = "<< (*m_ref_count) <<endl;

        // 自己都不是有效的数据了，异常

        if((*m_ref_count) < 1)

        {

            assert(false);

        }

        // 如果是最后一个引用，直接释放

        if((*m_ref_count) == 1)

        {

            delete m_ref_count;

            if(m_data != NULL)

            {

                delete m_data;

            }

        }

        else

        {

            (*m_ref_count) --; // 减少一次引用

        }

    }

};

测试代码如下：

    {
        Test* t0 = new Test();
        test_share_ptr ptr00(t0);
        test_share_ptr ptr02 = ptr00;
        Test* t1 = new Test();
        test_share_ptr ptr10(t1);
        test_share_ptr ptr12 = ptr10;
        ptr12 = ptr02; // 别人来替换自己
        ptr12 = ptr12; // 自己给自己
    }
    // 测试容器
    {
        vector<test_share_ptr> test_share_ptr_vec;
        test_share_ptr ptr00(new Test());
        test_share_ptr_vec.push_back(ptr00);
        test_share_ptr_vec.push_back(ptr00);
        test_share_ptr_vec.push_back(ptr00);

        ptr00 = test_share_ptr(new Test());
        test_share_ptr_vec.push_back(ptr00);
        test_share_ptr_vec.push_back(ptr00);

        test_share_ptr ptr10(new Test());
        test_share_ptr_vec.push_back(ptr10);
        test_share_ptr_vec.push_back(ptr10);
    }