Adding a general logger and integrating it with daggyr
This commit is contained in:
@@ -2,6 +2,7 @@ target_sources(${PROJECT_NAME} PRIVATE
|
||||
Serialization.cpp
|
||||
Utilities.cpp
|
||||
DAGRunner.cpp
|
||||
GeneralLogger.cpp
|
||||
)
|
||||
|
||||
add_subdirectory(executors)
|
||||
|
||||
66
libdaggy/src/GeneralLogger.cpp
Normal file
66
libdaggy/src/GeneralLogger.cpp
Normal file
@@ -0,0 +1,66 @@
|
||||
#include <daggy/GeneralLogger.hpp>
|
||||
#include <daggy/Serialization.hpp>
|
||||
|
||||
namespace daggy {
|
||||
GeneralLogger::GeneralLogger(std::ostream &os, LogLevel level)
|
||||
: running_(true)
|
||||
, os_(os)
|
||||
, level_(level)
|
||||
, messageEmiter_(&GeneralLogger::emitMessages, this)
|
||||
{
|
||||
}
|
||||
|
||||
GeneralLogger::~GeneralLogger()
|
||||
{
|
||||
shutdown();
|
||||
}
|
||||
|
||||
void GeneralLogger::shutdown()
|
||||
{
|
||||
if (!running_)
|
||||
return;
|
||||
running_ = false;
|
||||
newMessage_.notify_one();
|
||||
messageEmiter_.join();
|
||||
os_.flush();
|
||||
}
|
||||
|
||||
void GeneralLogger::setLevel(LogLevel level)
|
||||
{
|
||||
level_ = level;
|
||||
}
|
||||
void GeneralLogger::log(const std::string &msg, LogLevel level)
|
||||
{
|
||||
if (level > level_) {
|
||||
return;
|
||||
}
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(messageGuard_);
|
||||
messages_.emplace_back(
|
||||
LogMessage{.level = level, .msg = msg, .time = Clock::now()});
|
||||
}
|
||||
newMessage_.notify_one();
|
||||
}
|
||||
|
||||
void GeneralLogger::log(const std::function<std::string()> &fun,
|
||||
LogLevel level)
|
||||
{
|
||||
if (level > level_)
|
||||
return;
|
||||
log(fun(), level);
|
||||
}
|
||||
|
||||
void GeneralLogger::emitMessages()
|
||||
{
|
||||
while (running_ || !messages_.empty()) {
|
||||
std::unique_lock<std::mutex> lock(messageGuard_);
|
||||
newMessage_.wait(lock, [&] { return !(messages_.empty() && running_); });
|
||||
for (const auto &msg : messages_) {
|
||||
os_ << timePointToString(msg.time) << " [" << msg.level._to_string()
|
||||
<< "] " << msg.msg << '\n';
|
||||
}
|
||||
os_.flush();
|
||||
messages_.clear();
|
||||
}
|
||||
}
|
||||
} // namespace daggy
|
||||
Reference in New Issue
Block a user