首页 > 后端开发 > C++ > 正文

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

P粉602998670
发布: 2025-07-10 14:24:02
原创
179人浏览过

实现一个简化版的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示例

实现构造与析构逻辑

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

知网AI智能写作
知网AI智能写作

知网AI智能写作,写文档、写报告如此简单

知网AI智能写作 38
查看详情 知网AI智能写作

构造函数

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

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++中实现引用计数智能指针 手写简化版shared_ptr示例的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号