从shared_ptr的引用计数实现，聊聊C++11/14/17里原子操作的那些事儿

从shared_ptr的原子引用计数看C并发编程的演进在C的世界里智能指针一直是资源管理的利器。而shared_ptr作为其中最常用的智能指针之一其线程安全性问题一直是开发者关注的焦点。但很少有人深入思考为什么shared_ptr的引用计数是线程安全的这背后隐藏着怎样的并发编程奥秘1. shared_ptr的线程安全机制剖析shared_ptr的核心在于引用计数——这个看似简单的计数器在多线程环境下却需要精心的设计。让我们先看看shared_ptr的基本结构templatetypename T class shared_ptr { T* ptr; // 指向管理对象的指针 control_block* cb; // 控制块包含引用计数等元数据 };1.1 引用计数的原子性实现现代C实现中引用计数通常通过原子操作实现。以下是一个简化的引用计数实现class control_block { std::atomicint ref_count; public: void increment() { ref_count.fetch_add(1, std::memory_order_relaxed); } bool decrement() { return ref_count.fetch_sub(1, std::memory_order_acq_rel) 1; } };这里的关键点在于fetch_add和fetch_sub都是原子操作使用了不同的内存序memory_order来优化性能1.2 为什么资源访问不是线程安全的虽然引用计数是线程安全的但shared_ptr管理的资源本身并不自动具备线程安全性。考虑以下场景std::shared_ptrData shared_data std::make_sharedData(); // 线程A shared_data-modify(); // 线程B shared_data-read();即使引用计数操作是原子的对Data对象的并发访问仍可能导致数据竞争。这就是为什么文档强调多个shared_ptr对象对其所管理的资源的访问不是线程安全的。2. C原子操作的演进历程2.1 C11原子操作的诞生C11首次引入了atomic头文件提供了基本的原子类型和操作std::atomicint counter; counter.store(0, std::memory_order_seq_cst); int val counter.load(std::memory_order_seq_cst);早期的实现通常采用最严格的内存序memory_order_seq_cst虽然安全但性能不是最优。2.2 C14原子操作的优化C14对原子操作进行了多项改进新增了atomic_shared_ptr和atomic_weak_ptr优化了无锁算法的实现引入了更灵活的内存序选项2.3 C17原子操作的成熟C17进一步增强了原子操作新增std::atomic_ref允许对现有对象进行原子操作改进了内存模型使编译器能生成更高效的代码引入了std::atomic_flag的测试操作3. 内存序的深入理解内存序是理解原子操作的关键。C定义了6种内存序内存序描述典型使用场景memory_order_relaxed仅保证原子性不保证顺序计数器等简单场景memory_order_consume保证数据依赖顺序很少使用memory_order_acquire保证该操作后的读操作不会被重排到前面锁获取memory_order_release保证该操作前的写操作不会被重排到后面锁释放memory_order_acq_rel同时具有acquire和release语义读-修改-写操作memory_order_seq_cst完全顺序一致性需要严格顺序的场景在shared_ptr的实现中通常这样使用内存序// 增加引用计数只需要原子性不需要严格顺序 void increment() { ref_count.fetch_add(1, std::memory_order_relaxed); } // 减少引用计数需要确保资源释放发生在所有使用之后 bool decrement() { if(ref_count.fetch_sub(1, std::memory_order_acq_rel) 1) { // 只有最后一个引用被释放时才删除对象 delete ptr; return true; } return false; }4. 实战实现一个线程安全的shared_ptr让我们尝试实现一个简化版的线程安全shared_ptrtemplatetypename T class ThreadSafeSharedPtr { struct ControlBlock { std::atomicint ref_count; T* ptr; ControlBlock(T* p) : ref_count(1), ptr(p) {} void increment() { ref_count.fetch_add(1, std::memory_order_relaxed); } bool decrement() { if(ref_count.fetch_sub(1, std::memory_order_acq_rel) 1) { delete ptr; delete this; return true; } return false; } }; ControlBlock* cb; public: explicit ThreadSafeSharedPtr(T* ptr nullptr) : cb(ptr ? new ControlBlock(ptr) : nullptr) {} ~ThreadSafeSharedPtr() { if(cb) cb-decrement(); } ThreadSafeSharedPtr(const ThreadSafeSharedPtr other) : cb(other.cb) { if(cb) cb-increment(); } ThreadSafeSharedPtr operator(const ThreadSafeSharedPtr other) { if(this ! other) { if(cb) cb-decrement(); cb other.cb; if(cb) cb-increment(); } return *this; } T operator*() const { return *cb-ptr; } T* operator-() const { return cb-ptr; } };这个实现展示了几个关键点引用计数使用原子操作拷贝构造和赋值操作都正确处理了引用计数使用适当的内存序来平衡性能和正确性5. 性能考量与最佳实践在多线程环境下使用shared_ptr时需要注意以下几点避免频繁拷贝原子操作虽然线程安全但仍有性能开销合理使用std::atomic_shared_ptrC20引入的专门为多线程优化的版本考虑使用std::weak_ptr打破循环引用时不会增加引用计数局部变量优先尽量让shared_ptr作为局部变量减少跨线程共享以下是一个性能对比表格展示不同操作的开销操作非原子版本原子版本(seq_cst)原子版本(relaxed)递增1x10-20x5-10x递减1x10-20x5-10x读取1x5-10x1-2x在实际项目中我曾经遇到过一个性能问题一个高频调用的函数中大量使用shared_ptr的拷贝导致性能下降30%。通过改为传递引用并仅在必要时拷贝性能恢复了正常。