昨天在讨论需求时提到了google的一款开源软件,突然想起来在深圳工作那半年遇到该软件在搭配CUB库中AutoMsg使用时存在内存泄漏的问题。于是撰写本文对问题进行阐述,并给出解决方案。
cub 已经开源,其中封装了很多有用的组件例如链表,锁以及DCI框架。有兴趣的同学可以浏览下。本文主要探讨其中的一个组件AutoMsg.
AutoMsg主要是解决大对象占用较大栈内内存,可能导致栈溢出的问题。其原理如下:
- 1,构造对象(在全局数据区分配内存---> 构造对象)
- 2,销毁对象 (回收内存)
由上可以看出对象的构造分为两个阶段:分配内存和对象构造。然而在销毁时就只剩内存回收,也就是说缺省了对象析构这个过程。如果存在非默认的析构函数,那么我们设置的扫尾工作可能就没有人做了。例如动态内存释放。
~AutoMsg()
{
if(msg != 0)
{
typedef typename IS_POD<MSG>::value_type type;
_Destroy(getPointer(), type());
msg->operator delete((void*)msg);
msg = 0;
}
}
在最小修改以及不影响原有业务代码的情前提下,我们对原有的代码只是添加了如下两行:
typedef typename IS_POD<MSG>::value_type type;
_Destroy(getPointer(), type());
IS_POD是我们实现的一个类型萃取器,其主要作用是判断当前的数据类型是否需要析构。如果不需要,为了提高效率也就没有必要调用析构函数。_Destroy主要是进行对象析构,其实现如下:
template<typename T> void _Destroy(T* ptr, __false_type)
{
ptr->~T();
}
template<typename T> void _Destory(T* ptr, __true_type){}
综上的修改只是在内存回收的前面调用了下析构函数。当然在此也做了稍许的性能优化,毕竟对于我们来说性能还是很重要的。上述IS_POD类型萃取器的实现,这个比较简单主要就是使用了模板的特化在此就不作展开了。该问题看似实现简单,实际上用到了很多实现技巧,例如模板特化、类型萃取等。有兴趣的同学可以研究下