i2pd/Log.h

143 lines
2.5 KiB
C
Raw Normal View History

2013-12-10 17:00:13 +04:00
#ifndef LOG_H__
#define LOG_H__
2014-04-24 19:10:46 +04:00
#include <string>
2013-12-10 17:00:13 +04:00
#include <iostream>
#include <sstream>
2014-04-24 19:10:46 +04:00
#include <fstream>
2014-04-23 20:49:02 +04:00
#include <functional>
#include <chrono>
2016-02-04 20:36:58 +03:00
#include <memory>
2013-12-10 17:00:13 +04:00
#include "Queue.h"
2014-10-28 23:36:17 +03:00
enum LogLevel
{
eLogError = 0,
eLogWarning,
eLogInfo,
eLogDebug,
eNumLogLevels
};
class Log;
2013-12-10 17:00:13 +04:00
struct LogMsg
{
std::stringstream s;
Log * log;
2014-10-28 23:36:17 +03:00
LogLevel level;
2013-12-10 17:00:13 +04:00
LogMsg (Log * l = nullptr, LogLevel lv = eLogInfo): log (l), level (lv) {};
2013-12-10 17:00:13 +04:00
void Process();
2013-12-10 17:00:13 +04:00
};
2014-04-23 20:49:02 +04:00
class Log: public i2p::util::MsgQueue<LogMsg>
{
public:
2016-02-04 20:36:58 +03:00
Log () { SetOnEmpty (std::bind (&Log::Flush, this)); };
~Log () {};
2014-04-24 19:10:46 +04:00
void SetLogFile (const std::string& fullFilePath);
2016-02-04 21:53:38 +03:00
void ReopenLogFile ();
2015-12-28 03:00:00 +03:00
void SetLogLevel (const std::string& level);
2016-02-04 20:36:58 +03:00
void SetLogStream (std::shared_ptr<std::ostream> logStream);
std::shared_ptr<std::ostream> GetLogStream () const { return m_LogStream; };
const std::string& GetTimestamp ();
2015-12-28 03:00:00 +03:00
LogLevel GetLogLevel () { return m_MinLevel; };
2014-04-23 20:49:02 +04:00
private:
void Flush ();
2014-04-24 19:10:46 +04:00
private:
2016-02-04 21:53:38 +03:00
std::string m_FullFilePath; // empty if stream
2016-02-04 20:36:58 +03:00
std::shared_ptr<std::ostream> m_LogStream;
2015-12-28 03:00:00 +03:00
enum LogLevel m_MinLevel;
std::string m_Timestamp;
2015-11-20 18:02:54 +03:00
#if (__GNUC__ == 4) && (__GNUC_MINOR__ <= 6) && !defined(__clang__) // gcc 4.6
2015-05-07 02:18:00 +03:00
std::chrono::monotonic_clock::time_point m_LastTimestampUpdate;
#else
std::chrono::steady_clock::time_point m_LastTimestampUpdate;
#endif
2014-04-23 20:49:02 +04:00
};
2014-07-02 21:48:45 +04:00
extern Log * g_Log;
inline void StartLog (const std::string& fullFilePath)
{
if (!g_Log)
{
2015-05-09 04:42:28 +03:00
auto log = new Log ();
2014-07-02 22:25:57 +04:00
if (fullFilePath.length () > 0)
2015-05-09 04:42:28 +03:00
log->SetLogFile (fullFilePath);
g_Log = log;
2014-07-02 21:48:45 +04:00
}
}
2016-02-04 20:36:58 +03:00
inline void StartLog (std::shared_ptr<std::ostream> s)
{
if (!g_Log)
{
2015-05-09 04:42:28 +03:00
auto log = new Log ();
if (s)
2015-05-09 04:42:28 +03:00
log->SetLogStream (s);
g_Log = log;
}
}
2014-07-02 21:48:45 +04:00
inline void StopLog ()
{
if (g_Log)
{
2015-05-09 04:42:28 +03:00
auto log = g_Log;
2014-07-02 21:48:45 +04:00
g_Log = nullptr;
2015-05-09 04:42:28 +03:00
log->Stop ();
delete log;
2014-07-02 21:48:45 +04:00
}
}
2013-12-10 17:00:13 +04:00
2016-02-02 20:16:29 +03:00
inline void SetLogLevel (const std::string& level)
{
if (g_Log)
g_Log->SetLogLevel(level);
}
2016-02-04 21:53:38 +03:00
inline void ReopenLogFile ()
{
if (g_Log)
g_Log->ReopenLogFile ();
}
2013-12-10 17:00:13 +04:00
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>
2014-10-28 23:36:17 +03:00
void LogPrint (LogLevel level, TArgs... args)
2013-12-10 17:00:13 +04:00
{
2015-12-28 03:00:00 +03:00
if (g_Log && level > g_Log->GetLogLevel ())
return;
LogMsg * msg = new LogMsg (g_Log, level);
2013-12-10 17:00:13 +04:00
LogPrint (msg->s, args...);
msg->s << std::endl;
2015-12-28 03:00:00 +03:00
if (g_Log) {
2014-07-02 21:48:45 +04:00
g_Log->Put (msg);
2015-12-28 03:00:00 +03:00
} else {
2014-07-02 21:48:45 +04:00
msg->Process ();
delete msg;
}
2014-10-28 23:36:17 +03:00
}
2013-12-10 17:00:13 +04:00
#endif