码迷,mamicode.com
首页 > 其他好文 > 详细

定制类自己的的new_handler

时间:2015-06-25 22:40:06      阅读:174      评论:0      收藏:0      [点我收藏+]

标签:

  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。

 

定制类自己的的new_handler

标签:

原文地址:http://www.cnblogs.com/sjinsa/p/4600310.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!