模拟share_ptr的实现-02-基于模板的实现

最新推荐文章于 2023-03-06 22:02:04 发布

WhyHowWhat

最新推荐文章于 2023-03-06 22:02:04 发布

阅读量647

点赞数

分类专栏： C++

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

版权

C++ 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

这里只是基于《模拟share_ptr的实现-01-简单实现》文章，用模板的方式套了一下而已，

#pragma once

#include <iostream>

#include <assert.h>

using namespace std;

template <class T>

class share_ptr

public:

    share_ptr(T* 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;

    //拷贝构造函数

    share_ptr(const 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;

    share_ptr& operator = (const 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; // 自己传自己就直接返回

    ~share_ptr()

        free_one_ref();

private:

    T* 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) --; // 减少一次引用

};

测试如下：

    {
        share_ptr<Test> ptr = share_ptr<Test>(new Test());
        vector<share_ptr<Test> > share_ptr_vec;
        share_ptr_vec.push_back(ptr);
        share_ptr_vec.push_back(ptr);
        share_ptr_vec.push_back(ptr);
        ptr = ptr;

        share_ptr<Test> ptr2 = share_ptr<Test>(new Test());
        share_ptr_vec.push_back(ptr2);
        ptr2 = ptr;
        share_ptr_vec.push_back(ptr2);
        share_ptr_vec.push_back(ptr);
        return 0;
    }