HasPtr 智能指针的声明如下,保存一个指向U_Ptr对象的指针,U_Ptr对象指向实际的int基础对象,代码如下:
#include<iostream>
using namespace std;
// 定义仅由HasPtr类使用的U_Ptr类,用于封装使用计数和相关指针
// 这个类的所有成员都是private,我们不希望普通用户使用U_Ptr类,所以它没有任何public成员
// 将HasPtr类设置为友元,使其成员可以访问U_Ptr的成员
class U_Ptr
{
friend class HasPtr;
int *ip;
size_t use;
U_Ptr(int *p) : ip(p) , use(1)
{
cout << "U_ptr constructor called !" << endl;
}
~U_Ptr()
{
delete ip;
cout << "U_ptr distructor called !" << endl;
}
};
class HasPtr
{
public:
// 构造函数:p是指向已经动态创建的int对象指针
HasPtr(int *p, int i) : ptr(new U_Ptr(p)) , val(i)
{
cout << "HasPtr constructor called ! " << "use = " << ptr->use << endl;
}
// 复制构造函数:复制成员并将使用计数加1
HasPtr(const HasPtr& orig) : ptr(orig.ptr) , val(orig.val)
{
++ptr->use;
cout << "HasPtr copy constructor called ! " << "use = " << ptr->use << endl;
}
// 赋值操作符
HasPtr& operator=(const HasPtr&);
// 析构函数:如果计数为0,则删除U_Ptr对象
~HasPtr()
{
cout << "HasPtr distructor called ! " << "use = " << ptr->use << endl;
if (--ptr->use == 0)
delete ptr;
}
// 获取数据成员
int *get_ptr() const
{
return ptr->ip;
}
int get_int() const
{
return val;
}
// 修改数据成员
void set_ptr(int *p) const
{
ptr->ip = p;
}
void set_int(int i)
{
val = i;
}
// 返回或修改基础int对象
int get_ptr_val() const
{
return *ptr->ip;
}
void set_ptr_val(int i)
{
*ptr->ip = i;
}
private:
U_Ptr *ptr; //指向使用计数类U_Ptr
int val;
};
HasPtr& HasPtr::operator = (const HasPtr &rhs) //注意,这里赋值操作符在减少做操作数的使用计数之前使rhs的使用技术加1,从而防止自我赋值
{
// 增加右操作数中的使用计数
++rhs.ptr->use;
// 将左操作数对象的使用计数减1,若该对象的使用计数减至0,则删除该对象
if (--ptr->use == 0)
delete ptr;
ptr = rhs.ptr; // 复制U_Ptr指针
val = rhs.val; // 复制int成员
return *this;
}
int main(void)
{
int *pi = new int(42);
HasPtr *hpa = new HasPtr(pi, 100); // 构造函数
HasPtr *hpb = new HasPtr(*hpa); // 拷贝构造函数
HasPtr *hpc = new HasPtr(*hpb); // 拷贝构造函数
HasPtr hpd = *hpa; // 拷贝构造函数
cout << hpa->get_ptr_val() << " " << hpb->get_ptr_val() << endl;
hpc->set_ptr_val(10000);
cout << hpa->get_ptr_val() << " " << hpb->get_ptr_val() << endl;
hpd.set_ptr_val(10);
cout << hpa->get_ptr_val() << " " << hpb->get_ptr_val() << endl;
delete hpa;
delete hpb;
delete hpc;
cout << hpd.get_ptr_val() << endl;
return 0;
}
这里的赋值操作符比较麻烦,且让我用图表分析一番:
假设现在又两个智能指针p1、 p2,一个指向内容为42的内存,一个指向内容为100的内存,如下图:
[img]http://files.jb51.net/file_images/article/201211/201211281133037.jpg[/img]
现在,我要做赋值操作,p2 = p1。对比着上面的
HasPtr& operator=(const HasPtr&); // 赋值操作符
此时,rhs就是p1,首先将p1指向的ptr的use加1,
++rhs.ptr->use; // 增加右操作数中的使用计数
然后,做:
if (--ptr->use == 0)
delete ptr;
因为,原先p2指向的对象现在p2不在指向,那么该对象就少了一个指针去指,所以,use做自减1;
此时,条件成立。因为u2的use为1。那么,运行U_Ptr的析构函数,而在U_Ptr的析构函数中,做了delete ip操作,所以释放了内存,不会有内存泄露的问题。
接下来的操作很自然,无需多言:
ptr = rhs.ptr; // 复制U_Ptr指针
val = rhs.val; // 复制int成员
return *this;
做完赋值操作后,那么就成为如下图所示了。红色标注的就是变化的部分:
[img]http://files.jb51.net/file_images/article/201211/201211281133038.jpg[/img]
而还要注意的是,重载赋值操作符的时候,一定要注意的是,检查自我赋值的情况。
如图所示:
[img]http://files.jb51.net/file_images/article/201211/201211281133039.jpg[/img]
此时,做p1 = p1的操作。那么,首先u1.use自增1,为2;然后,u1.use自减1,为1。那么就不会执行delete操作,剩下的操作都可以顺利进行。按《C++ primer》说法,“这个赋值操作符在减少左操作数的使用计数之前使rhs的使用计数加1,从而防止自身赋值”。哎,反正我是那样理解的。当然,赋值操作符函数中一来就可以按常规那样:
if(this == &rhs)
return *this;
运行结果如下图:
[img]http://files.jb51.net/file_images/article/201211/2012112811330310.png[/img]