搭配异常可以让异常的代码更简洁
文章目录
- 智能指针
- 内存泄漏的危害
- 1.auto_ptr(非常不建议使用)
- 2.unique_ptr
- 3.shared_ptr
- 4.weak_ptr
- 总结
智能指针
C++中为什么会需要智能指针呢?下面我们看一下样例:
int div()
{
int a, b;
cin >> a >> b;
if (b == 0)
throw invalid_argument("除0错误");
return a / b;
}
void Func()
{
// 1、如果p1这里new 抛异常会如何?
// 2、如果p2这里new 抛异常会如何?
// 3、如果div调用这里又会抛异常会如何?
int* p1 = new int;
int* p2 = new int;
cout << div() << endl;
delete p1;
delete p2;
}
int main()
{
try
{
Func();
}
catch (exception& e)
{
cout << e.what() << endl;
}
return 0;
}
在上面的代码中,一旦出现异常那就会造成内存泄漏,什么是内存泄漏呢:
如果func函数里的p1new的时候抛异常该怎么办呢?对于第一个new如果抛异常会直接跳到main函数中的catch被捕获,那么p2new失败了会怎么办呢?div函数抛异常我们可以捕获一下:
那么如果是p2失败了我们还要再catch一下:
void Func()
{
// 1、如果p1这里new 抛异常会如何?
// 2、如果p2这里new 抛异常会如何?
// 3、如果div调用这里又会抛异常会如何?
int* p1 = new int;
int* p2 = nullptr;
try
{
p2 = new int;
try
{
cout << div() << endl;
}
catch (...)
{
delete p1;
delete p2;
throw;
}
}
catch (...)
{
delete p1;
delete p2;
}
}
那么这样的代码看起来会不会有些冗余呢?为了处理这样的问题,智能指针就能起到很好的作用:
template <class T>
class SmartPtr
{
public:
SmartPtr(T* ptr)
:_ptr(ptr)
{
}
~SmartPtr()
{
delete _ptr;
cout << _ptr << endl;
}
private:
T* _ptr;
};
int div()
{
int a, b;
cin >> a >> b;
if (b == 0)
throw invalid_argument("除0错误");
return a / b;
}
void Func()
{
int* p1 = new int;
SmartPtr<int> sp1(p1);
int* p2 = new int;
SmartPtr<int> sp2(p2);
cout << div() << endl;
delete p1;
delete p2;
}
int main()
{
try
{
Func();
}
catch (exception& e)
{
cout << e.what() << endl;
}
return 0;
}
可以看到有了智能指针即使抛异常了我们没有释放的空间也会被自动释放,因为抛异常后自定义类型出了作用域我们智能指针的析构函数会将这个空间释放。当然,我们的智能指针也可以直接创建资源,比如下面这样:
但是我们这样写不就不可以对指针解引用访问指针的资源了吗?其实我们只需要再给智能指针多加加个功能让它变得像指针一样就解决了这个问题:
template <class T>
class SmartPtr
{
public:
//保存资源
SmartPtr(T* ptr)
:_ptr(ptr)
{
}
//释放资源
~SmartPtr()
{
delete _ptr;
cout << _ptr << endl;
}
//像指针一样
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
T& operator[](size_t pos)
{
return _ptr[pos];
}
private:
T* _ptr;
};
这个时候我们来使用一下这个指针:
可以看到现在我们的这个智能指针也可以正常使用了,再次说明一下:智能指针的构造函数是为了保存资源,析构函数是为了释放资源,其他功能是为了和指针一样。我们上面将资源管理的责任托管给对象的做法就叫做RAII(资源获得即初始化),这就是避免内存泄漏的一种方法。