c++ - 从现有 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;
};

只需添加一个自定义删除器。对于现有流，它不执行任何操作。相反，如果流是手动创建的，则释放该流。