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

最新推荐文章于 2023-10-08 22:42:11 发布

WhyHowWhat

最新推荐文章于 2023-10-08 22:42:11 发布

阅读量1.1k

点赞数

分类专栏： C++

本文链接：https://blog.csdn.net/whyhowwhat/article/details/39027093

版权

C++ 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

真正的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);
    }