Adding a general logger and integrating it with daggyr

This commit is contained in:
Ian Roddis
2022-02-04 11:58:45 -04:00
parent 57e93b5045
commit ea3f67f226
8 changed files with 245 additions and 20 deletions

View File

@@ -2,6 +2,7 @@ target_sources(${PROJECT_NAME} PRIVATE
Serialization.cpp
Utilities.cpp
DAGRunner.cpp
GeneralLogger.cpp
)
add_subdirectory(executors)

View 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