shared_ptr的代码实现
时间: 2023-10-08 16:06:55 浏览: 80
shared_ptr是C++11中的一种智能指针,用于管理动态分配的内存资源,它基于引用计数的思想,可以让多个智能指针共享同一个对象,以便有效地避免内存泄漏和重复释放等问题。
以下是shared_ptr的精简版代码实现:
```c++
template <typename T>
class shared_ptr {
public:
// 构造函数
shared_ptr(T* ptr = nullptr) : m_ptr(ptr), m_ref_count(new size_t(1)) {}
// 拷贝构造函数
shared_ptr(const shared_ptr<T>& other) {
m_ptr = other.m_ptr;
m_ref_count = other.m_ref_count;
++(*m_ref_count);
}
// 析构函数
~shared_ptr() {
release();
}
// 获取指针
T* get() const {
return m_ptr;
}
// 重载箭头运算符
T* operator->() const {
return m_ptr;
}
// 重载解引用运算符
T& operator*() const {
return *m_ptr;
}
// 获取引用计数
size_t use_count() const {
return *m_ref_count;
}
// 重载赋值运算符
shared_ptr<T>& operator=(const shared_ptr<T>& other) {
if (this != &other) {
release();
m_ptr = other.m_ptr;
m_ref_count = other.m_ref_count;
++(*m_ref_count);
}
return *this;
}
// 释放资源
void release() {
if (*m_ref_count == 1) {
delete m_ref_count;
delete m_ptr;
} else {
--(*m_ref_count);
}
m_ref_count = nullptr;
m_ptr = nullptr;
}
private:
T* m_ptr; // 指向对象的指针
size_t* m_ref_count; // 引用计数指针
};
```
这个实现中,m_ptr指向被管理的对象,m_ref_count指向引用计数。在构造函数中,m_ptr被赋值为传递进来的指针,m_ref_count被初始化为1。在拷贝构造函数中,m_ptr和m_ref_count都被赋值为other的成员变量,而other的引用计数也被加1。在析构函数中,如果引用计数为1,则删除m_ptr和m_ref_count,否则引用计数减1。在重载赋值运算符中,如果this指针不等于other,则先调用release()方法释放当前对象的资源,然后再将m_ptr和m_ref_count赋值为other的成员变量,并将引用计数加1。
这样一来,就可以通过shared_ptr来管理动态分配的内存资源,达到自动释放的效果。同时,由于shared_ptr基于引用计数的思想,可以让多个智能指针共享同一个对象,以便有效地避免内存泄漏和重复释放等问题。
阅读全文