将已有的对象拷贝给另一个对象时,会用到赋值运算符函数, 尤其是当对象的内存分配在堆上时
函数原型:
ClassName & operator = (const ClassName &obj)
<span style="font-family:Microsoft YaHei;">#include <iostream>
using namespace std;
class CMyString {
public:
CMyString(char* pData = NULL) ;
CMyString(const CMyString &str);
~CMyString();
private:
char* m_data;
};</span>
为上面的类添加赋值运算符函数
CMyString & operator = (const CMyString &other) ;
注意
1. 是否把返回值的类型声明为类的引用,并在函数结束前返回实例自身的引用(即*this)。只有返回一个引用,才可以允许连续赋值。否则,如果函数的返回值是void,那么这个赋值运算符不能连续赋值,假设有3个CMyString对象:str1, str2, str3, 在程序语句中str1 = str2 =str3将不能通过编译。
2. 是否把传入的参数类型声明为常量引用。如果传入的参数不是引用而是实例,那么从形参到实参会调用一次拷贝构造函数。把参数声明为引用可以避免这样无谓的消耗,提高代码的效率。同时,在赋值运算符内不会改变传入的实例的状态,因此应该为传入的引用参数加上const关键字。
3. 是否释放实例自身已有的内存。如果我们忘记在分配新内存之前释放自身已有的空间,程序将出现内存泄露。
4. 是否判断传入的参数和当前的实例(*this)是不是同一个实例。如果是同一个,就不进行赋值操作,直接返回。如果实现不判断就赋值,那么在释放实例自身的内存时会导致严重的问题:当*this和传入的参数是同一个实例时,一旦释放了实例自身的内存,传入的参数的内存也同时被释放了,因此再也找不到需要赋值的内容了。
考虑这4点,我们应该可以写出下面的代码:
CMyString& CMyString::operator = (const CMyString &other) {
if (this == &other) {
return *this;
}
else {
delete []m_data;
m_data = NULL;
m_data = new char[strlen(other.m_data) + 1];
strcpy(m_data, other.m_data);
return *this;
}
}
考虑异常安全性呢?
在上面的函数中,我们在分配内存之前先用delete释放了实例m_data的内存。如果此时内存不足导致new char抛出异常,m_data将会是一个空指针,这样非常容易导致程序崩溃。也就是说,一旦在赋值运算符函数内部跑出一个异常,CMyString的实例就不再保有有效状态,这违背了异常安全性原则。
要想在赋值运算符内部实现异常安全性,有两种方法。
一个简单的办法时先用new分配新的内容,再用delete释放已有的内容。这样只在分配内容成功之后才释放原来的内容,也就是当分配内存失败时我们能确保CMyString的实例不会被修改。
还有个更好的办法是,先创建一个临时实例,再交换临时实例和原来的实例。
CMyString& CMyString::operator = (const CMyString &other) {
if (this == &other) {
return *this;
}
else {
CMyString strTemp(other);
char* pTemp = strTemp.m_data;
strTemp.m_data = m_data;
m_data = pTemp;
return *this;
}
}
在这个函数中,我们先创建了一个临时实例strTemp,接着把strTemp.m_data和实例自身的m_data互换。由于strTemp是一个局部变量,当程序运行到if的外面时也就出了该变量的作用域,就会自动调用strTemp的析构函数,把strTemp.m_data所指向的内存释放掉。由于strTemp.m_data指向的内存就是实例之前m_data的内存,这就相当于自动调用析构函数释放实例的内存。
在新的代码中,我们在CMyString的构造函数里用new分配内存,如果由于内存不足抛出了异常,我们还没有修改原来实例的状态,因此实例的状态还是有效的,也就保证了异常安全性。