给shared_ptr添加自定义删除器的几种方式

将回调函数传递给 shared_ptr 的构造函数,该构造函数将从其析构函数中调用以进行删除

方式一、使用普通函数

// 自定义删除器
void deleter(Sample * x)
{
	std::cout << "Deleter function called" << std::endl;
	delete[] x;
}
// 构造函数传递自定义删除器指针
std::shared_ptr<Sample> p1(new Sample[5], deleter);

方式二、使用仿函数

class Deleter
{
public:
	void operator() (Sample *x) {
		std::cout << "Deleter function called" << std::endl;
		delete[] x;
	}
};
// 构造函数传递自定义删除器指针
std::shared_ptr<Sample> p2(new Sample[5], Deleter);

方式三、使用lambda表达式

std::shared_ptr<Sample> p3(new Sample[5], [](Sample *x) {
	std::cout << "Deleter function called" << std::endl;
	delete[] x;
});

方式四、使用std::default_delete

std::shared_ptr<Sample> p(new Sample[5], std::default_delete<Sample[]>());

举例:

#include <iostream>
#include <memory>

class Sample {
public:
	Sample() {
		std::cout << "Sample Constructor" << std::endl;
	}
	~Sample() {
		std::cout << "~Sample Destrutor" << std::endl;
	}
};

void deleter(Sample *x)
{
	std::cout << "Deleter function called" << std::endl;
	delete[] x;
}

class Deleter
{
public:
	void operator() (Sample *x) {
		std::cout << "Deleter function called" << std::endl;
		delete[] x;
	}
};

int main()
{
	{
		std::shared_ptr<Sample> p0(new Sample[5]);
		//std::shared_ptr<Sample> p1(new Sample[5], deleter);		//方式一
		//std::shared_ptr<Sample> p2(new Sample[5], Deleter());		//方式二
		//std::shared_ptr<Sample> p3(new Sample[5], [](Sample *x) {  //方式三
		//	std::cout << "Deleter function called" << std::endl;
		//	delete[] x;
		//});
		//std::shared_ptr<Sample> p(new Sample[5], std::default_delete<Sample[]>()); //方式四
	}
	
	getchar();
	return 0;
}

运行结果:

Sample Constructor
Sample Constructor
Sample Constructor
Sample Constructor
Sample Constructor
~Sample Destrutor

只看到一个析构函数,这时候已经发生了内存泄露,数组中的5个对象,只有一个发生了析构。

int main()
{
	{
		std::shared_ptr<Sample> p1(new Sample[5], deleter);		//方式一
	}
	
	getchar();
	return 0;
}

此时的运行结果:

Sample Constructor
Sample Constructor
Sample Constructor
Sample Constructor
Sample Constructor
Deleter function called
~Sample Destrutor
~Sample Destrutor
~Sample Destrutor
~Sample Destrutor
~Sample Destrutor

发现p1析构时候,会自动调用deleter(你构造的时候传进去的),对数组中的所有元素一一进行删除。此时没有内存泄露。

修改为这样也是一样的结果,只是调用的的Deleter中的operator() (Sample *x)函数

std::shared_ptr<Sample> p2(new Sample[5], Deleter());		//方式二

采用lambda函数的时候,就是:

std::shared_ptr<Sample> p3(new Sample[5], [](Sample *x) {  //方式三
			std::cout << "Deleter function called" << std::endl;
			delete[] x;
		});

使用std::default_delete作为删除器,default_delete的内部是通过调用delete来说实现功能的。

std::shared_ptr<int> p(new int[10], [](int *p) {delete[] p; });

也可以写成

std::shared_ptr<int> p(new int[5], std::default_delete<int[]>());

封装一个make_shared_array方法来让shared_ptr支持数组

template<typename T>
std::shared_ptr<T> make_shared_array(size_t size)
{
	return std::shared_ptr<T>(new T[size], std::default_delete <T[]>());
}

测试代码:

std::shared_ptr<Sample> p5 = make_shared_array<Sample>(5);
std::shared_ptr<int> p5 = make_shared_array<int>(5);

参考:
C++ 智能指针 shared_ptr 详解与示例
std::default_delete官方文档和实例
智能指针拾遗

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值