C++ shared_ptr的实现
在前面放一个源码剖析中看到的auto_ptr的定义文件,补充一点构造函数必须要为explicit,避免一些隐式转换。
补充一点:reset(rhs)函数会删除auto_ptr指向的对象,然后用rhs指向的新建一个
auto_ptr release函数意在将调⽤该函数的智能指针的所有权转移,如 ptr = my_auto2.release ();就是将my_auto2的 所有权转给ptr。
比较完整的复现了STL中的shared_ptr中的功能,可以加载删除器,并且加入了互斥,可以在多线程中使用。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 | #include <iostream> #include <mutex> using namespace std; template < typename T> class Shared_ptr { public : Shared_ptr(T * t_ptrToObject) { m = new mutex(); this ->haveDeleterOrNot = t_ptrToObject; this ->count = malloc ( sizeof ( int )); (* this ->count) = 1; this ->haveDeleterOrNot = false ; } Shared_ptr(T * t_ptrToObject, void (*t_deleter)(T *)) { this ->m = new mutex(); this ->ptrToObject = t_ptrToObject; this ->count = ( int *) malloc ( sizeof ( int )); (* this ->count) = 1; this ->deleter = t_deleter; this ->haveDeleterOrNot = true ; } Shared_ptr( const Shared_ptr & rhs) { this ->m = rhs.m; (*m).lock(); this ->ptrToObject = rhs.ptrToObject; this ->count = rhs.count; (*(rhs.count)) = (*(rhs.count)) + 1; this ->haveDeleterOrNot = rhs.haveDeleterOrNot; this ->deleter = rhs.deleter; (*m).unlock(); } Shared_ptr & operator= ( const Shared_ptr & rhs) { this ->m = rhs.m; (*m).lock(); this ->ptrToObject = rhs.ptrToObject; this ->count = rhs.count; (*(rhs.count)) = (*(rhs.count)) + 1; this ->haveDeleterOrNot = rhs.haveDeleterOrNot; this ->deleter = rhs.deleter; (*m).unlock(); return (* this ); } //rewrite the operator T & operator* () { return *( this ->ptrToObject); } T * operator-> () { return this ->ptrToObject; } ~Shared_ptr() { (*m).lock(); (*count) = (*count) - 1; if ((*count) == 0 && this ->haveDeleterOrNot) { (*( this ->deleter))( this ->ptrToObject); } else if ((*count) == 0 && ! this ->haveDeleterOrNot) { delete this ->ptrToObject; } (*m).unlock(); } int getCount() const { return * this ->count; } private : T * ptrToObject; int * count; bool haveDeleterOrNot; void (*deleter)(T *); mutex * m; }; void test( int * num) { cout << "num: " << *num << endl; } int main() { int * num = new int (5); Shared_ptr< int > sp1(num, test); Shared_ptr< int > sp2(sp1); Shared_ptr< int > sp3 = sp1; cout << sp1.getCount() << endl; // void (*deleter)(int *); // deleter = test; // void (*t_deleter)(int *); // t_deleter = deleter; // (*t_deleter)(num); return 0; } |
分类:
c++学习
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· winform 绘制太阳,地球,月球 运作规律
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· Manus的开源复刻OpenManus初探
· 写一个简单的SQL生成工具
· AI 智能体引爆开源社区「GitHub 热点速览」
2020-12-09 制作光环