如何在C++中实现引用计数智能指针 手写简化版shared_ptr示例

实现一个简化版的c++++引用计数智能指针(类似shared_ptr)的核心步骤包括:设计包含资源指针和引用计数的类,实现构造函数、拷贝构造函数、析构函数和赋值操作符重载,以及提供获取原始指针和解引用的操作符。2. 构造函数初始化资源并设置初始引用计数为1;拷贝构造函数使新对象共享资源并递增引用计数;析构函数递减引用计数并在其变为0时释放资源;赋值操作符处理自我赋值并正确释放旧资源后接管新资源。3. 通过重载*和->操作符以支持指针访问语义,并提供use_count方法查看当前引用计数值。4. 示例代码演示了该智能指针的基本用法及引用计数的变化过程,验证了其核心机制的有效性。

如何在C++中实现引用计数智能指针 手写简化版shared_ptr示例

c++中,实现一个简化版的引用计数智能指针(类似shared_ptr)可以帮助你理解其底层机制。下面会一步步带你写一个基础版本,虽然不考虑线程安全、自定义删除器等复杂情况,但足以展示核心原理。

如何在C++中实现引用计数智能指针 手写简化版shared_ptr示例


基本结构设计

要实现一个带有引用计数的智能指针,我们需要两个关键部分:

如何在C++中实现引用计数智能指针 手写简化版shared_ptr示例

  • 一个资源管理类,用来持有实际的对象指针;
  • 一个引用计数管理类,记录当前有多少个智能指针指向这个资源。

我们可以用一个结构体来保存这些信息:

立即学习C++免费学习笔记(深入)”;

template<typename T> class RefCountedPtr { private:     struct ControlBlock {         T* ptr;         int* ref_count;          ControlBlock(T* p) : ptr(p), ref_count(new int(1)) {}         ~ControlBlock() {             delete ptr;             delete ref_count;         }     };      ControlBlock control_block; };

当然这只是骨架,后面我们会把它封装成更像shared_ptr的形式。

如何在C++中实现引用计数智能指针 手写简化版shared_ptr示例


实现构造与析构逻辑

我们要让多个智能指针共享同一个资源,并在最后一个指针被销毁时释放内存。

构造函数

构造函数负责初始化资源和引用计数:

template<typename T> class SimpleSharedPtr { private:     T* ptr;     int* ref_count;  public:     explicit SimpleSharedPtr(T* p = nullptr)         : ptr(p), ref_count(new int(1)) {}      // ... };

拷贝构造函数

拷贝构造函数需要让新对象指向同一资源,并增加引用计数:

SimpleSharedPtr(const SimpleSharedPtr& other)     : ptr(other.ptr), ref_count(other.ref_count) {     ++(*ref_count); }

析构函数

当引用计数变为0时才真正释放资源:

~SimpleSharedPtr() {     if (--(*ref_count) == 0) {         delete ptr;         delete ref_count;     } }

赋值操作符重载

赋值操作要考虑自我赋值和旧资源释放的问题:

SimpleSharedPtr& operator=(const SimpleSharedPtr& other) {     if (this != &other) {         // 先减少当前引用计数         if (--(*ref_count) == 0) {             delete ptr;             delete ref_count;         }          // 接管对方资源         ptr = other.ptr;         ref_count = other.ref_count;         ++(*ref_count);     }     return *this; }

获取原始指针和解引用操作

为了让我们的智能指针像普通指针一样使用,还需要重载一些操作符:

T& operator*() const { return *ptr; } T* operator->() const { return ptr; }  T* get() const { return ptr; }

示例代码完整版

下面是简化版shared_ptr的一个最小可用实现:

#include <iostream>  template<typename T> class SimpleSharedPtr { private:     T* ptr;     int* ref_count;  public:     explicit SimpleSharedPtr(T* p = nullptr)         : ptr(p), ref_count(new int(1)) {}      SimpleSharedPtr(const SimpleSharedPtr& other)         : ptr(other.ptr), ref_count(other.ref_count) {         ++(*ref_count);     }      SimpleSharedPtr& operator=(const SimpleSharedPtr& other) {         if (this != &other) {             if (--(*ref_count) == 0) {                 delete ptr;                 delete ref_count;             }              ptr = other.ptr;             ref_count = other.ref_count;             ++(*ref_count);         }         return *this;     }      ~SimpleSharedPtr() {         if (--(*ref_count) == 0) {             delete ptr;             delete ref_count;         }     }      T& operator*() const { return *ptr; }     T* operator->() const { return ptr; }      int use_count() const { return *ref_count; } };  int main() {     SimpleSharedPtr<int> p1(new int(42));     std::cout << "Use count: " << p1.use_count() << std::endl; // 输出 1      {         SimpleSharedPtr<int> p2 = p1;         std::cout << "Use count: " << p1.use_count() << std::endl; // 输出 2     }      std::cout << "Use count after p2 destroyed: " << p1.use_count() << std::endl; // 输出 1     return 0; }

基本上就这些了。
这个版本没有考虑线程安全、自定义删除器、数组支持等功能,但在学习理解引用计数机制上已经足够。

以上就是如何在C++中实现引用计数智能指针 手写简化版sha<a

© 版权声明
THE END
喜欢就支持一下吧
点赞12 分享