C++11/14介绍(五)——智能指针和引用计数(四)

std::weak_ptr

#include <iostream>
#include <memory>

class A;
class B;

class A {
public:
    std::shared_ptr<B> pointer;
    ~A() {
        std::cout << "A 被销毁" << std::endl;
    }
};
class B {
public:
    std::shared_ptr<A> pointer;
    ~B() {
        std::cout << "B 被销毁" << std::endl;
    }
};
int main() {
    std::shared_ptr<A> a = std::make_shared<A>();
    std::shared_ptr<B> b = std::make_shared<B>();
    a->pointer = b;
    b->pointer = a;

    return 0;
}

以上代码中shared_ptr存在着资源无法释放的问题,因为a,b内部的pointer同时又引用了b,a,这使得a,b的引用计数变为了2,而离开作用域时,a,b智能指针被析构,引用计数减一,导致了a,b对象指向的内存区域引用计数不为0,而外部已经无法找到这块区域了,因此造成了内存泄漏。

解决方法为使用std::weak_ptr,这是一种弱引用,不会引起引用计数增加,

#include <iostream>
#include <memory>

class A;
class B;

class A {
public:
    // A 或 B 中至少有一个使用 weak_ptr
    std::weak_ptr<B> pointer;
    ~A() {
        std::cout << "A 被销毁" << std::endl;
    }
};
class B {
public:
    std::shared_ptr<A> pointer;
    ~B() {
        std::cout << "B 被销毁" << std::endl;
    }
};
int main() {
    std::shared_ptr<A> a = std::make_shared<A>();
    std::shared_ptr<B> b = std::make_shared<B>();
    a->pointer = b;
    b->pointer = a;

    return 0;
}

结果:

B 被销毁
A 被销毁

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章