C++ 智能指针 shared_ptr
最近接触智能指针比较多,对智能指针的使用做下基本的总结。
shared_ptr 是C++11提供的一种智能指针类,本质上是一个类,具有构造函数和析构函数,于是能够自动释放没有指针引用的资源。
其核心实现便是计数。
每个 shared_ptr 对象在内部指向两个内存位置:
1、指向对象的指针。
2、用于控制引用计数数据的指针。
共享所有权如何在参考计数的帮助下工作:
1、当新的 shared_ptr 对象与指针关联时,则在其构造函数中,将与此指针关联的引用计数增加1。
2、当任何 shared_ptr 对象超出作用域时,则在其析构函数中,它将关联指针的引用计数减1。如果引用计数变为0,则表示没有其他 shared_ptr 对象与此内存关联,在这种情况下,它使用delete函数删除该内存。
同时:
(1) 智能指针主要的用途就是方便资源的管理,自动释放没有指针引用的资源。
(2) 使用引用计数来标识是否有多余指针指向该资源。(注意,shart_ptr本身指针会占1个引用)
(3) 在赋值操作中, 原来资源的引用计数会减一,新指向的资源引用计数会加一(计数是在堆上)。
std::shared_ptr<Test> p1(new Test); std::shared_ptr<Test> p2(new Test); p1 = p2;
(4) 引用计数加一/减一操作是原子性的,所以线程安全的。
(5) make_shared要优于使用new,make_shared可以一次将需要内存分配好。
std::shared_ptr<Test> p = std::make_shared<Test>(); std::shared_ptr<Test> p(new Test);
(6) std::shared_ptr的大小是原始指针的两倍,因为它的内部有一个原始指针指向资源,同时有个指针指向引用计数。
(7) 引用计数是分配在动态分配的,std::shared_ptr支持拷贝,新的指针获可以获取前引用计数个数。
示例:
ps:尽量不用使用get函数去获取,可能会出错。
include <iostream> #include <memory> #include <thread> #include <chrono> #include <mutex> struct Test { Test() { std::cout << " Test::Test() "; } ~Test() { std::cout << " Test::~Test() "; } }; //线程函数 void thr(std::shared_ptr<Test> p) { //线程暂停1s std::this_thread::sleep_for(std::chrono::seconds(1)); //赋值操作, shared_ptr引用计数use_cont加1(c++11中是原子操作) std::shared_ptr<Test> lp = p; { //static变量(单例模式),多线程同步用 static std::mutex io_mutex; //std::lock_guard加锁 std::lock_guard<std::mutex> lk(io_mutex); std::cout << "local pointer in a thread: " << " lp.get() = " << lp.get() << ", lp.use_count() = " << lp.use_count() << ' '; } } int main() { //使用make_shared一次分配好需要内存 std::shared_ptr<Test> p = std::make_shared<Test>(); //std::shared_ptr<Test> p(new Test); std::cout << "Created a shared Test " << " p.get() = " << p.get() << ", p.use_count() = " << p.use_count() << ' '; //创建三个线程,t1,t2,t3 //形参作为拷贝, 引用计数也会加1 std::thread t1(thr, p), t2(thr, p), t3(thr, p); std::cout << "Shared ownership between 3 threads and released " << "ownership from main: " << " p.get() = " << p.get() << ", p.use_count() = " << p.use_count() << ' '; //等待结束 t1.join(); t2.join(); t3.join(); std::cout << "All threads completed, the last one deleted "; return 0; }
详细解释
创建空的 shared_ptr 对象
因为带有参数的 shared_ptr 构造函数是 explicit 类型的,所以不能像这样std::shared_ptr<int> p1 = new int();隐式调用它构造函数。创建新的shared_ptr对象的最佳方法是使用std :: make_shared:
std::shared_ptr<int> p1 = std::make_shared<int>();
std::make_shared 一次性为int对象和用于引用计数的数据都分配了内存,而new操作符只是为int分配了内存。
分离关联的原始指针
要使 shared_ptr 对象取消与相关指针的关联,可以使用reset()函数:
不带参数的reset():
p1.reset();
它将引用计数减少1,如果引用计数变为0,则删除指针。
带参数的reset():
p1.reset(new int(34));
在这种情况下,它将在内部指向新指针,因此其引用计数将再次变为1。
使用nullptr重置:
p1 = nullptr;
shared_ptr是一个伪指针
shared_ptr充当普通指针,我们可以将*和->与 shared_ptr 对象一起使用,也可以像其他 shared_ptr 对象一样进行比较;
代码如下
#include <iostream> #include <memory> // 需要包含这个头文件 int main() { // 使用 make_shared 创建空对象 std::shared_ptr<int> p1 = std::make_shared<int>(); *p1 = 78; std::cout << "p1 = " << *p1 << std::endl; // 输出78 // 打印引用个数:1 std::cout << "p1 Reference count = " << p1.use_count() << std::endl; // 第2个 shared_ptr 对象指向同一个指针 std::shared_ptr<int> p2(p1); // 下面两个输出都是:2 std::cout << "p2 Reference count = " << p2.use_count() << std::endl; std::cout << "p1 Reference count = " << p1.use_count() << std::endl; // 比较智能指针,p1 等于 p2 if (p1 == p2) { std::cout << "p1 and p2 are pointing to same pointer "; } std::cout<<"Reset p1 "<<std::endl; // 无参数调用reset,无关联指针,引用个数为0 p1.reset(); std::cout << "p1 Reference Count = " << p1.use_count() << std::endl; // 带参数调用reset,引用个数为1 p1.reset(new int(11)); std::cout << "p1 Reference Count = " << p1.use_count() << std::endl; // 把对象重置为NULL,引用计数为0 p1 = nullptr; std::cout << "p1 Reference Count = " << p1.use_count() << std::endl; if (!p1) { std::cout << "p1 is NULL" << std::endl; // 输出 } return 0; }
自定义删除器 Deleter
下面将讨论如何将自定义删除器与 std :: shared_ptr 一起使用。
当 shared_ptr 对象超出范围时,将调用其析构函数。在其析构函数中,它将引用计数减1,如果引用计数的新值为0,则删除关联的原始指针。
析构函数中删除内部原始指针,默认调用的是delete()函数。
delete Pointer;
有些时候在析构函数中,delete函数并不能满足我们的需求,可能还想加其他的处理。
当 shared_ptr 对象指向数组
std::shared_ptr<int> p3(new int[12]);
像这样申请的数组,应该调用delete []释放内存,而shared_ptr析构函数中默认delete并不能满足需求。
给shared_ptr添加自定义删除器
在上面在这种情况下,我们可以将回调函数传递给 shared_ptr 的构造函数,该构造函数将从其析构函数中调用以进行删除,即
// 自定义删除器 void deleter(Sample * x) { std::cout << "DELETER FUNCTION CALLED "; delete[] x; } // 构造函数传递自定义删除器指针 std::shared_ptr<Sample> p3(new Sample[12], deleter);
下面看一个完整的示例:
#include <iostream> #include <memory> struct Sample { Sample() { std::cout << "Sample "; } ~Sample() { std::cout << "~Sample "; } }; void deleter(Sample * x) { std::cout << "Custom Deleter "; delete[] x; } int main() { std::shared_ptr<Sample> p3(new Sample[3], deleter); return 0; }
输出: Sample Sample Sample Custom Deleter ~Sample ~Sample ~Sample
使用Lambda 表达式 / 函数对象作为删除器
class Deleter { public: void operator() (Sample * x) { std::cout<<"DELETER FUNCTION CALLED "; delete[] x; } }; // 函数对象作为删除器 std::shared_ptr<Sample> p3(new Sample[3], Deleter()); // Lambda表达式作为删除器 std::shared_ptr<Sample> p4(new Sample[3], [](Sample * x){ std::cout<<"DELETER FUNCTION CALLED "; delete[] x; });
shared_ptr 相对于普通指针的优缺点
缺少 ++, – – 和 [] 运算符
与普通指针相比,shared_ptr仅提供-> 、*和==运算符,没有+、-、++、--、[]等运算符。
示例:
#include<iostream> #include<memory> struct Sample { void dummyFunction() { std::cout << "dummyFunction" << std::endl; } }; int main() { std::shared_ptr<Sample> ptr = std::make_shared<Sample>(); (*ptr).dummyFunction(); // 正常 ptr->dummyFunction(); // 正常 // ptr[0]->dummyFunction(); // 错误方式 // ptr++; // 错误方式 //ptr--; // 错误方式 std::shared_ptr<Sample> ptr2(ptr); if (ptr == ptr2) // 正常 std::cout << "ptr and ptr2 are equal" << std::endl; return 0; }
NULL检测
当我们创建 shared_ptr 对象而不分配任何值时,它就是空的;普通指针不分配空间的时候相当于一个野指针,指向垃圾空间,且无法判断指向的是否是有用数据。
shared_ptr 检测空值方法
std::shared_ptr<Sample> ptr3; if(!ptr3) std::cout<<"Yes, ptr3 is empty" << std::endl; if(ptr3 == NULL) std::cout<<"ptr3 is empty" << std::endl; if(ptr3 == nullptr) std::cout<<"ptr3 is empty" << std::endl;
创建 shared_ptr 时注意事项
不要使用同一个原始指针构造 shared_ptr
创建多个 shared_ptr 的正常方法是使用一个已存在的shared_ptr 进行创建,而不是使用同一个原始指针进行创建。
示例:
int *num = new int(23); std::shared_ptr<int> p1(num); std::shared_ptr<int> p2(p1); // 正确使用方法 std::shared_ptr<int> p3(num); // 不推荐 std::cout << "p1 Reference = " << p1.use_count() << std::endl; // 输出 2 std::cout << "p2 Reference = " << p2.use_count() << std::endl; // 输出 2 std::cout << "p3 Reference = " << p3.use_count() << std::endl; // 输出 1
假如使用原始指针num创建了p1,又同样方法创建了p3,当p1超出作用域时会调用delete释放num内存,此时num成了悬空指针,当p3超出作用域再次delete的时候就可能会出错。
不要用栈中的指针构造 shared_ptr 对象
shared_ptr 默认的构造函数中使用的是delete来删除关联的指针,所以构造的时候也必须使用new出来的堆空间的指针。
示例:
#include<iostream> #include<memory> int main() { int x = 12; std::shared_ptr<int> ptr(&x); return 0; }
当 shared_ptr 对象超出作用域调用析构函数delete 指针&x时会出错。
建议使用 make_shared
为了避免以上两种情形,建议使用make_shared()<>创建 shared_ptr 对象,而不是使用默认构造函数创建。
std::shared_ptr<int> ptr_1 = make_shared<int>(); std::shared_ptr<int> ptr_2 (ptr_1);
另外不建议使用get()函数获取 shared_ptr 关联的原始指针,因为如果在 shared_ptr 析构之前手动调用了delete函数,同样会导致类似的错误。
再论shared_ptr 的线程安全
虽然我们借shared_ptr 来实现线程安全的对象释放,但是shared_ptr 本身不是100% 线程安全的。它的引用计数本身是安全且无锁的,但对象的读写则不是,因为shared_ptr 有两个数据成员,读写操作不能原子化。根据文档11,shared_ptr 的线程安全级别和内建类型、标准库容器、std::string 一样,即:
一个shared_ptr 对象实体可被多个线程同时读取;
两个shared_ptr 对象实体可以被两个线程同时写入,“析构”算写操作;
如果要从多个线程读写同一个shared_ptr 对象,那么需要加锁。
请注意,以上是shared_ptr 对象本身的线程安全级别,不是它管理的对象的线程安全级别。
要在多个线程中同时访问同一个shared_ptr,正确的做法是用mutex 保护:
MutexLock mutex; // No need for ReaderWriterLock shared_ptr<Foo> globalPtr; // 我们的任务是把globalPtr 安全地传给doit() void doit(const shared_ptr<Foo>& pFoo);
globalPtr 能被多个线程看到,那么它的读写需要加锁。注意我们不必用读写锁,而只用最简单的互斥锁,这是为了性能考虑。因为临界区非常小,用互斥锁也不会阻塞并发读。
为了拷贝globalPtr,需要在读取它的时候加锁,即:
void read() { shared_ptr<Foo> localPtr; { MutexLockGuard lock(mutex); localPtr = globalPtr; // read globalPtr } // use localPtr since here,读写localPtr 也无须加锁 doit(localPtr); }
写入的时候也要加锁:
void write() { shared_ptr<Foo> newPtr(new Foo); // 注意,对象的创建在临界区之外 { MutexLockGuard lock(mutex); globalPtr = newPtr; // write to globalPtr } // use newPtr since here,读写newPtr 无须加锁 doit(newPtr); }
注意到上面的read() 和write() 在临界区之外都没有再访问globalPtr,而是用了一个指向同一Foo 对象的栈上shared_ptr local copy。下面会谈到,只要有这样的local copy 存在,shared_ptr 作为函数参数传递时不必复制,用reference to const 作为参数类型即可。
另外注意到上面的new Foo 是在临界区之外执行的,这种写法通常比在临界区内写globalPtr.reset(new Foo) 要好,因为缩短了临界区长度。如果要销毁对象,我们固然可以在临界区内执行globalPtr.reset(),但是这样往往会让对象析构发生在临界区以内,增加了临界区的长度。
一种改进办法是像上面一样定义一个localPtr,用它在临界区内与globalPtr 交换(swap()),这样能保证把对象的销毁推迟到临界区之外。练习:在write() 函数中,globalPtr = newPtr; 这一句有可能会在临界区内销毁原来globalPtr 指向的Foo 对象,设法将销毁行为移出临界区。