C++ 智能指针简单实现和循环引用问题解决

参考:

https://baike.baidu.com/item/%E6%99%BA%E8%83%BD%E6%8C%87%E9%92%88/10784135?fr=aladdin

https://baike.baidu.com/item/%E5%8F%A5%E6%9F%84%E7%B1%BB/358880?fr=aladdin

https://blog.csdn.net/zhangye3017/article/details/80429780

当类中有指针成员时,一般有两种方式来管理指针成员:一是采用值型的方式管理,每个类对象都保留一份指针指向的对象的拷贝;另一种更优雅的方式是使用智能指针,从而实现指针指向的对象的共享。

智能指针(smart pointer)的一种通用实现技术是使用引用计数(reference count)。智能指针类将一个计数器与类指向的对象相关联,引用计数跟踪该类有多少个对象的指针指向同一对象。

原理

每次创建类的新对象时,初始化指针并将引用计数置为1;当对象作为另一对象的副本而创建时,拷贝构造函数拷贝指针并增加与之相应的引用计数;对一个对象进行赋值时,赋值操作符减少左操作数所指对象的引用计数(如果引用计数为减至0,则删除对象),并增加右操作数所指对象的引用计数;调用析构函数时,析构函数减少引用计数(如果引用计数减至0,则删除基础对象)。

实现代码如下:

// ConsoleApplication.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <stdio.h>
#include<stdlib.h>
#include<iostream> 

class TestP//实际原始对象类
{
public:
	TestP(int a, int b) :m(a), n(b) {}
	int GetM() const { return m; };
	int GetN() const { return n; };
	void SetM(int v) { m = v; };
	void SetN(int v) { n = v; };

private:
	int m, n;
};

class UPoint//引用计数器类 存放实际原始对象和引用
{
private:
	friend class HandlePtr;//声明HandlePtr为UPoint友元类 
	UPoint(TestP *TP) :p(TP), RefCout(1) {} //每次创建类的新对象时,初始化指针并将引用计数置为1;
	~UPoint() { delete p; }
	int RefCout;//引用计数器
	TestP *p;//指向实际原始对象
};

class HandlePtr//智能指针类 存储和管理所指对象 操作UPoint
{
public:
	HandlePtr(TestP *TP) :UP(new UPoint(TP)) {};//智能指针类 默认有参数构造函数 传入实际原始对象TestP
	HandlePtr(const HandlePtr &HP) :UP(HP.UP) { ++UP->RefCout; }//拷贝构造函数 当对象作为另一对象的副本而创建时,拷贝构造函数拷贝指针并增加与之相应的引用计数;
	HandlePtr& operator=(const HandlePtr &R_HP)//对一个对象进行赋值时,赋值操作符减少左操作数所指对象的引用计数(如果引用计数为减至0,则删除对象),并增加右操作数所指对象的引用计数;
	{
		++R_HP.UP->RefCout;
		if (--UP->RefCout == 0)
			delete UP;
		UP = R_HP.UP;
		return *this;
	}
	
	~HandlePtr()//调用析构函数时,析构函数减少引用计数(如果引用计数减至0,则删除基础对象)。
	{
		if (--UP->RefCout == 0)
			delete UP;
		else
		std::cout << "剩余" << UP->RefCout << "个指针指向原始对象" << std::endl;
	}

	//重载* -> 智能指针就可以实际原始对象使用* ->
	TestP& operator*()//重载运算符*
	{
		return *(UP->p);
	}
	TestP* operator->()//重载运算符->
	{
		return UP->p;
	}

private:

	UPoint *UP;//引用计数器类对象,通过管理UPoint 达到管理 实际原始对象TestP
};

int main()
{
	TestP *P = new TestP(5, 6);
	{
		HandlePtr HP1(P);//这时RefCout=1
		{
			HandlePtr HP2(HP1);//这时RefCout=2
			{
				HandlePtr HP3 = HP1;//这时RefCout=3
			}
			//HP3作用域外回收 这时RefCout=2
		}
		//HP2作用域外回收 这时RefCout=1
	}
	//HP1作用域外回收 这时RefCout=0 释放内存

	std::cout << P->GetM() << std::endl;

	TestP *PP = new TestP(10, 11);
	HandlePtr _HP(PP);
	std::cout << _HP->GetM() << std::endl;
	std::cout << (*_HP).GetN() << std::endl;

	system("pause");
    return 0;
}

 

