#include <iostream>
#include <memory>
/*
// pa与pb相互引用,导致引用计数不能减为0,导致内存泄漏
class B;
class A {
public:
std::shared_ptr<B> m_pb;
~A() {
std::cout << "A deleted" << std::endl;
}
};
class B {
public:
std::shared_ptr<A> m_pa;
~B() {
std::cout << "B deleted" << std::endl;
}
};
*/
// 为了解决上述问题,将其中一个成员改为weak_ptr,weak_ptr与shared_ptr的区别就是不计数。
// 资源B的引用开始就只有1,当pb析构时,B的计数变为0,B得到释放,B释放的同时也会使A的计数减1。
// 同时pa析构时使A的计数减1,那么A的计数为0,A得到释放。
class B;
class A {
public:
std::shared_ptr<B> m_pb;
~A() {
std::cout << "A deleted" << std::endl;
}
};
class B {
public:
std::weak_ptr<A> m_pa;
~B() {
std::cout << "B deleted" << std::endl;
}
};
void func() {
std::shared_ptr<A> pa(new A());
std::shared_ptr<B> pb(new B());
std::cout << "before reference pa.use_count():" << pa.use_count() << std::endl;
std::cout << "before reference pb.use_count():" << pb.use_count() << std::endl;
pa->m_pb = pb;
pb->m_pa = pa;
// weak_ptr只能使用shared_ptr或者weak_ptr进行初始化
std::cout << "after reference pa.use_count():" << pa.use_count() << std::endl;
std::cout << "after reference pb.use_count():" << pb.use_count() << std::endl;
// 获取当前资源被多少个指针进行引用
std::cout << "pb->m_pa.use_count():" << pb->m_pa.use_count() << std::endl;
// weak_ptr管理的资源不能被直接访问,需要通过lock函数从其中获取shared_ptr才可以被访问。
std::shared_ptr<A> tempA = pb->m_pa.lock();
// 使用expired查看当前weak_ptr是否已经不指向任何数据
std::cout << "pb->m_pa.expired() :" << (pb->m_pa.expired() ? " is " : " is not ") << "expire"<< std::endl;
pb->m_pa.reset();
std::cout << "pb->m_pa.expired() :" << (pb->m_pa.expired() ? " is " : " is not ") << "expire" << std::endl;
}
int main() {
func();
}
使用示例方式记录weak_ptr使用方法
最新推荐文章于 2025-07-05 00:00:00 发布