编程的时候经常会遇到这样的问题:由于要分析一段数据,我们需要new一段内存出来做临时内存区域。但是分析的过程比较复杂,后面很多地方都需要终止分析。于是我们只能在每个return前面都加上delete。后面代码维护的时候就非常麻烦,很可能一个疏忽就导致内存块被重复delete(比如原来某个地方需要return,后来加入新规则,就转而进行其他处理。等跳出循环后,忘记了之前统一还有一个delete,然后就悲剧了;或者添加了新的return分支,但是忘记了delete)
有没有什么办法,可以避免这种情况?
答案当然是有!那就是建立一个TempBuf的模板类!
先不废话,上代码:
#pragma once
template<typename T>
class TempBuf
{
public:
TempBuf(size_t nSize)
{//禁止默认构造函数,避免空buf的情况
m_Size = 0;
m_Buf = new T[nSize+1];
if(m_Buf){
memset(m_Buf,0,sizeof(T)*(nSize+1));
m_Size = nSize;
}
}
~TempBuf(void)
{//析构
empty();
}
T& operator[](int nIndex)
{//允许取指定元素,并且防止溢出
if(nIndex>m_Size || (nIndex<0)){
nIndex = m_Size;//这里还可以加入其他异常处理等等
}
return m_Buf[nIndex];
}
operator void*()
{//方便memset等函数
return (vo