使用模版智能指针类实现如下:

// ConsoleApplication.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <stdio.h>
#include<stdlib.h>
#include<iostream> 

class TestP//实际原始对象类
{
public:
	TestP(int a, int b) :m(a), n(b) {}
	int GetM() const { return m; };
	int GetN() const { return n; };
	void SetM(int v) { m = v; };
	void SetN(int v) { n = v; };

private:
	int m, n;
};

template <typename T>
class HandlePtr;//前置声明

template <typename T>
class UPoint//引用计数器类 存放实际原始对象和引用
{
private:
	friend class HandlePtr<T>;//声明HandlePtr为UPoint友元类 
	UPoint(T *TP) :p(TP), RefCout(1) {} //每次创建类的新对象时,初始化指针并将引用计数置为1;
	~UPoint() { delete p; }
	int RefCout;//引用计数器
	T *p;//指向实际原始对象
};

template <typename T>
class HandlePtr//智能指针类 存储和管理所指对象 操作UPoint
{
public:
	HandlePtr(T *TP) :UP(new UPoint<T>(TP)) {};//智能指针类 默认有参数构造函数 传入实际原始对象TestP
	HandlePtr(const HandlePtr &HP) :UP(HP.UP) { ++UP->RefCout; }//拷贝构造函数 当对象作为另一对象的副本而创建时,拷贝构造函数拷贝指针并增加与之相应的引用计数;
	HandlePtr& operator=(const HandlePtr &R_HP)//对一个对象进行赋值时,赋值操作符减少左操作数所指对象的引用计数(如果引用计数为减至0,则删除对象),并增加右操作数所指对象的引用计数;
	{
		++R_HP.UP->RefCout;
		if (--UP->RefCout == 0)
			delete UP;
		UP = R_HP.UP;
		return *this;
	}
	
	~HandlePtr()//调用析构函数时,析构函数减少引用计数(如果引用计数减至0,则删除基础对象)。
	{
		if (--UP->RefCout == 0)
			delete UP;
		else
		std::cout << "剩余" << UP->RefCout << "个指针指向原始对象" << std::endl;
	}

	//重载* -> 智能指针就可以实际原始对象使用* ->
	T& operator*()//重载运算符*
	{
		return *(UP->p);
	}
	T* operator->()//重载运算符->
	{
		return UP->p;
	}

private:

	UPoint<T> *UP;//引用计数器类对象,通过管理UPoint 达到管理 实际原始对象TestP
};

int main()
{
	TestP *P = new TestP(5, 6);
	{
		HandlePtr<TestP> HP1(P);//这时RefCout=1
		{
			HandlePtr<TestP> HP2(HP1);//这时RefCout=2
			{
				HandlePtr<TestP> HP3 = HP1;//这时RefCout=3
			}
			//HP3作用域外回收 这时RefCout=2
		}
		//HP2作用域外回收 这时RefCout=1
	}
	//HP1作用域外回收 这时RefCout=0 释放内存

	std::cout << P->GetM() << std::endl;

	TestP *PP = new TestP(10, 11);
	HandlePtr<TestP> _HP(PP);
	std::cout << _HP->GetM() << std::endl;
	std::cout << (*_HP).GetN() << std::endl;

	system("pause");
    return 0;
}

上面这种是通过辅佐类UPoint类操作实际原始对象和引用计数器实现,以下方法通过句柄类操作实际原始对象和引用计数器实现代码如下:

// ConsoleApplication.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <stdio.h>
#include<stdlib.h>
#include<iostream> 

class TestP//实际原始对象类
{
public:
	TestP(int a, int b) :m(a), n(b) {}
	int GetM() const { return m; };
	int GetN() const { return n; };
	void SetM(int v) { m = v; };
	void SetN(int v) { n = v; };

private:
	int m, n;
};


template <typename T>
class HandlePtr//智能指针类 存储和管理所指对象 
{
 
public:
	HandlePtr(T *TP) :p(TP), RefCout(new int(1)){};//智能指针类 默认有参数构造函数 传入实际原始对象 并将引用计数置为1
	HandlePtr(const HandlePtr &HP) :p(HP.p), RefCout(HP.RefCout) 
	{
		++(*RefCout);//拷贝构造函数 当对象作为另一对象的副本而创建时,拷贝构造函数拷贝指针并增加与之相应的引用计数;
	}
	HandlePtr& operator=(const HandlePtr &R_HP)//对一个对象进行赋值时,赋值操作符减少左操作数所指对象的引用计数(如果引用计数为减至0,则删除对象),并增加右操作数所指对象的引用计数;
	{
		if (this != &R_HP)
		{
			++(*(R_HP.RefCout));
			if (--(*RefCout) == 0)
			{
				delete p;
				delete RefCout;
			}

			p = R_HP.p;
			RefCout = R_HP.RefCout;

		}
		
		return *this;
	}
	
