如果我想在new调用出现异常时,捕捉住它而不是由于它而终止程序,那么就需要利用C++提供的set_new_handler函数,内部定义如下:
namespace std
{
typedef void (*new)handler)();
new_handler set_new_handler(new_handler p) throw();
}
为了不改变全局的new_handler可以这样设计:
class Widget
{
public:
static std::new_handler set_new_handler(std::new_handler p) throw();
static void* operator new(std::size_t size) throw(std::bad_alloc);
private:
static std::new_handler handler;
};
std::new_handler Widget::handler = 0;
std::new_handler Widget::set_new_handler(std::new_handler p) throw()
{
std::new_handler old = handler;
handler = p;
return old;
}
在设计operator new之前,先定义一个中间类,利用类析构函数的特性:
class NewHandlerHolder
{
public:
explicit NewHandlerHolder(std::new_handler p):handler(p) {};
~NewHandlerHolder()
{std::set_new_handler(handler);}
private:
std::new_handler handler;
};
operator new 设计如下:
void* Widget::operator new(std::size_t size) throw(std::bad_alloc)
{
NewHandlerHolder tmp(std::set_new_handler(handler));
return ::operator new(size);
}
更一般地,我们可以将Widget泛化:
template<class T>
class Widget
{
public:
static std::new_handler set_new_handler(std::new_handler p) throw();
static void* operator new(std::size_t size) throw(std::bad_alloc);
private:
static std::new_handler handler;
};
template<class T>
std::new_handler Widget<T>::handler = 0;
template<class T>
std::new_handler Widget<T>::set_new_handler(std::new_handler p) throw()
{
std::new_handler old = handler;
handler = p;
return old;
}
template<class T>
void* Widget<T>::operator new(std::size_t size) throw(std::bad_alloc)
{
NewHandlerHolder tmp(std::set_new_handler(handler));
return ::operator new(size);
}
那么,我们这样继承就能生成具备这种能力的类:
class Windows : public Widget<Windows>
{
};
或许你会疑问为什么T模板参数没用到,泛化作用在哪里?
原因来自基类都是静态函数!模板实例化针对不同的形参实例化。