实现智能指针(Smart Pointer)

版权声明:本文为博主原创文章,如有需要,可以随意转载,请注明出处。 https://blog.csdn.net/xunye_dream/article/details/82824857
#ifndef SMARTPTR_H_
#define SMARTPTR_H_

template<typename T>
struct SmartPtr
{
	explicit SmartPtr(T* ptr = nullptr) : dumpPtr(ptr){}
	explicit SmartPtr(const SmartPtr& rhs)
	{
		if (this != &rhs)
		{
			dumpPtr = rhs.operator ->();
		}
	}

	~SmartPtr()
	{
		if (dumpPtr)
		{
			delete dumpPtr;
		}
	}

	SmartPtr& operator=(const SmartPtr& rhs)
	{
		if (this != &rhs)
		{
			if (dumpPtr)
				delete dumpPtr;
			dumpPtr = rhs.operator ->();
		}

		return *this;
	}

	T* operator->() const
	{
		return dumpPtr;
	}

	T& operator*() const
	{
		return *dumpPtr;
	}

private:
	T* dumpPtr;
};

#endif 
#include <iostream>
#include "SmartPtr.h"

using namespace std;

struct Pig
{
	void Print()
	{
		cout << "this is pigsy" << endl;
	}

	~Pig()
	{
		cout << "Kill the pigsy" << endl;
	}
};

int main(int argc, char **argv)
{
	SmartPtr<Pig> pigsy(new Pig());

	pigsy->Print();
	pigsy.operator *().Print();

	return 0;
}

猜你喜欢

转载自blog.csdn.net/xunye_dream/article/details/82824857