	~HandlePtr()//调用析构函数时,析构函数减少引用计数(如果引用计数减至0,则删除基础对象)。
	{
		if (--(*RefCout) == 0)
		{
			delete p;
			delete RefCout;
		}
		
		else
		std::cout << "剩余" << *RefCout << "个指针指向原始对象" << std::endl;
	}

	//重载* -> 智能指针就可以实际原始对象使用* ->
	T& operator*()//重载运算符*
	{
		return *(p);
	}
	T* operator->()//重载运算符->
	{
		return p;
	}

private:
	int* RefCout;//引用计数器
	T *p;//指向实际原始对象
};

int main()
{
	TestP *P = new TestP(5, 6);
	{
		HandlePtr<TestP> HP1(P);//这时RefCout=1
		{
			HandlePtr<TestP> HP2(HP1);//这时RefCout=2
			{
				HandlePtr<TestP> HP3 = HP1;//这时RefCout=3
			}
			//HP3作用域外回收 这时RefCout=2
		}
		//HP2作用域外回收 这时RefCout=1
	}
	//HP1作用域外回收 这时RefCout=0 释放内存

	std::cout << P->GetM() << std::endl;

	TestP *PP = new TestP(10, 11);
	HandlePtr<TestP> _HP(PP);
	std::cout << _HP->GetM() << std::endl;
	std::cout << (*_HP).GetN() << std::endl;

	system("pause");
    return 0;
}

智能指针问题:循环引用问题,实现描述循环引用问题代码如下:

// ConsoleApplication.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <stdio.h>
#include<stdlib.h>
#include<iostream> 




template <typename T>
class HandlePtr//智能指针类 存储和管理所指对象 
{
 
public:
	HandlePtr(T *TP) :p(TP), RefCout(new int(1)){};//智能指针类 默认有参数构造函数 传入实际原始对象 并将引用计数置为1
	HandlePtr(const HandlePtr &HP) :p(HP.p), RefCout(HP.RefCout) 
	{
		++(*RefCout);//拷贝构造函数 当对象作为另一对象的副本而创建时,拷贝构造函数拷贝指针并增加与之相应的引用计数;
	}
	HandlePtr& operator=(const HandlePtr &R_HP)//对一个对象进行赋值时,赋值操作符减少左操作数所指对象的引用计数(如果引用计数为减至0,则删除对象),并增加右操作数所指对象的引用计数;
	{
		if (this != &R_HP)
		{
			++(*(R_HP.RefCout));
			if (--(*RefCout) == 0)
			{
				delete p;
				delete RefCout;
			}

			p = R_HP.p;
			RefCout = R_HP.RefCout;

		}
		
		return *this;
	}
	
	~HandlePtr()//调用析构函数时,析构函数减少引用计数(如果引用计数减至0,则删除基础对象)。
	{
		if (--(*RefCout) == 0)
		{
			std::cout << "已经销毁原始对象" << std::endl;
			delete p;
			delete RefCout;
		}
		else
		std::cout << "剩余" << *RefCout << "个指针指向原始对象" << std::endl;
	}

	//重载* -> 智能指针就可以实际原始对象使用* ->
	T& operator*()//重载运算符*
	{
		return *(p);
	}
	T* operator->()//重载运算符->
	{
		return p;
	}

private:
	int* RefCout;//引用计数器
	T *p;//指向实际原始对象
};


class TestP//实际原始对象类
{
public:
	HandlePtr<TestP> _A;
	HandlePtr<TestP> _B;

	TestP()
		:_A(NULL)
		, _B(NULL)
	{}

	~TestP()
	{
		std::cout << "delete TestP" << std::endl;
	}
};

void TestRefCycle()
{
	//A B里成员对象都互相指向,循环引用
	HandlePtr<TestP> A = new TestP();
	HandlePtr<TestP> B = new TestP();
	A->_B = B;
	B->_A = A;
}

