Program Listing for File logger.cc

↰ Return to documentation for file (src/sparsebase/utils/logger.cc)

#include "logger.h"

#include <algorithm>
#include <ctime>
#include <iostream>
#include <sstream>
#include <string>
#include <typeindex>

#include "utils.h"
namespace sparsebase::utils {

Logger::Logger(std::type_index root_type) {
  root_ = utils::demangle(root_type);
  root_.erase(std::remove(root_.begin(), root_.end(), '*'), root_.end());
  if (!Logger::filename_.empty()) file_.open(filename_);
}

Logger::Logger() {
  if (!Logger::filename_.empty()) file_.open(filename_);
}

Logger::~Logger() {
  if (file_.is_open()) file_.close();
}

void Logger::Log(const std::string& message, LogLevel msg_level) {
  if (msg_level < Logger::level_) {
    return;
  }

  std::time_t current = std::time(0);
  auto now_tm = std::localtime(&current);
  char buffer[30];
  size_t size = strftime(buffer, 30, "%x %X", now_tm);
  std::string now_str(buffer, buffer + size);

  std::stringstream ss;
  ss << "[" << now_str << "]"
     << " ";

  std::string level_str = "INFO";
  if (msg_level == LOG_LVL_WARNING) {
    level_str = "WARNING";
  }
  ss << "[" << level_str << "]"
     << " ";

  ss << "[" << root_ << "]"
     << " ";

  ss << message;

  std::string log = ss.str();

  if (file_.is_open()) file_ << log << std::endl;
  if (Logger::use_stdout_) std::cout << log << std::endl;
  if (Logger::use_stderr_) std::cerr << log << std::endl;
}

LogLevel Logger::level_ = LOG_LVL_WARNING;
std::string Logger::filename_;
bool Logger::use_stdout_ = true;
bool Logger::use_stderr_ = false;
}  // namespace sparsebase::utils