两种智能指针的简单实现

C++11的智能指针确实带来了很多方便,但是shared_ptr因为要维护很多元信息,会牺牲一些效率,如果大量使用智能指针,一定不要忘记开启编译器的优化选项。。

下面是仿照unique_ptr和shared_ptr的实现。

unique_ptr

#pragma once

namespace m{

template<typename T>
struct DefaultDeleter{
    void operator()(T *ptr){
        if(ptr!=nullptr){
            delete ptr;
            ptr = nullptr;
        }
    }
};

template<typename T,typename Deleter = DefaultDeleter<T>>
class UniquePtr{
    T *ptr;
    Deleter deleter;
    
    // delete copy constructor and operator=
    UniquePtr(const UniquePtr &){}
    UniquePtr &operator=(const UniquePtr &){}
public:
    UniquePtr():ptr(nullptr){}
    UniquePtr(T *_ptr):ptr(_ptr){}
    ~UniquePtr(){ deleter(ptr); } 

    //replaces the managed object 
    void reset(T *_ptr);
    //returns a pointer to the managed object and releases the ownership
    T *release();
    //returns a pointer to the managed object
    T *get();

    operator bool() const;
    T &operator*();
    T *operator->();
};

template<typename T,typename Deleter>
void UniquePtr<T,Deleter>::reset(T *_ptr){
    if(ptr==_ptr) return ;
    T *old_ptr = ptr;
    ptr = _ptr;
    deleter(old_ptr);
}

template<typename T,typename Deleter>
T *UniquePtr<T,Deleter>::release(){
    T *old_ptr = ptr;
    ptr = nullptr;
    return old_ptr;
}

template<typename T,typename Deleter>
T *UniquePtr<T,Deleter>::get(){
    return ptr;
}

template<typename T, typename Deleter>
UniquePtr<T,Deleter>::operator bool() const{
    return ptr!=nullptr;
}

template<typename T, typename Deleter>
T &UniquePtr<T,Deleter>::operator*(){
    return *ptr;
}

template<typename T, typename Deleter>
T *UniquePtr<T,Deleter>::operator->(){
    return ptr;
}

}

shared_ptr

#pragma once
namespace m{

template<typename T>
struct DefaultDeleter{
    void operator()(T *ptr){
        if(ptr!=nullptr){
            delete ptr;
            ptr = nullptr;
        }
    }
};

template<typename T,typename Deleter>
struct Control{
    int reference_count;
    Deleter deleter;
    Control():reference_count(1){}
};

template <typename T,typename Deleter = DefaultDeleter<T>>
class SharedPtr{
    T *ptr;
    Control<T,Deleter> *ctrl;
public:

    SharedPtr():ptr(nullptr),ctrl(new Control<T,Deleter>()){}
    SharedPtr(T *_ptr):ptr(_ptr),ctrl(new Control<T,Deleter>()){}
    SharedPtr(const SharedPtr<T> &other):ptr(other->ptr),ctrl(other->ctrl){
        ctrl->reference_count++;
    }
    ~SharedPtr() {
        ctrl->reference_count--;
        if(ctrl->reference_count == 0){
            ctrl->deleter(ptr);
            delete ctrl;
            ctrl = nullptr;
        }
    }
    T &operator*(){
        return *ptr;
    }
    T *operator->(){
        return ptr;
    }
    operator bool() const{
        return ptr!=nullptr;
    }
};
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 1. 什么是智能指针? 智能指针是行为类似于指针的类对象,但这种对象还有其他功能。 2. 为什么设计智能指针? 引...
    MinoyJet阅读 646评论 0 1
  • 导读## 最近在补看《C++ Primer Plus》第六版,这的确是本好书,其中关于智能指针的章节解析的非常清晰...
    小敏纸阅读 2,020评论 1 12
  • 原作者:Babu_Abdulsalam 本文翻译自CodeProject,转载请注明出处。 引入### Ooops...
    卡巴拉的树阅读 30,162评论 13 74
  • C++ 智能指针详解 一、简介由于 C++ 语言没有自动内存回收机制,程序员每次 new 出来的内存都要手动 de...
    yangqi916阅读 1,386评论 0 2
  • C++智能指针 原文链接:http://blog.csdn.net/xiaohu2022/article/deta...
    小白将阅读 6,889评论 2 21