c++ 重定向 cout / cerr 到自定义日志方法

Posted 2023-02-03 Eritque arcus

代码:

#include <iostream>
#include <sstream>
#define log(x)
#define err(x)

class OStreamRedirector 
public:
    ~OStreamRedirector() 
        obj.rdbuf(old);
    

    /**
         * @brief 重定向 obj 的流
         * @param obj 需要重定向的流
         * @param new_buffer 重定向到的新缓冲区
         */
    explicit OStreamRedirector(std::ostream &obj, std::streambuf *newBuffer)
        : old(obj.rdbuf(newBuffer)), obj(obj) 

private:
    // 被重定向的流
    std::ostream &obj;
    // 旧的缓冲区目标
    std::streambuf *old;
;

class OString : private std::streambuf, public std::ostream 
public:
    // 输出是否为 info 级别
    explicit OString(bool info) : std::ostream(this), info(info) 

private:
    bool info;
    // 缓冲区
    std::ostringstream result;
    // 加入缓冲区
    int overflow(std::streambuf::int_type c) override 
        if (c == EOF)
            out();
        else
            result.put((std::streambuf::char_type) c);
            return c;
        
        // std::streambuff 的 sync
        int sync() override 
            out();
            return 0;
        
        // 输出缓冲区内容, 相当于 flush
        void out() 
        // 执行log系统输出
            if (info)
                log(result.str());
            else
                err(result.str());
            result.str("");
        
    ;

用法:

OString outTarget(true);
OStreamRedirector outRedirectorstd::cout, outTarget.rdbuf();
OString errTarget(false);
OStreamRedirector errRedirectorstd::cerr, errTarget.rdbuf();

值得注意两个点:

1. 是在overflow()里面不能判断输入流是否结束, 不会出现传入 c == EOF 的情况
2. 是 cout 和 cerr 不会自动 flush(), 可能需要调用std::endl或者手动 flush

--EOF--