智能指针管理第三方库

本节记录一下通过智能指针来管理第三方库内存的分配。

在实际应用中会用到很多别人的lib，当该lib分配内存返回的指针一般是原始指针，其需要手动释放内存。这时候就可以使用智能指针来自动管理分配的内存了。

void* p = GetHandle()->Create();
//do something...
GetHandle()->Release(p);

在实际试用中可能会忘记使用Release，故可以通过使用智能指针的方法来实现，防止出现内存泄露的情况。

//写成一个函数接口如下
std::shared_ptr<void> Guard(void* p)
{
    return std::shared_ptr<void> sp(p,[this](void* p){GetHandle()->Release(p);});
}

void* p = GetHandle()->Create();
auto sp = Guard(p);
//do something...

#include<iostream>
#include<memory>
#include<cstdlib>

class Obj 
{
public:
	Obj() 
	{
		std::cout << "Obj" << std::endl;
	}
	~Obj() 
	{
		std::cout << "~Obj" << std::endl;
	}
};
struct R 
{
	void* create() 
	{
		std::cout << "enter create" << std::endl;
		Obj* i = new Obj();
		std::cout << "create" << std::endl;
		return i;
	}
	void Release(void* p) 
	{
		std::cout << "enter Release" << std::endl;
		Obj* p1 = (Obj*)p;
		delete p1;
		std::cout << "out Release" << std::endl;
	}
	~R() 
	{
		std::cout << "~R" << std::endl;
	}
}r;

R* GetHandle() 
{
	return &r;
}

//不安全，guard2之后可能还会用到p指针，但是其在guard2之后就释放了
void guard2(void* p) 
{
  std::shared_ptr<void> sp(p, [](void* p){GetHandle()->Release(p);});     
}
//ok，p指针的使用权被转移出来，可以继续使用
std::shared_ptr<void> guard(void* p) 
{
	return std::shared_ptr<void>(p, [](void* p) {GetHandle()->Release(p); });
}

//最佳方法
#define GUARD(P) std::shared_ptr<void>p2(p,[](void* p){GetHandle()->Release(p);}); 

int main() 
{
	void* p = GetHandle()->create();
	//guard2(p);
	//std::cout << "out gurad2" << std::endl; 
	//auto p2 = guard(p);
	GUARD(p);
	std::cout << "out test" << std::endl;
}