如何在自定义TmpFile类中重载fstream的<<运算符？

完全可以通过重载operator<<来实现你想要的写法，核心思路是让重载的运算符直接调用内部std::ofstream对象的写入功能，并支持链式调用。

修改后的TmpFile类示例代码：

#include <fstream>
#include <string>
#include <iostream> // 用于endl等流操作符

class TmpFile {
private:
    std::string tmp_file_name;
    std::ofstream tmp_file;

public:
    TmpFile(const std::string &name) : tmp_file_name(name),
                                       tmp_file(tmp_file_name, std::ios::out | std::ios::trunc) {}
    
    ~TmpFile() {
        // std::ofstream的析构函数会自动关闭文件，这里的close()可以省略
        // tmp_file.close();
        // 后续额外操作
        ...
    }

    // 通用模板版本，支持所有std::ostream能处理的数据类型
    template <typename T>
    TmpFile& operator<<(const T& data) {
        if (tmp_file.is_open()) { // 先检查文件是否成功打开
            tmp_file << data;
        }
        return *this; // 返回自身以支持链式调用
    }

    // 单独重载用于处理endl、flush这类流操作函数
    TmpFile& operator<<(std::ostream& (*manipulator)(std::ostream&)) {
        if (tmp_file.is_open()) {
            manipulator(tmp_file);
        }
        return *this;
    }
};

使用示例：

int main() {
    TmpFile f("test.txt");
    f << "hello" << " world" << std::endl;
    f << 123 << " " << 3.14 << std::flush;
    return 0;
}

说明：

• 模板版本的operator<<可以兼容所有标准输出流支持的类型（字符串、整数、浮点数等），不用逐个重载。
• 针对流操作符（如std::endl）的重载是因为这类操作符本质是函数指针，需要单独处理才能正常使用。
• 添加文件打开状态检查可以避免文件打开失败时的无效写入操作。

内容的提问来源于stack exchange，提问作者f1msch