c++ std::enable_shared_from_this 的作用

在类的内部获取shared_ptr是在所难免的。

举个例子：

在日常c++编程中，为了更好的管理资源，我们通常借助shared_ptr来达到对资源的自动管理。由于其原理是通过过跟踪引用计数实现的，也就是说在使用了shared_ptr后就不能再使用裸指针this。比如说在类的内部直接使用std::shared_ptr(this)，就会导致智能指针失效。

常见的场景：

通过std::bind生成functor函数传递给其他实例时（暴露了this），当其他实例调用functor函数时，该实例有可能已经被析构，会引起异常。

首先我们看下shared_ptr的用法，

class Test{
        public:
                Test() {
                        a = 123;
                        std::cout << "construct\n";
                };
                void func() {
                        std::cout << "test func called data="<< a << "\n";
                };
                ~Test() {
                        std::cout << "destroy\n";
                };
        private:
                int a;
};

int main()
{
        {
                std::shared_ptr b(new Test());
                b->func();
        }
}

// construct
// test func called data=123
// destroy

在某些情况，当我们在func内部将this传递给其他对象后，由于使用了裸指针，导致shared_ptr无法正常跟踪计数。最终Test会被析构两次，导致未定义行为。下面的代码展示了这个问题。

#include 
#include 

class Test
{
public:
    std::shared_ptr getptr() {
        return std::shared_ptr(this);
    }
    ~Test() { std::cout << "Test::~Test() called" << std::endl; }
};

int main()
{

    std::shared_ptr a(new Test());
    std::shared_ptr b = a->getptr();

    std::cout << "a.use_count() = " << a.use_count() << std::endl; // 1
    std::cout << "b.use_count() = " << b.use_count() << std::endl; // 1

}

使用 enable_shared_from_this来修正这个问题

#include 
#include 

class Test :public std::enable_shared_from_this
{
public:
        std::shared_ptr getptr() {
                return shared_from_this();
        }
        ~Test() { std::cout << "Test::~Test() called" << std::endl; }
};

int main()
{

        std::shared_ptr a(new Test());
        std::shared_ptr b = a->getptr();

        std::cout << "a.use_count() = " << a.use_count() << std::endl; // 2
        std::cout << "b.use_count() = " << b.use_count() << std::endl; // 2

}

这样shared_ptr就能正确的取得计数。避免了在内部无法正确取得shared_ptr的困境。