C++ stream as a member variable

Madeleine P. Vincent picture Madeleine P. Vincent · Dec 9, 2011 · Viewed 7.3k times · Source

I've got a C++ class which I would like to hold a stream used for logging.

The stream should be able to be set (and possibly reset) after the construction of the object.

It should be possible to set the stream as std::cout, or as a file stream to log to a file, or as a stringstream which does nothing more than ignore the data (a /dev/null of sorts). In any case, it should be an ostream type object, which the creator of the object can reset at any time. The class itself is oblivious to the concrete stream type.

I could accomplish this with a pointer to an ostream, but then the syntax becomes a little annoying, having to use the deref operator:

(*m_log) << "message";

rather than

m_log << "message";

But I can't use references, as the stream object needs to be possibly reset after the object has been initialized.

Is there an elegant way to achieve this, i.e., avoid using pointers, but still be able to reset after construction?

Answer

sehe picture sehe · Dec 9, 2011

You can reset streams: see it live on https://ideone.com/Ci4eo

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

struct Logger
{
    Logger(std::ostream& os) : m_log(os.rdbuf()) { }

    std::streambuf* reset(std::ostream& os) 
    {
        return m_log.rdbuf(os.rdbuf());
    }

    template <typename T> friend Logger& operator<<(Logger& os, const T& t)
    { os.m_log << t; return os; }

    friend Logger& operator<<(Logger& os, std::ostream& ( *pf )(std::ostream&))
    { os.m_log << pf; return os; }

  private:
    std::ostream m_log;
};

int main(int argc, const char *argv[])
{
    Logger logto(std::cout);

    logto << "Hello world" << std::endl;

    logto.reset(std::cerr);
    logto << "Error world" << std::endl;

    return 0;
}