读书笔记_Effective_C++_条款四十九:了解new_handler的行为

本章开始讨论内存分配的一些用法,C/C++内存分配采用new和delete。在new申请内存时,可能会遇到的一种情况就是,内存不够了,这时候会抛出out of memory的异常。有的时候,我们希望能够调用自己定制的异常处理函数,这就是本条款要说的。

 

在声明于<new>的一个标准程序库中,有如下的接口:

1 namespace std
2 {
3     typedef void (*new_handler)();
4     new_handler set_new_handler(new handler p) throw();
5 }

注意这里面typedef了一个函数指针new_handler,它指向一个函数,这个函数的返回值为void,形参也是void。set_new_handler就是将new_handler指向具体的函数,在这个函数里面处理out of memory异常(函数末尾的throw()表示它不抛出任务异常),如果这个new_handler为空,那么这个函数没有执行,就会抛出out of memory异常。

 1 void MyOutOfMemory()
 2 {
 3     cout << "Out of memory error!" << endl;
 4     abort();
 5 }
 6 
 7 int main()
 8 {
 9     set_new_handler(MyOutOfMemory);
10     int *verybigmemory = new int[0x1fffffff];
11     delete verybigmemory;
12 }

这里预先设定好new异常时调用的函数为MyOutOfMemory,然后故意申请一个很大的内存,就会走到MyOutOfMemory中来了。

 

好,我们更进一步,现在想要在不同的类里面定制不同的new_handler处理机制,一种想法是在类内部定义set_new_handler函数,将new_handler作为私有的成员变量,具体的new_handler函数可以由构造函数传入,但编译器要求set_new_handler是静态的,所以通过构造函数传入new_handler不被编译器支持,只能将set_new_handler与operator new都写成静态的,同时定义一个静态的new_handler变量,像下面这样:

 1 class Widget
 2 {
 3 private:
 4     static new_handler CurrentHandler;
 5 
 6 public:
 7     void set_new_handler(new_handler h) throw()
 8     {
 9         CurrentHandler = h;
10     }
11     static void* operator new(size_t size)
12     {
13         Widget::set_new_handler(CurrentHandler);
14         return ::operator new(size);
15     }
16 };
17 
18 new_handler Widget::CurrentHandler = 0;

 

属于类的静态变量CurrentHandler用于保存当前环境下的new_handler函数,在operator_new中,先设置成当前的new异常处理函数,再去调用std的operator new,执行内存分配操作。但这里就存在问题了,set_new_handler到下一次设置它为止,一直都是生效的,我们只想在处理这个类对象的分配时用自定义的new_handler函数,但是类似于new int,new char这些基本类型,还是希望走默认的new_handler(就是null,就是什么也不执行,如我们期望,这样会抛出异常)。

一种自然的想法,就是在调用operator new末尾处还原new_handler,这就需要保存之前的new_handler,为此,我们构造一个NewHandlerHolder类,像下面这样:

 1 class NewHandlerHolder
 2 {
 3 private:
 4     new_handler SavedHandler;
 5     NewHandlerHolder(const NewHandlerHolder&);
 6     NewHandlerHolder& operator= (const NewHandlerHolder&);
 7 
 8 public:
 9     explicit NewHandlerHolder(new_handler h) :SavedHandler(h){}
10     ~NewHandlerHolder()
11     {
12         set_new_handler(SavedHandler);
13     }
14 };

这里有一个SavedHandler成员变量,它在NewHandlerHolder构造时确定具体的指向(其实就是指向系统默认的new_handler函数(即null),将拷贝构造函数与赋值运算符重载设置为private是为了防止出现拷贝的行为(在编译期就可以阻止),这点可以参照之前的条款。还要特别注意这里的析构函数,它调用了set_new_handler,将new异常的处理恢复成SavedHandler(其实就是null)。

 

这样我们重新整理一下Widget,如下:

 1 class Widget
 2 {
 3 private:
 4     static new_handler CurrentHandler;
 5 
 6 
 7 public:
 8     static new_handler set_new_handler(new_handler h) throw()
 9     {
10         new_handler OldHandler = CurrentHandler;
11         CurrentHandler = h;
12         return OldHandler;
13     }
14     static void* operator new(size_t size)
15     {
16         NewHandlerHolder h(Widget::set_new_handler(CurrentHandler));
17         return ::operator new(size);
18     }
19 };
20 
21 new_handler Widget::CurrentHandler = 0;

为了返回系统默认的new_handler,我们在set_new_handler处理完之后,进行了旧handler的返回,同时在operator new的调用中进行了NewHandlerHolder的包装,这样在return之后,h会自动调用析构函数,恢复成默认的new_handler。

 

到这一步,本条款的重要内容已经说完了,但为了避免重复劳动,即为每一个需要重定义new_handler的类都写一份set_new_handler和operator new,书上在最后对之进行了封装,其实就是将Widget专门作为这两个函数(set_new_handler和operator new)的类,然后将需要自行处理new_handler的类public继承于Widget即可。

 

最后总结一下:

set_new_handler允许客户指定一个函数,在内存分配无法获得满足时被调用。

posted @ 2014-05-11 23:15  Jerry19880126  阅读(2940)  评论(0编辑  收藏  举报