如何改进我的类来创建输出文件

Question

由于我经常需要这个，我想写一个类来处理流的主要活动。

我可以这样使用的东西：

OutputFile out("output.txt");
for (int i = 0; i < 10; ++i)
out << i << "n";

为此，我编写了以下类：

class OutputFile {
std::string filename;
std::ofstream out;
public:
explicit OutputFile(std::string filename) {
this->filename = filename;
out.open("output/" + filename);
}
~OutputFile() {
out.close();
}
std::ofstream& operator()() { return out; }
};

这几乎是我想要的，但是我重载了运算符()，所以在上面的例子中，我必须使用

out() << i << "n";

我应该如何修改我的类，以便用作

out << i << "n";

Answer 1

您可以在类中重载operator<<。

class OutputFile {
std::string filename;
std::ofstream out;
public:
explicit OutputFile(const std::string &filename)
: filename(filename), out("output/" + filename) {}
template<typename T>
OutputFile& operator<<(const T &value) {
out << value;
return *this;
}
};