从现有 istream 或类本身创建的 istream 初始化成员 istream

Question

我试图弄清楚如何拥有一个成员变量，它要么代表传入的 istream，要么代表类自己创建的成员变量。

我认为如果我动态分配类创建的istream，则使用指向istream 的指针可以工作；但是，问题在于unique_ptr 将尝试释放非动态分配的内存。

这里有一些代码可以重现我遇到的问题：

#include <iostream>
#include <fstream>
#include <memory>
#include <string>

class example {
public:
  explicit example(std::istream& i)
    : m_input(&i)
  {}
  explicit example(const std::string& path)
    : m_input(new std::ifstream(path))
  {}
private:
  std::unique_ptr<std::istream> m_input;
};

int main() {
  example e1(std::cin);
  example e2("./test.txt");
}

e1 将尝试释放 std::cin，这会导致错误。我知道我可以使用多个成员，例如

#include <iostream>
#include <fstream>
#include <memory>
#include <string>

class example {
public:
  explicit example(std::istream& i)
    : m_i(),
      m_input(&i)
  {}
  explicit example(const std::string& path)
    : m_i(path),
      m_input(&m_i)
  {}
private:
  std::ifstream m_i;
  std::istream *m_input;
};

int main() {
  example e1(std::cin);
  example e2("./test.txt");
}

但我想知道是否有一种方法可以只使用一个成员变量

Answer 1

我猜你可能会通过存储指向流的指针来扩展你的方法：

class X {
    public:
        explicit X(std::istream& stream)
            : stream(&stream, [](std::istream* stream){})
        {}

        explicit X(const std::string& path)
            : stream(new std::ifstream(path),
                [](std::istream* stream) { delete stream; })
        {}

    private:
        using StreamDeleter = std::function<void(std::istream* stream)>;

        std::unique_ptr<std::istream, StreamDeleter> stream;
};

只需添加一个自定义删除器。在现有流的情况下，它什么也不做。相反，如果流是手动创建的，它会释放流。