不检查复制赋值运算符是否将对象分配给自身是不安全的,因为这可能导致对象数据的损坏或未定义的行为。以下是一个示例代码,展示了如何在复制赋值运算符中检查自赋值并避免这个问题:
#include
class MyClass {
private:
int* data;
public:
MyClass(int val) {
data = new int(val);
}
// 复制构造函数
MyClass(const MyClass& other) {
data = new int(*other.data);
}
// 复制赋值运算符
MyClass& operator=(const MyClass& other) {
// 检查自赋值
if (this == &other) {
return *this;
}
// 释放旧的数据
delete data;
// 复制数据
data = new int(*other.data);
return *this;
}
int getValue() const {
return *data;
}
~MyClass() {
delete data;
}
};
int main() {
MyClass obj1(5);
MyClass obj2(10);
// 将obj2赋值给obj2
obj2 = obj2;
std::cout << "obj1 value: " << obj1.getValue() << std::endl;
std::cout << "obj2 value: " << obj2.getValue() << std::endl;
return 0;
}
在上述代码中,复制赋值运算符重载函数中添加了自赋值检查。如果检测到自赋值,函数会直接返回当前对象,而不进行数据复制操作。这样可以避免出现悬挂指针和内存泄漏的问题。
在示例中,obj2 = obj2;
是一个自赋值操作,但由于我们在复制赋值运算符中进行了检查,所以不会对对象造成任何损害。
输出结果将是:
obj1 value: 5
obj2 value: 10