int main()
{
	TestRefCycle();
	//在函数TestRefCycle执行后,并没有销毁原始对象,打印日志:
	//剩余1个指针指向原始对象
	//剩余1个指针指向原始对象
	//  A B互相等对方释放 导致最终都没能释放,内存泄漏
	system("pause");
    return 0;
}

解决智能指针循环引用问题:通过弱指针,弱指针仅指向原始对象不操作引用计数(引用计数不增加),这样就解决循环引用没有释放内存问题了,实现代码如下:

// ConsoleApplication.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <stdio.h>
#include<stdlib.h>
#include<iostream> 



template<typename T>
class HandleWeakPtr;//前置声明

template <typename T>
class HandlePtr//智能指针类 存储和管理所指对象 
{
	friend class HandleWeakPtr<T>;//HandleWeakPtr 为 HandlePtr友元类   弱指针HandleWeakPtr可以访问HandlePtr
public:
	HandlePtr(T *TP) :p(TP), RefCout(new int(1)){};//智能指针类 默认有参数构造函数 传入实际原始对象 并将引用计数置为1
	HandlePtr(const HandlePtr &HP) :p(HP.p), RefCout(HP.RefCout) 
	{
		++(*RefCout);//拷贝构造函数 当对象作为另一对象的副本而创建时,拷贝构造函数拷贝指针并增加与之相应的引用计数;
	}
	HandlePtr& operator=(const HandlePtr &R_HP)//对一个对象进行赋值时,赋值操作符减少左操作数所指对象的引用计数(如果引用计数为减至0,则删除对象),并增加右操作数所指对象的引用计数;
	{
		if (this != &R_HP)
		{
			++(*(R_HP.RefCout));
			if (--(*RefCout) == 0)
			{
				delete p;
				delete RefCout;
			}

			p = R_HP.p;
			RefCout = R_HP.RefCout;

		}
		
		return *this;
	}
	
	~HandlePtr()//调用析构函数时,析构函数减少引用计数(如果引用计数减至0,则删除基础对象)。
	{
		if (--(*RefCout) == 0)
		{
			std::cout << "已经销毁原始对象" << std::endl;
			delete p;
			delete RefCout;
		}
		else
		std::cout << "剩余" << *RefCout << "个指针指向原始对象" << std::endl;
	}

	//重载* -> 智能指针就可以实际原始对象使用* ->
	T& operator*()//重载运算符*
	{
		return *(p);
	}
	T* operator->()//重载运算符->
	{
		return p;
	}

private:
	int* RefCout;//引用计数器
	T *p;//指向实际原始对象
};


// 弱指针仅指向原始对象不操作引用计数(不在管理和释放原始对象,引用计数不增加,解决了循环引用的问题)
template<typename T>
class HandleWeakPtr
{
public:
	HandleWeakPtr(HandlePtr<T>& HP)//HandleWeakPtr构造函数 参数传入HandlePtr对象
		:p(HP.p)    //通过参数HandlePtr对象 指向实际原始对象
	{}

	HandleWeakPtr(const HandlePtr<T> &HP)
	{
		p = HP.p;
	}
	HandleWeakPtr& operator=(const HandlePtr<T> &R_HP)
	{
		p = R_HP.p;
		return *this;
	}

	HandleWeakPtr()
		:p(NULL)
	{}

	//重载* -> 弱指针就可以实际原始对象使用* ->
	T& operator*()//重载运算符*
	{
		return *p;
	}
	T* operator->()//重载运算符->
	{
		return p;
	}
protected:
	T * p;
};


class TestP//实际原始对象类
{
public:

	//使用弱指针 不操作引用计数器
	HandleWeakPtr<TestP> _A;
	HandleWeakPtr<TestP> _B;

	~TestP()
	{
		std::cout << "delete TestP" << std::endl;
	}
};


void TestRefCycle()
{
	//A B里成员对象都互相指向,循环引用
	HandlePtr<TestP> A = new TestP();
	HandlePtr<TestP> B = new TestP();
	A->_B = B;
	B->_A = A;
}

int main()
{
	TestRefCycle();
	//在函数TestRefCycle执行后,并没有销毁原始对象,打印日志:
	//已经销毁原始对象
    //delete TestP
	//已经销毁原始对象
    //delete TestP
	//  A B通过弱指针不增加引用计数器 所以都销毁对象 A B了
	system("pause");
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值