处理共享堆内存——std::shared_ptr

上一节中,我们了解了如何使用unique_ptr。这个类型非常有用,能帮助我们管理动态分配的对象。不过,所有权只能让一个类型对象所有,不能让多个对象指向同一个动态分配的对象。

指针类型shared_ptr就是为了应对这种情况所设计的。共享指针可以随时进行拷贝,其内部有一个计数器,记录了有多少对象持有这个指针。只有当最后一个持有者被销毁时,才会对动态分配的对象进行删除。同样,其也不会让我们陷入内存泄漏的窘境,因为对象也会在使用之后进行自动删除。同时,需要确定对象没有过早的被删除,或是删除的过于频繁(每次对象的创建都要进行一次删除)。

本节中,你将了解到如何使用shared_ptr自动的对动态对象进行管理,并且能在多个所有者间共享动态对象,而后了解其与unique_ptr之间的区别:

How to do it...

我们将完成一个与unique_ptr节类似的程序,以展示shared_ptr的用法:

  1. 包含必要的头文件,并声明所使用的命名空间:

    #include <iostream>
    #include <memory>
    
    using namespace std;
  2. 然后定义一个辅助类,其能帮助我们了解类何时创建和销毁。我们将会使用shared_ptr对内存进行管理:

    class Foo
    {
    public:
        string name;
    
        Foo(string n)
            : name{move(n)}
        { cout << "CTOR " << name << '\n'; }
    
        ~Foo() { cout << "DTOR " << name << '\n'; }
    };
  3. 接下来,我们将实现一个函数Foo,其参数的类型为共享指针。接受共享指针作为参数的方式,要比引用有意思的多,因为这样我们不会进行拷贝,但是会改变共享这指针内部的计数器:

    void f(shared_ptr<Foo> sp)
    {
        cout << "f: use counter at "
            << sp.use_count() << '\n';
    }
  4. 主函数中声明一个空的共享指针。通过默认构造方式对其进行构造,其实际上是一个null指针:

    int main()
    {
        shared_ptr<Foo> fa;
  5. 下一步,我们将创建一个代码段,并创建两个Foo对象。使用new操作符对第一个对象进行创建,然后使用构造函数在shared_ptr中创建这一对象。直接使用make_shared<Foo>对第二个实例进行创建,使用我们提供的参数创建一个Foo实例。这种创建的方式很优雅,使用auto进行类型推断,对象也算第一次访问。这里与unique_ptr很类似:

        {
            cout << "Inner scope begin\n";
    
            shared_ptr<Foo> f1 {new Foo{"foo"}};
            auto f2 (make_shared<Foo>("bar"));
  6. 当共享指针被共享时,需要记录有多少个指针共享了这个对象。这需要内部引用计数器或使用独立计数器完成,我们可以使用use_count将这个值进行输出。现在其值为1,因为其还没进行拷贝。我们可以f1拷贝到fa,其计数值将会为2。

        cout << "f1's use counter at " << f1.use_count() << '\n';
        fa = f1;
        cout << "f1's use counter at " << f1.use_count() << '\n';
  7. 离开这个代码段时,共享指针f1f2将会被销毁。f1变量引用计数将会减少1,现在只有fa拥有这个Foo实例。当f2被回收时,其引用计数将减为0。因此,shared_ptr指针将对对象进行销毁:

        }
        cout << "Back to outer scope\n";
    
        cout << fa.use_count() << '\n';
  8. 现在,让我们用两种方式调用f函数。第一种,我们使用直接拷贝fa的方式。f函数将会将引用计数输出,值为2。在第二次对f的调用时,我们将指针移动到函数中。现在只有f函数对其指向的对象具有所有权:

        cout << "first f() call\n";
        f(fa);
        cout << "second f() call\n";
        f(move(fa));
  9. f返回之后,Foo实例就被立即销毁,因为没有任何指针对其有所有权。因此,在主函数返回前,所有对象就都会被销毁:

        cout << "end of main()\n";
    }
  10. 编译并运行程序就会得到如下输出。起初,我们可以看到foobar被创建。然后指针的副本f1出现(其指向foo实例),引用计数增加到2。当离开代码段时,因为没有任何指针在对指向bar实例的共享指针具有所有权,所以其会自动进行销毁。现在fa的引用计数为1,因为现在只有fafoo对象具有所有权。之后,我们调用了两次f函数。第一次调用,我们对fa进行了拷贝,会再次将引用计数增为2。第二次调用时,我们将fa移动到f中,其对引用计数的数值并无影响。此外,因为f函数具有了foo对象指针的所有权,所以当f函数结束时,foo对象就自动销毁了。主函数打印出最后一行前,堆上分配的动态对象就会被全部销毁:

    $ ./shared_ptr
    Inner scope begin
    CTOR foo
    CTOR bar
    f1's use counter at 1
    f1's use counter at 2
    DTOR bar
    Back to outer scope
    1
    first f() call
    f: use counter at 2
    second f() call
    f: use counter at 1
    DTOR foo
    end of main()

