C++ 自定义调试类如何重载<<运算符实现终端或文件流式输出

结论 #

不需要自定义流子类，直接通过重载<<运算符即可实现需求，核心逻辑是把所有输出操作转发给类内部持有的std::ostream指针指向的流对象。

实现方案 #

注意事项 #

• 重载<<需要用模板实现，才能兼容所有标准库支持的可输出类型、以及std::endl这类流操纵符
• 切换输出文件时需要先关闭之前已经打开的文件流，避免资源泄漏
• 构造函数建议初始化流指针为std::cout，避免未调用setOutFile直接输出时空指针崩溃
• 单独增加流操纵符的重载，避免std::endl这类函数模板参数推导失败

完整实现代码 #

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

class MyDbg {
protected:
    std::ostream* fp;
    std::ofstream fout;

public:
    // 构造函数默认输出到终端
    MyDbg() : fp(&std::cout) {}

    // 析构函数自动关闭打开的文件流
    ~MyDbg() {
        if (fout.is_open()) {
            fout.close();
        }
    }

    void setOutFile(std::string filename) {
        // 先关闭之前打开的文件
        if (fout.is_open()) {
            fout.close();
        }

        if (filename == "terminal") {
            fp = &std::cout;
        } else {
            fout.open(filename);
            fp = &fout;
        }
    }

    // 模板重载，兼容所有可输出类型
    template<typename T>
    MyDbg& operator<<(const T& val) {
        *fp << val;
        return *this;
    }

    // 单独重载，兼容std::endl等流操纵符
    MyDbg& operator<<(std::ostream& (*manip)(std::ostream&)) {
        *fp << manip;
        return *this;
    }
};

测试代码 #

和你期望的调用方式完全兼容，同时支持所有基础类型、流对象输出：

int main() {
  MyDbg dbg;
  dbg.setOutFile("terminal");
  dbg << "foo" << std::endl;
  dbg.setOutFile("dbg.log");
  dbg << "bar" << std::endl;
  // 支持其他类型输出
  dbg << 123 << " " << 3.14 << std::endl;
  return 0;
}

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