C++中的new操作符首先使用operator new函数来分配空间,然后再在此空间上调用类的构造函数构造对象。当operator new无法分配所需的内存空间时,默认的情况下会抛出一个bad_alloc异常,在抛出这个异常之前,如果用户指定了错误处理函数即new_handler,则程序会先执行new_handler函数进行错误处理。为了设置这个错误处理函数,我们需要调用set_new_handler函数,它在std命名空间内的情况如下所示
namespace std{ typedef void (*new_handler)(); new_handler set_new_handler(new_handler p)throw (); }
set_new_handler函数承若不抛出任何异常。该函数接受一个new_handler函数指针,且返回一个这样的指针。pNew代表的是如果operator new发生失败,采用的错误处理函数是pNew所指向的函数,set_new_handler函数返回的是以前的错误处理函数指针。比如,
void myErrorHandler(){ cerr<<"can‘t need the request for memory"; abort(); } int main(){ std::set_new_handler(myErrorHandler); int *ptr=new int[1000000000L]; delete [] ptr; return ; }
当出现无法满足内存分配要求的时候,程序会不断的调用myErrorHandler函数进行内存的分配,如果错误处理函数中没有abort(),终端会不断的打印错误提示。设置全局的set_new_handler,只要出现operator new失败,都会调用该错误处理函数。但是如果我们需要给不同的类设置不同的错误处理函数时,该怎么办呢?也就是说我们需要定制类自己的set_new_handler函数。
class Widget{ public: static std::new_handler set_new_handler(std::new_handler pNew) throw (); static void* operator new(std::size_t size)throw (std::bad_alloc); private: static std::new_handler currentHandler; int array[100000000L];//为了容易让new Widget出现异常 };std::new_handler Widget::currentHandler=0;
如上所示,类Widget需要定义自己的new_handler变量,而且需要定义自己的set_new_handler函数和operator new函数,set_new_handler函数比较简单,参照全局的set_new_handler函数的行为即可。
std::new_handler Widget::set_new_handler(std::new_handler pNew){ std::new_handler oldHandler=currentHandler; currentHandler=pNew; return oldHandler; }
函数set_new_handler设置新的new_handler,并且返回以前的new_handler。
那么Widget类中的operator new函数应该怎么定义呢?
首先,它应该设置本类的new_handler,怎么给Widget设置自己的new_handler并让其正常工作呢?调用std::set_new_handler即可,类中的currentHandler可作为其参数;
其次,调用std::operator new函数请求分配内存,但是,这里可能存在一个资源泄露的问题,如第一步,我们将全局的set_new_handler设置成类Widget的错误处理函数,如果分配失败,我们怎么将原来的new_handler再设置回去呢?很显然的一种方法就是通过对象来管理new_handler,如下
class NewHandlerHolder{ public: explicit NewHandlerHolder(new_handler p):oldHandler(p){} ~NewHandlerHoler(){ std::set_new_handler(oldHander); } private: std::new_handler oldHandler; NewHandlerHolder(const NewHandlerHolder&); NewHandlerHolder& operator=(const NewHandlerHolder&); };
这是一个资源管理类,如果我们定义了一个NewHandlerHolder对象来管理原来的new_handler,当内存分配失败时,抛出异常且函数退栈,同时也会调用已经构造的对象的析构函数,当我们调用NewHandlerHolder的析构函数时,会重新将原来的new_handler设置回去。用于资源管理的类在较多情况下不允许进行赋值和复制构造的动作,因为我们希望这个资源管理的对象具有资源唯一的管理权。
此时,我们就可以写出来Widget类中的operator new函数了。
void* Widget::operator new(std::size_t size) throw (std::bad_alloc){ NewHandlerHolder temp(std::set_new_handler(currentHandler)); return ::operator new(size); }
定义完所有所需的内容后,我们可以使用定制的new_handler。
int main(){ Widget::set_new_handler(myErrorHandler); //Widget::set_new_handler(0); while(1) Widget *ptr=new Widget; }
运行结果如下图
将注释的解注,运行结果没有"can‘t need the request for memory"。
以上思想就能为各个类编写自己的定制的new_handler,但是为每个需要编写定制new_handler的类写如此相似的代码,似乎有点不合理,为了避免代码重复,我们可以使用模板
template<typename T> class NewHandlerSupport{ public: static std::new_handler set_new_handler(std::new_handler newHandler)throw (); static void* operator new(std::size_t size)throw (std::bad_alloc); private: static std::new_handler currentHandler; }; template<typename T> void* NewHandlerSupport<T>::operator new(std::size_t size){ NewHandlerHolder temp(std::set_new_handler(currentHandler)); return ::operator new(size); } template<typename T> std::new_handler NewHandlerSupport<T>::set_new_handler(std::new_handler newHandler)throw(){ std::new_handler oldHander=currentHandler; currentHandler=newHandler; return oldHandler; } template<typename T> std::new_handler NewHandlerSupport<T>::currentHandler=0;
如果我们希望类widget获得定制的new_handler行为,只需要将widget类以public继承NewHandlerSupport<widget>即可。我们只是希望获得不同模板参数所带来的不同的static std::new_handler currentHandler,我们需要用它来定制各个类的new_handler。