shared_ptr的实现
时间: 2023-07-18 17:29:32 浏览: 73
shared_ptr
shared_ptr的实现可以分为两部分:引用计数和资源管理。引用计数是指记录有多少个shared_ptr对象共享同一个资源的计数器,当计数器为0时,资源被释放。资源管理是指shared_ptr对象的构造、拷贝、赋值和析构等操作,它们需要保证引用计数的正确性。
下面是shared_ptr的一个简化实现:
```
template <typename T>
class shared_ptr {
public:
shared_ptr() : ptr(nullptr), ref_count(nullptr) {}
shared_ptr(T* p) : ptr(p), ref_count(new size_t(1)) {}
shared_ptr(const shared_ptr<T>& sp) : ptr(sp.ptr), ref_count(sp.ref_count) {
++(*ref_count);
}
~shared_ptr() {
if (--(*ref_count) == 0) {
delete ptr;
delete ref_count;
}
}
shared_ptr<T>& operator=(const shared_ptr<T>& sp) {
if (this != &sp) {
if (--(*ref_count) == 0) {
delete ptr;
delete ref_count;
}
ptr = sp.ptr;
ref_count = sp.ref_count;
++(*ref_count);
}
return *this;
}
T* get() const {
return ptr;
}
size_t use_count() const {
return *ref_count;
}
private:
T* ptr;
size_t* ref_count;
};
```
在这个实现中,我们使用了一个计数器ref_count来记录有多少个shared_ptr对象共享同一个资源。当一个新的shared_ptr对象被创建时,它的计数器被初始化为1,表示它自己是唯一的资源使用者。当有一个新的shared_ptr对象与它共享同一个资源时,它的计数器加1,表示有多了一个资源使用者。当一个shared_ptr对象被销毁时,它的计数器减1,如果计数器变为0,则表示没有任何资源使用者,此时资源被释放。
除了计数器,我们还需要一个指针ptr来指向所管理的资源。在构造函数中,我们需要传入一个指向资源的指针,然后创建一个新的计数器并初始化为1。在拷贝构造函数和赋值操作符中,我们需要共享已有的计数器,因此直接将ref_count指针赋值给新的shared_ptr对象即可。在析构函数中,我们需要判断计数器是否为0,如果为0,则释放资源和计数器。在get()函数中,我们返回指向资源的指针;在use_count()函数中,我们返回计数器的值。
这个实现仅仅是shared_ptr的一个简化版本,实际的shared_ptr实现可能还包括线程安全、自定义删除器等功能。
阅读全文