C++如何实现一个智能指针_手动实现C++ shared_ptr的引用计数功能

新闻中心 NEWS CENTER

您当前位置：首页 > 新闻中心 > 网络学院

C++如何实现一个智能指针_手动实现C++ shared_ptr的引用计数功能

2025-12-01

浏览次数：次

返回列表

答案：实现简易 shared_ptr 需定义引用计数控制块，通过构造、拷贝、赋值和析构操作管理资源；1. 使用 int* 动态记录引用计数，多个指针共享同一块内存；2. 拷贝时递增计数，析构或赋值时调用 release 函数递减，为 0 则 delete 资源；3. 支持解引用、get、use_count 等接口；4. 示例验证了构造、作用域析构、赋值等场景下引用计数正确变化；5. 关键点包括独立分配计数、自赋值保护、reset 正确释放旧资源；6. 局限性有非线程安全、无自定义删除器、未优化内存布局，生产环境需改进。

c++如何实现一个智能指针_手动实现c++ shared_ptr的引用计数功能

实现一个简易的 shared_ptr 智能指针，核心是模拟引用计数机制，确保多个指针共享同一块内存时，只有最后一个指针释放时才真正 delete 资源。下面是一个手动实现的基本版本，包含构造、拷贝、赋值和析构等关键操作。

1. 定义引用计数控制块

我们需要一个额外的结构体来管理原始指针和引用计数，这个结构体被所有共享该对象的智能指针共用。

template <typename T>
class SharedPtr {
private:
    T* ptr;                    // 指向实际数据的指针
    int* ref_count;            // 指向引用计数的指针
<pre class='brush:php;toolbar:false;'>void release() {
    if (ref_count && --(*ref_count) == 0) {
        delete ptr;
        delete ref_count;
        ptr = nullptr;
        ref_count = nullptr;
    }
}

public: // 构造函数：接管原始指针 explicit SharedPtr(T* p = nullptr) : ptr(p), ref_count(nullptr) { if (ptr) { ref_count = new int(1); } }

// 拷贝构造函数
SharedPtr(const SharedPtr& other) : ptr(other.ptr), ref_count(other.ref_count) {
    if (ref_count) {
        ++(*ref_count);
    }
}

// 拷贝赋值运算符
SharedPtr& operator=(const SharedPtr& other) {
    if (this != &other) {
        release();  // 释放当前资源
        ptr = other.ptr;
        ref_count = other.ref_count;
        if (ref_count) {
            ++(*ref_count);
        }
    }
    return *this;
}

// 析构函数
~SharedPtr() {
    release();
}

// 解引用
T& operator*() const { return *ptr; }
T* operator->() const { return ptr; }

// 获取原始指针
T* get() const { return ptr; }

// 获取引用计数（调试用）
int use_count() const { return ref_count ? *ref_count : 0; }

// 判断是否唯一拥有
bool unique() const { return use_count() == 1; }

// 重置指针
void reset(T* p = nullptr) {
    release();
    ptr = p;
    if (ptr) {
        ref_count = new int(1);
    } else {
        ref_count = nullptr;
    }
}

};

2. 使用示例

测试我们实现的 SharedPtr 是否正确管理引用计数。

#include <iostream>
<p>struct MyClass {
int value;
MyClass(int v) : value(v) { std::cout << "MyClass(" << v << ") created\n"; }
~MyClass() { std::cout << "MyClass destroyed\n"; }
};</p><p>int main() {
SharedPtr<MyClass> sp1(new MyClass(42));
std::cout << "use count: " << sp1.use_count() << "\n";</p><pre class='brush:php;toolbar:false;'>{
    SharedPtr<MyClass> sp2 = sp1;
    std::cout << "use count after copy: " << sp1.use_count() << "\n";
} // sp2 析构，引用计数减一

std::cout << "use count after sp2 destroyed: " << sp1.use_count() << "\n";

SharedPtr<MyClass> sp3;
sp3 = sp1; // 赋值测试
std::cout << "use count after assignment: " << sp1.use_count() << "\n";

return 0;

}