简单的 wostream 日志记录类（带有自定义流操纵器）

Question

我已经阅读了大量的问题、文章和文档，但我还没有找到解决我的问题的方法。

我想创建一个简单的类用于调试。最终结果将允许我做这样的事情：

logger << error << L"This is a problem!" << endl;
logger << warning << L"This might be a problem!" << endl;
logger << info << L"This isn't a problem but I thought you should know about it" << endl;

有了这样的想法，在记录器类中，我可以切换这些东西是否进入控制台/调试文件。

logger.setLevel(ERROR);

我已经有了一个骨架，但我无法让操作员重载以使操纵器工作。

这是 Logger.h:

class LoggerBuffer : public wfilebuf {
// Functions
    public:
        LoggerBuffer() { wfilebuf::open("NUL", ios::out); currentState = 1;}
        ~LoggerBuffer() {wcout << "DELETED!" << endl;}
        void open(const char fname[]);
        void close() {wfilebuf::close();} 
        virtual int sync();
        void setState(int newState);
// Variables
    private:
         int currentState;
};

class LoggerStream : public wostream {
// Functions
    public:
         LoggerStream() : wostream(new LoggerBuffer()), wios(0) {}
         ~LoggerStream() { delete rdbuf(); }
         void open(const char fname[] = 0) { 
    wcout << "Stream Opening " << fname << endl;((LoggerBuffer*)rdbuf())->open(fname); }
         void close() { ((LoggerBuffer*)rdbuf())->close(); }
         void setState(int newState);
};

和 Logger.cpp:

void LoggerBuffer::open(const char fname[]) {
    wcout << "Buffer Opening " << fname << endl;
    close();
    wfilebuf* temp = wfilebuf::open(fname, ios::out);
    wcout << "Temp: " << temp << endl;
}
int LoggerBuffer::sync() {
    wcout << "Current State: " << currentState << ", Data: " << pbase();
    return wfilebuf::sync();
}
void LoggerBuffer::setState(int newState) {
    wcout << "New buffer state = " << newState << endl;
    currentState = newState;
}

void LoggerStream::setState(int newState) {
    wcout << "New stream state = " << newState << endl;
    ((LoggerBuffer*)rdbuf())->setState(newState);
}

和 main.cpp:

struct doSetState {
    int _l;    
    doSetState ( int l ): _l ( l ) {}

    friend LoggerStream& operator<< (LoggerStream& os, doSetState fb ) {
        os.setState(3);
        return (os);
    }
};

...
LoggerStream test;
test.open("debug.txt");
test << "Setting state!" << doSetState(1) << endl;
...

这种混乱在 VS2005 中产生以下错误：

“错误 C2679：二进制 '<<' ： 不 发现运算符需要一个 类型的右侧操作数 'doSetState'（或者没有 可接受的转换）”

非常感谢任何帮助。

谢谢！

Answer 1

您的 ostream 操作员没有正确的签名。应该是：

friend LoggerStream&运算符<< (LoggerStream&os, const doSetState&fb)

（必须使用引用，因为单遍编译器在类定义中途不知道 doSetState 的大小。）

Answer 2

问题是，当您这样做时：

test << "Setting state!"

它返回一个基本的 wostream 对象。因此链接它不起作用，因为没有重载：

wostream& operator<< (wostream& os, const doSetState& fb )

但是您可以在单独的行上执行此操作，如下所示：

test << "Setting state!";
test << doSetState(1) << endl;

Answer 3

我会采取稍微不同的方法。

我的记录器类中有一个 std::wfostream 成员，而不是从 std::wostream 继承。然后，您可以拥有一个通用模板化operator<<，它有选择地转发到嵌入流。

例如：

class Logger;

template<class T> Logger& operator<<(Logger&, const T&);

enum LogLevel
{
    debug,
    info,
    warning,
    error
};

class Logger
{
public:
    void open(const char* file) { stream.open(file); }
    void close() { stream.close(); }
    void passLevel(Loglevel level) { pass = level; }
    void logLevel(LogLevel level) { current = level; }
private:
    bool passThrough() { return current >= pass; }

    std::wofstream stream;
    LogLevel pass;
    LogLevel current;

    friend template<class T> Logger& operator<<(Logger&, const T&);
};

template<class T> 
Logger& operator<<(Logger& log, const T& rhs)
{
    if (log.passthrough())
    {
        log.stream << rhs; 
    }
    return log;
}

Logger& operator<<(Logger&, LogLevel level)
{
    log.logLevel(level);
    return log;
}

struct setLogLevel {
    setLogLevel(LogLevel l) : level(l) { }
    LogLevel level;
};

Logger& operator<<(Logger&, const setLogLevel setter)
{
    log.passLevel(setter.level);
    return log;
}