How it works...

shared_ptr的工作方式与unique_ptr的类似。构造共享指针和唯一指针的方法也非常类似(使用make_shared函数创建共享对象的指针,使用make_unique创建unique_pointer)。

unique_ptrshared_pointer的最大区别在于可复制性上,因为共享指针内部具有一块控制区域(control block),其中有用来管理对象的指针,还有一个计数器。当有N个shared_ptr实例指向某个对象时,其内部的计数器的值就为N。当shared_ptr实例销毁时,内部计数器会减1。当没有指针对对象具有所有权时,计数器的值即为0,对象就会被自动销毁。这样我们就不用担心内存泄漏了。

为了更加形象的说明,我们来看一下下面的图:

第1步中,我们具有两shared_ptr实例用于管理Foo类型的一个对象。所以其引用个数为2。然后,shared_ptr2被销毁,计数就会变为1。因为还有指针指向其实例,所以Foo对象并未被销毁。第3步中,最后一个共享指针也被销毁了,这就导致引用计数为0。第4步会很快在第3步之后发生,所有控制块和Foo实例都会被销毁,并且其内存也会在堆上释放。

了解了shared_ptrunique_ptr,我们将能很容易的对动态分配的对象进行管理,并且不用担心出现内存泄漏。不过,这里有个忠告——共享指针避免在循环引用的指针间进行,这样会让计数器无法归零,导致内存泄漏。

There's more...

来看一下下面的代码。你能告诉我这段代码是否会发生内存泄漏吗?

void function(shared_ptr<A>, shared_ptr<B>, int);
// "function" is defined somewhere else

// ...somewhere later in the code:
function(new A{}, new B{}, other_function());

你可能会反问我,“怎么可能有内存泄漏呢?”,A和B在分配后就放入shared_ptr类型中,并且其之后会进行释放,所以不会有内存泄漏。

你说的没错,当我们确定shared_ptr实例获取了对应的指针时,那么我们不会遇到内存泄漏。不过这个问题又好像有点琢磨不透。

我们调用函数f(x(), y(), z())时,编译器需要在f前找到xyz函数的定义,并先运行这些函数,然后将其返回值传入f函数中。结合我们上面的例子来说的话,对于编译器来说执行xyz函数的顺序并没有被规定。

回看下我们给出的例子,当编译器决定首先进行new A{}的操作,然后进行other_function(),再进行new B{}操作时,那么能确保这些操作的结果会传到function中吗?当other_function()抛出一个异常,因为没有使用shared_ptr对A进行管理,所以会造成内存泄漏。无论我们如何捕获这个异常,对这个对象的处理机会已经不在,我们无法将其删除。

这里有两种方法可以规避这个问题:

// 1.)
function(make_shared<A>(), make_shared<B>(), other_function());

// 2.)
shared_ptr<A> ap {new A{}};
shared_ptr<B> bp {new B{}};
function(ap, bp, other_function());

这样,对象在传入函数之前就被shared_ptr所保管,也就无所谓函数是否会在中途抛出异常了。

Last updated