自己实现一个智能指针，类比shared_ptr

最新推荐文章于 2023-10-09 14:28:14 发布

不会科研

最新推荐文章于 2023-10-09 14:28:14 发布

阅读量139

点赞数

文章标签： c++

本文链接：https://blog.csdn.net/DD_Huang/article/details/130692354

版权

本文展示了C++中一个简单的智能指针类模板的实现，包括构造函数、拷贝构造函数、运算符重载以及析构函数。智能指针用于自动管理动态分配的对象，确保在不再使用时正确释放内存。示例中，当多个智能指针共享同一个对象时，会通过计数器来跟踪引用计数，防止过早释放内存。

摘要由CSDN通过智能技术生成

包括：

构造函数

拷贝构造函数

运算符重载

析构函数

#include <iostream>
#include <assert.h>

using namespace std;

class Person {
public:

	string m_name;
	int m_age;
	
public:
   
	Person() {
		m_name = "hxx";
		m_age = 24;
		cout << "Person的构造函数" << endl;
	}

	~Person() {
		cout << "Person的析构函数" << endl;
	}
};

template<class T>
class smartPointer {

private:
	T* m_ptr;
	size_t* m_count;

public:
	smartPointer(T* p = nullptr) {
		m_ptr = p;
		if (p == nullptr) {
			m_count = new size_t(0);
		}
		else {
			m_count = new size_t(1);
		}
		cout << "智能指针的构造函数" << endl;
	}

	smartPointer(smartPointer& sp) {
		m_ptr = sp.m_ptr;
		m_count = sp.m_count;
		(*m_count)++;
		cout << "智能指针的拷贝构造函数" << endl;
	}

    T* operator->() {
		assert(m_ptr != nullptr);
		return m_ptr;
    }

	T& operator* () {
		assert(m_ptr != nullptr);
		return *m_ptr;
	}

	smartPointer& operator= (smartPointer& sp) {
		if (m_ptr == sp.m_ptr) return *this;

		(*m_count)--;
		if (m_count == 0)
		{
			delete m_ptr;
			delete m_count;
			cout << "智能指针指向的旧对象已释放" << endl;
		}

		m_count = sp.m_count;
		m_ptr = sp.m_ptr;
		(*m_count)++;
		return *this;

	}

	size_t use_count() {
		return *m_count;
	}

	~smartPointer()
	{
		if (m_ptr == nullptr) {
			delete m_count;
			cout << "智能指针的析构函数：指针为空" << endl;
			return;
		}
		(*m_count)--;
		if (*m_count == 0)
		{
			delete m_ptr;
			delete m_count;
			cout << "智能指针的析构函数：智能指针指向的对象已释放" << endl;
		}
	}
	
};

int main()
{
	smartPointer<Person> sp1(new Person());
	cout << sp1->m_name << endl;
	cout << (*sp1).m_age << endl;
	cout << sp1.use_count() << endl;
	smartPointer<Person> sp2(sp1);
	cout << sp1.use_count() << sp2.use_count() << endl;
	smartPointer<Person> sp3;
	sp3 = sp2;
	cout << sp1.use_count() << sp2.use_count() << sp3.use_count() << endl;
}