条款7:预先准备好内存不够的情况

1. operator new在无法完成内存分配请求时会抛出异常(在以前或者某些特定情况(声明nothrow)下new会返回0)

2. 使用C语言的宏定义处理会是这样

#define NEW(PTR, TYPE)							\
	try {(PTR) = new TYPE; }					\
	catch (std::bad_alloc&) { assert(0); }

3. 用C++的异常处理机制,处理方法是这样

#include <iostream>
#include <new>

using namespace std;

/*
#define NEW(PTR, TYPE)							\
	try {(PTR) = new TYPE; }					\
	catch (std::bad_alloc&) { assert(0); }
*/

//new异常处理模版类
template <class T>
class NewHandlerSupport
{
public:
	static new_handler set_new_handler(new_handler handler);
	static void* operator new(size_t size);
private:
	static new_handler current_handler;
};

//设置异常处理函数
template <class T>
new_handler NewHandlerSupport<T>::set_new_handler(new_handler handler)
{
	new_handler old_handler = current_handler;
	current_handler = handler;
	return old_handler;
}

//重写new
template<class T>
void* NewHandlerSupport<T>::operator new(size_t size)
{
	void* memory = NULL;
	new_handler old_handler = ::set_new_handler(current_handler);

	try
	{
		memory = ::operator new(size);
	}
	catch (std::bad_alloc&)
	{
		set_new_handler(old_handler);
		throw;
	}

	set_new_handler(old_handler);
	return memory;
}

//初始化静态数据
template<class T>
new_handler NewHandlerSupport<T>::current_handler;

//具体类继承模版实例
class Example : public NewHandlerSupport<Example>
{
public:
	//...
private:
	//...
};

int main()
{
	Example* e = new Example;
	return 0;
}

4. 如果想让new分配失败不抛出异常而是返回0,可以像下面这样

int* i = new (nothrow) int;


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值