智能指针实现

1 篇文章 0 订阅

智能指针

  • 智能指针的使用:

    模拟C++11中的shared_ptr实现一个智能指针,包含初始化,拷贝,赋值等操作。
    采用引用计数的方式来控制指针的析构。

  • 类的定义


template<typename T>
class MySmartPoint 
{
	T* m_ptr;
	size_t* ref_count = 0;

public:

	MySmartPoint(T* ptr = nullptr);
	~MySmartPoint();

	MySmartPoint(const MySmartPoint& s_point);
	MySmartPoint& operator=(const MySmartPoint& s_point);

	size_t GetCount();

	T& operator*();
	T* operator->();
};

  • 类的实现

智能指针的构造函数和析构函数

template<typename T>
inline MySmartPoint<T>::MySmartPoint(T* ptr)
	:m_ptr(ptr)
	,ref_count(new size_t(1))
{
}

template<typename T>
inline MySmartPoint<T>::~MySmartPoint()
{
	if (--(*ref_count) == 0)
	{
		delete m_ptr;
		delete ref_count;
		m_ptr = nullptr;
		ref_count = nullptr;
		std::cout << "引用为0,调用析构函数" << endl;
	}
}

智能指针的拷贝构造和赋值运算符

template<typename T>
MySmartPoint<T>::MySmartPoint(const MySmartPoint& s_point)
{
	if (this != &s_point)
	{
		this->m_ptr = s_point.m_ptr;
		this->ref_count = s_point.ref_count;
		++(*this->ref_count);
	}
}

template<typename T>
MySmartPoint<T>& MySmartPoint<T>::operator=(const MySmartPoint& s_point)
{
	if(m_ptr == s_point.m_ptr)
	{
		return *this;
	}
	else
	{
		if (--(*ref_count) == 0)
		{
			delete m_ptr;
			delete ref_count;
			m_ptr = nullptr;
			ref_count = nullptr;
			std::cout << "引用为0,调用析构函数" << endl;
		}
		m_ptr = s_point.m_ptr;
		ref_count = s_point.ref_count;
		++(*ref_count);
		return *this;
	}
}

智能指针引用计数和 *,->的重载

template<typename T>
size_t MySmartPoint<T>::GetCount()
{
	return *ref_count;
}

template<typename T>
inline T& MySmartPoint<T>::operator*()
{
	if (this->m_ptr)
	{
		return *m_ptr;
	}
}

template<typename T>
inline T* MySmartPoint<T>::operator->()
{
	if (this->m_ptr)
	{
		return m_ptr;
	}
}

测试程序

#include <iostream>
#include "MySmartPoint.h"

using namespace std;

int main()
{
	MySmartPoint<int> p(new int());
	MySmartPoint<int> q(p);
	MySmartPoint<int> x;
	*p = 10;

	cout << p.GetCount() << endl;
	cout << *p << endl;

	p = x;
	cout << x.GetCount() << endl;

	q = x;
	cout << q.GetCount() << endl;
	cout << x.GetCount() << endl;
	{
		MySmartPoint<int> c;
	}

	system("pause");
	return 0;
}

运行结果
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值