ThreadSafeSharedPtr

#include <atomic>

template<typename T>
class ThreadSafeSharedPtr {
public:
    ThreadSafeSharedPtr() : m_ptr(nullptr), m_refCount(nullptr) {}

    ThreadSafeSharedPtr(T* ptr) : m_ptr(ptr), m_refCount(new std::atomic<int>(1)) {}

    ThreadSafeSharedPtr(const ThreadSafeSharedPtr& other) {
        m_ptr = other.m_ptr;
        m_refCount = other.m_refCount;
        if (m_refCount) {
            m_refCount->fetch_add(1);
        }
    }

    ThreadSafeSharedPtr& operator=(const ThreadSafeSharedPtr& other) {
        if (this != &other) {
            reset();
            m_ptr = other.m_ptr;
            m_refCount = other.m_refCount;
            if (m_refCount) {
                m_refCount->fetch_add(1);
            }
        }
        return *this;
    }

    ~ThreadSafeSharedPtr() {
        reset();
    }

    T* get() const {
        return m_ptr;
    }

    void reset() {
        if (m_refCount && m_refCount->fetch_sub(1) == 1) {
            delete m_ptr;
            delete m_refCount;
        }
        m_ptr = nullptr;
        m_refCount = nullptr;
    }

    T& operator*() const {
        return *m_ptr;
    }

    T* operator->() const {
        return m_ptr;
    }

private:
    T* m_ptr;
    std::atomic<int>* m_refCount;
};
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值