mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2024-11-04 06:00:37 +00:00
90 lines
1.5 KiB
C++
90 lines
1.5 KiB
C++
#ifndef LOG_H__
|
|
#define LOG_H__
|
|
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <fstream>
|
|
#include <functional>
|
|
#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<LogMsg>
|
|
{
|
|
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<typename TValue>
|
|
void LogPrint (std::stringstream& s, TValue arg)
|
|
{
|
|
s << arg;
|
|
}
|
|
|
|
template<typename TValue, typename... TArgs>
|
|
void LogPrint (std::stringstream& s, TValue arg, TArgs... args)
|
|
{
|
|
LogPrint (s, arg);
|
|
LogPrint (s, args...);
|
|
}
|
|
|
|
template<typename... TArgs>
|
|
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
|