博客: www.cyhone.com
公众号:编程沉思录
C++11中推出了三种智能指针,unique_ptr、shared_ptr和weak_ptr,同时也将auto_ptr置为废弃(deprecated)。
但是在实际的使用过程中,很多人都会有这样的问题:
本文试图理清楚三种智能指针的具体使用场景,并讲解三种智能指针背后的性能消耗。同时也解释了为什么要用shared_from_this以及智能指针的函数传参问题。
首先需要理清楚的概念就是对象所有权的概念。所有权在rust语言中非常严格,写rust的时候必须要清楚自己创建的每个对象的所有权。
但是C++比较自由,似乎我们不需要明白对象的所有权,写的代码也能正常运行。但是明白了对象所有权,我们才可以正确管理好对象生命周期和内存问题。
C++引入了智能指针,也是为了更好的描述对象所有权,简化内存管理,从而大大减少我们C++内存管理方面的犯错机会。
我们大多数场景下用到的应该都是unique_ptr。
unique_ptr代表的是专属所有权,即由unique_ptr管理的内存,只能被一个对象持有。
所以,unique_ptr不支持复制和赋值,如下:
auto w = std::make_unique<Widget>();
auto w2 = w; // 编译错误
如果想要把w复制给w2, 是不可以的。因为复制从语义上来说,两个对象将共享同一块内存。
因此,unique_ptr只支持移动, 即如下:
auto w = std::make_unique<Widget>();
auto w2 = std::move(w); // w2获得内存所有权,w此时等于nullptr
unique_ptr代表的是专属所有权,如果想要把一个unique_ptr的内存交给另外一个unique_ptr对象管理。只能使用std::move转移当前对象的所有权。转移之后,当前对象不再持有此内存,新的对象将获得专属所有权。
如上代码中,将w对象的所有权转移给w2后,w此时等于nullptr,而w2获得了专属所有权。
因为C++的zero cost abstraction的特点,unique_ptr在默认情况下和裸指针的大小是一样的。
所以内存上没有任何的额外消耗,性能是最优的。
unique_ptr一个最简单的使用场景是用于类属性。代码如下:
class Box{
public:
Box() : w(new Widget())
{}
~Box()
{
// 忘记delete w
}
private:
Widget* w;
};
如果因为一些原因,w必须建立在堆上。如果用裸指针管理w,那么需要在析构函数中delete w
;
这种写法虽然没什么问题,但是容易漏写delete语句,造成内存泄漏。
如果按照unique_ptr的写法,不用在析构函数手动delete属性,当对象析构时,属性w
将会自动释放内存。
假如我们在一段代码中,需要创建一个对象,处理一些事情后返回,返回之前将对象销毁,如下所示:
void process()
{
Widget* w = new Widget();
w->do_something(); // 可能会发生异常
delete w;
}
在正常流程下,我们会在函数末尾delete创建的对象w,正常调用析构函数,释放内存。
但是如果w->do_something()发生了异常,那么delete w
将不会被执行。此时就会发生内存泄漏。
我们当然可以使用try...catch捕捉异常,在catch里面执行delete,但是这样代码上并不美观,也容易漏写。
如果我们用std::unique_ptr,那么这个问题就迎刃而解了。无论代码怎么抛异常,在unique_ptr离开函数作用域的时候,内存就将会自动释放。
在使用shared_ptr之前应该考虑,是否真的需要使用shared_ptr, 而非unique_ptr。
shared_ptr代表的是共享所有权,即多个shared_ptr可以共享同一块内存。
因此,从语义上来看,shared_ptr是支持复制的。如下:
auto w = std::make_shared<Widget>();
{
auto w2 = w;
cout << w.use_count() << endl; // 2
}
cout << w.use_count() << endl; // 1
shared_ptr内部是利用引用计数来实现内存的自动管理,每当复制一个shared_ptr,引用计数会+1。当一个shared_ptr离开作用域时,引用计数会-1。当引用计数为0的时候,则delete内存。
同时,shared_ptr也支持移动。从语义上来看,移动指的是所有权的传递。如下:
auto w = std::make_shared<Widget>();
auto w2 = std::move(w); // 此时w等于nullptr,w2.use_count()等于1
我们将w对象move给w2,意味着w放弃了对内存的所有权和管理,此时w对象等于nullptr。
而w2获得了对象所有权,但因为此时w已不再持有对象,因此w2的引用计数为1。
BlockingQueue<std::shared_ptr<void>>
将对象转移到另外一个线程中释放,从而解放关键线程。我们往往会需要在类内部使用自身的shared_ptr,例如:
class Widget
{
public:
void do_something(A& a)
{
a.widget = 该对象的shared_ptr;
}
}
我们需要把当前shared_ptr对象同时交由对象a进行管理。意味着,当前对象的生命周期的结束不能早于对象a。因为对象a在析构之前还是有可能会使用到a.widget
。
如果我们直接a.widget = this;
, 那肯定不行, 因为这样并没有增加当前shared_ptr的引用计数。shared_ptr还是有可能早于对象a释放。
如果我们使用a.widget = std::make_shared<Widget>(this);
,肯定也不行,因为这个新创建的shared_ptr,跟当前对象的shared_ptr毫无关系。当前对象的shared_ptr生命周期结束后,依然会释放掉当前内存,那么之后a.widget
依然是不合法的。
对于这种,需要在对象内部获取该对象自身的shared_ptr, 那么该类必须继承std::enable_shared_from_this<T>
。代码如下:
class Widget : public std::enable_shared_from_this<Widget>
{
public:
void do_something(A& a)
{
a.widget = shared_from_this();
}
}
这样才是合法的做法。
weak_ptr是为了解决shared_ptr双向引用的问题。即:
class B;
struct A{
shared_ptr<B> b;
};
struct B{
shared_ptr<A> a;
};
auto pa = make_shared<A>();
auto pb = make_shared<B>();
pa->b = pb;
pb->a = pa;
pa和pb存在着循环引用,根据shared_ptr引用计数的原理,pa和pb都无法被正常的释放。
对于这种情况, 我们可以使用weak_ptr:
class B;
struct A{
shared_ptr<B> b;
};
struct B{
weak_ptr<A> a;
};
auto pa = make_shared<A>();
auto pb = make_shared<B>();
pa->b = pb;
pb->a = pa;
weak_ptr不会增加引用计数,因此可以打破shared_ptr的循环引用。
通常做法是parent类持有child的shared_ptr, child持有指向parent的weak_ptr。这样也更符合语义。
很多时候,函数的参数是个指针。这个时候就会面临选择困难症,这个参数应该怎么传,应该是shared_ptr<T>,还是const shared_ptr<T>&,还是直接raw pointer更合适。
1. 只在函数使用指针,但并不保存
假如我们只需要在函数中,用这个对象处理一些事情,但不打算涉及其生命周期的管理,不打算通过函数传参延长shared_ptr的生命周期。
对于这种情况,可以使用raw pointer或者const shared_ptr<T>&。
即:
void func(Widget*);
void func(const shared_ptr<Widget>&)
实际上第一种裸指针的方式可能更好,从语义上更加清楚,函数也不用关心智能指针的类型。
对于智能指针的使用,实际上是对所有权和生命周期的思考,一旦想明白了这两点,那对智能指针的使用也就得心应手了。
同时理解了每种智能指针背后的性能消耗、使用场景,那智能指针也不再是黑盒子和洪水猛兽。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。