c++ - 如何在父类和子类中使用std::enable_shared_from_this？

Question

我有两个类A和B，其中B是A的子类。我需要两个类都可以使用std::enable_shared_from_this。

我尝试了这个:

#include <memory>
#include <iostream>
#include <vector>


class A : public std::enable_shared_from_this<A> {
  public:
    void insertme(std::vector<std::shared_ptr<A>>& v) {
        std::cout << "A::insertme\n";
        v.push_back(shared_from_this());
        std::cout << "OK\n";
    }
};

class B : public A, public std::enable_shared_from_this<B> {
  public:
    void insertme(std::vector<std::shared_ptr<B>>& v) {
        std::cout << "B::insertme\n";
        v.push_back(std::enable_shared_from_this<B>::shared_from_this());
        std::cout << "OK\n";
    }
};

int main()
{
    std::vector<std::shared_ptr<A>> va;
    std::vector<std::shared_ptr<B>> vb;

    std::shared_ptr<A> pa = std::make_shared<A>();
    std::shared_ptr<B> pb = std::make_shared<B>();

    pa->insertme(va);
    pb->insertme(vb);
}

(为了避免 shared_from_this()含糊不清，我必须在 B::insertme中完全限定它。)

当我运行上面的程序时，得到以下输出:

A::insertme
OK
B::insertme
terminate called after throwing an instance of 'std::bad_weak_ptr'
  what():  bad_weak_ptr
Aborted (core dumped)

因此， A::insertme有效，但 B::insertme无效。

我在Linux下使用GCC 9.1.0。

我究竟做错了什么？

Answer 1

您只需要(并且只能)从基类的shared_from_this继承:

class A : public std::enable_shared_from_this<A> {
  public:
    void insertme(std::vector<std::shared_ptr<A>>& v) {
        std::cout << "A::insertme\n";
        v.push_back(shared_from_this());
        std::cout << "OK\n";
    }
};

class B : public A {
  public:
    void insertme(std::vector<std::shared_ptr<B>>& v) {
        std::cout << "B::insertme\n";
        v.push_back(shared_from_this()->static_pointer_cast<B>());
        std::cout << "OK\n";
    }
};

这意味着您需要显式的 static_pointer_cast来获取 shared_ptr<B>，但是如果需要，可以将其包装在 B中的替代项中:

std::shared_ptr<B> shared_from_this() { return A::shared_from_this()->static_pointer_cast<B>(); }