赋值运算符函数的代码需要关注几点
1) 是否把返回值的类型声明为该类型的引用,并在函数结束前返回实例自身的引用(即*this)。只有返回一个引用,才可以允许连续赋值。否则如果函数的返回值void,该赋值运算符将不能做连续赋值。
2) 是否把传入的参数的类型声明为常量引用。参数声明为引用可以避免无谓的消耗,提高代码的效率。同时我们希望赋值运算符函数内不会改变传入的实例的状态。因此应该为传入的引用参数加上const关键字。
3) 是否释放实例自身已有的内存。如果我们忘记在分配新内存之前释放自身已有的空间,程序将出现内存泄露。
4) 是否判断传入的参数和当前的实例(*this)是不是为同一实例。如果是同一个,则不进行赋值操作,直接返回。如果事先不判断就进行赋值,当*this和传入的参数是同一个实例时,那么一旦释放了自身的内存,传入的参数的内存也同样被释放了,因此再也找不到需要赋值的内容。
CMyString& operator=(constCMyString& str)
{
if(this!=&str)
{
delete[]m_pData;
m_pData=NULL;
m_pData=newchar[strlen(str.m_pData)+1];
strcpy(m_pData,str.m_pData);
}
return*this;
}
但是这个函数有一个缺陷,在分配内存之前先用delete释放了实例m_pData的内存。如果此时内存不足导致new char抛出异常,m_pData将是一个空指针,这样非常容易导致程序崩溃。也就是说一旦在赋值运算符函数内部抛出一个异常,CMyString的实例不再保持有效状态,这就违背了异常安全性原则。
书中提出了一个很新颖的方式,创建一个临时实例,再交换临时实例和原来的实例。
CMyString&operator=(const CMyString& str)
{
if(this!=&str)
{
CMyStringtemp(str);
char*pTemp=temp.m_pData;
temp.m_pData=m_pData;
m_pData=pTemp;
}
return*this;
}
在这个函数中,我们先创建一个临时temp,接着把temp.m_pData和自身的m_pData做交换。由于temp是个局部变量,但程序运行到if的外面也就出了该变量的作用域,就会自动调用temp的析构函数,把temp所指向的内存释放掉。由于temp.m_pData指向的内存就是实例之前m_pData的内存,相当于自动调用析构函数释放内存
给出CMyString类的完整代码
class CMyString
{
public:
CMyString(char*pData=NULL)
{
if(pData==NULL)
{
m_pData=newchar[1];
m_pData[0]='\0';
}
else
{
m_pData=newchar[strlen(pData)+1];
strcpy(m_pData,pData);
}
}
CMyString(constCMyString& str)
{
m_pData=newchar[strlen(str.m_pData)+1];
strcpy(m_pData,str.m_pData);
}
~CMyString()
{
delete[]m_pData;
m_pData=NULL;
}
/*
CMyString&operator=(const CMyString& str)
{
if(this!=&str)
{
delete[]m_pData;
m_pData=NULL;
m_pData=newchar[strlen(str.m_pData)+1];
strcpy(m_pData,str.m_pData);
}
return*this;
}*/
CMyString&operator=(const CMyString& str)
{
if(this!=&str)
{
CMyStringtemp(str);
char*pTemp=temp.m_pData;
temp.m_pData=m_pData;
m_pData=pTemp;
}
return*this;
}
private:
char*m_pData;
};