#ifndef LOG_H__ #define LOG_H__ #include #include #include #include #include #include "Queue.h" struct LogMsg { std::stringstream s; std::ostream& output; LogMsg (std::ostream& o = std::cout): output (o) {}; void Process(); }; class Log: public i2p::util::MsgQueue { public: Log (): m_LogFile (nullptr) { SetOnEmpty (std::bind (&Log::Flush, this)); }; ~Log () { delete m_LogFile; }; void SetLogFile (const std::string& fullFilePath); std::ofstream * GetLogFile () const { return m_LogFile; }; private: void Flush (); private: std::ofstream * m_LogFile; }; extern Log * g_Log; inline void StartLog (const std::string& fullFilePath) { if (!g_Log) { g_Log = new Log (); if (fullFilePath.length () > 0) g_Log->SetLogFile (fullFilePath); } } inline void StopLog () { if (g_Log) { delete g_Log; g_Log = nullptr; } } template void LogPrint (std::stringstream& s, TValue arg) { s << arg; } template void LogPrint (std::stringstream& s, TValue arg, TArgs... args) { LogPrint (s, arg); LogPrint (s, args...); } template void LogPrint (TArgs... args) { LogMsg * msg = (g_Log && g_Log->GetLogFile ()) ? new LogMsg (*g_Log->GetLogFile ()) : new LogMsg (); LogPrint (msg->s, args...); msg->s << std::endl; if (g_Log) g_Log->Put (msg); else { msg->Process (); delete msg; } } #endif