#include<iostream>
#include<cstring>
using namespace std;
class CMyString
{
public:
CMyString(char* pData);
CMyString(const CMyString& str);
CMyString& operator =(const CMyString &str);
~CMyString(void);
private:
char* m_pData;
};
CMyString& CMyString :: operator =(const CMyString &str)
{
if (this == &str)
{
return *this;
}
delete []m_pData;
m_pData = NULL;
m_pData = new char[strlen(str.m_pData) + 1];
strcpy(m_pData,str.m_pData);
return *this;
}
这个函数中,我们在分配内存之前先用delete释放了实例m_pDatade内存。如果此时内存不足导致new char抛出异常,则m_pData将会是一个空指针,执行 strcpy(m_pData,str.m_pData)将会出错,现在面临的问题是原来的内容被delete掉了,新内容也没有分配好。
为了解决原来的内容被delete掉了,新内容也没有分配好。我们可以采用下边的代码。
CMyString& CMyString::operatoe =(const CMyString &str)
{
if (this != &str)
{
CMyString strTemp(str);
char* pTemp = strTemp.m_pData;
strTemp.m_pData = m_pData;
m_pData = pTemp;
}
return *this;
}
在这个函数中,我们先创建一个临时实例 strTemp,要在拷贝构造函数中使用深拷贝,接着把 strTemp.m_pData 和实例自身的m_pData做交换。由于 strTemp 是一个局部变量,但程序运行到if的外面时也就出来该变量的作用域,就会自动调用strTemp的析构函数,把strTemp.m_pData所指向的内存释放掉。由于strTemp.m_pData指向的内存就是实例之前m_pData的内存,这就相当于自动调用析构函数释放实例的内存。也就是说新的创建好了,旧的也释放了。如果CMyString strTemp(str);没有创建成功,后边的代码不会执行,也就不会改变原来实例的状态。保证了异常安全性。