定制类自己的的new_handler

　　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。