LogConfiguration.cpp

Go to the documentation of this file.

#include "LogConfiguration.hpp"
#include <algorithm>
#include <boost/core/null_deleter.hpp>
#include <boost/log/attributes/mutable_constant.hpp>
#include <boost/log/core.hpp>
#include <boost/log/expressions.hpp>
#include <boost/log/sinks/sink.hpp>
#include <boost/log/support/date_time.hpp>
#include <boost/log/trivial.hpp>
#include <boost/log/utility/setup/console.hpp>
#include <boost/program_options.hpp>
#include <deque>
#include <filesystem>
#include <fstream>
#include <iostream>
#include <iterator>
#include <map>
#include <sstream>
#include <string>
#include <utility>
#include "utils/String.hpp"
#include "utils/assertion.hpp"
 
namespace precice::logging {
 
class timestamp_formatter_factory : public boost::log::basic_formatter_factory<char, boost::posix_time::ptime> {
public:
  formatter_type create_formatter(boost::log::attribute_name const &name, args_map const &args) override
  {
    namespace expr              = boost::log::expressions;
    args_map::const_iterator it = args.find("format");
    if (it != args.end())
      return expr::stream << expr::format_date_time<boost::posix_time::ptime>(expr::attr<boost::posix_time::ptime>(name), it->second);
    else
      return expr::stream << expr::attr<boost::posix_time::ptime>(name);
  }
};
 
class colorized_severity_formatter_factory : public boost::log::formatter_factory<char> {
public:
  formatter_type create_formatter(boost::log::attribute_name const &name, args_map const &args) override
  {
    namespace expr = boost::log::expressions;
    auto severity  = expr::attr<boost::log::trivial::severity_level>("Severity");
 
    return expr::stream
           << expr::if_(severity == boost::log::trivial::severity_level::error)
                  [expr::stream << "\033[31m" //red
                                << "ERROR: "]
           << expr::if_(severity == boost::log::trivial::severity_level::warning)
                  [expr::stream << "\033[36m" //cyan
                                << "WARNING: "]
           << "\033[0m";
  }
};
 
class severity_formatter_factory : public boost::log::formatter_factory<char> {
public:
  formatter_type create_formatter(boost::log::attribute_name const &name, args_map const &args) override
  {
    namespace expr = boost::log::expressions;
    auto severity  = expr::attr<boost::log::trivial::severity_level>("Severity");
 
    return expr::stream
           << expr::if_(severity == boost::log::trivial::severity_level::error)
                  [expr::stream
                   << "ERROR: "]
           << expr::if_(severity == boost::log::trivial::severity_level::warning)
                  [expr::stream
                   << "WARNING: "];
  }
};
 
class NullSink final : public boost::log::sinks::sink {
public:
  NullSink()
      : boost::log::sinks::sink(false) {}
 
  bool will_consume(boost::log::attribute_value_set const &) override
  {
    return false;
  }
 
  void consume(boost::log::record_view const &) override {}
 
  bool try_consume(boost::log::record_view const &) override
  {
    return false;
  }
 
  void flush() override {}
 
  bool is_cross_thread() const noexcept
  {
    return false;
  }
};
 
 
class StreamBackend final : public boost::log::sinks::text_ostream_backend {
private:
  boost::shared_ptr<std::ostream> _ostream;
 
public:
  explicit StreamBackend(boost::shared_ptr<std::ostream> ostream)
      : _ostream(std::move(ostream)) {}
 
  void consume(boost::log::record_view const &rec, string_type const &formatted_record)
  {
    *_ostream << formatted_record << '\n'
              << std::flush;
  }
};
 
LoggingConfiguration readLogConfFile(std::string const &filename)
{
  if (!std::filesystem::exists(filename)) {
    return {};
  }
 
  namespace po = boost::program_options;
  po::options_description desc;
  std::ifstream           ifs(filename);
 
  std::map<std::string, BackendConfiguration> configs;
  try {
    po::parsed_options parsed = parse_config_file(ifs, desc, true);
    for (auto const &opt : parsed.options) {
      std::string section = opt.string_key.substr(0, opt.string_key.find('.'));
      std::string key     = opt.string_key.substr(opt.string_key.find('.') + 1);
      if (BackendConfiguration::isValidOption(key)) {
        PRECICE_ASSERT(!opt.value.empty());
        configs[section].setOption(key, opt.value[0]);
      } else {
        std::cerr << "WARNING: section [" << section << "] in configuration file \"" << filename << "\" contains invalid key \"" << key << "\"\n";
      }
    }
  } catch (po::error &e) {
    std::cout << "ERROR reading logging configuration: " << e.what() << "\n\n";
    std::exit(-1);
  }
 
  LoggingConfiguration retVal;
 
  for (auto const &c : configs)
    if (c.second.enabled)
      retVal.push_back(c.second);
 
  return retVal;
}
 
// Default values for filter and format. They are also used from config/LogConfiguration.cpp
const std::string BackendConfiguration::default_filter    = "(%Severity% > debug) and not ((%Severity% = info) and (%Rank% != 0))";
const std::string BackendConfiguration::default_formatter = "(%Rank%) %TimeStamp(format=\"%H:%M:%S\")% [%Module%]:%Line% in %Function%: %ColorizedSeverity%%Message%";
const std::string BackendConfiguration::default_type      = "stream";
const std::string BackendConfiguration::default_output    = "stdout";
 
void BackendConfiguration::setOption(std::string key, std::string value)
{
  boost::algorithm::to_lower(key);
  if (key == "type") {
    boost::algorithm::to_lower(value);
    type = value;
  }
  if (key == "output")
    output = value;
  if (key == "filter")
    filter = value;
  if (key == "format")
    format = value;
}
 
bool BackendConfiguration::isValidOption(std::string key)
{
  boost::algorithm::to_lower(key);
  return key == "output" || key == "filter" || key == "format" || key == "type";
}
 
void BackendConfiguration::setEnabled(bool enabled)
{
  this->enabled = enabled;
}
 
void setupLogging(LoggingConfiguration configs, bool enabled)
{
  if (getGlobalLoggingConfig().locked)
    return;
 
  namespace bl = boost::log;
  bl::register_formatter_factory("TimeStamp", boost::make_shared<timestamp_formatter_factory>());
  bl::register_formatter_factory("ColorizedSeverity", boost::make_shared<colorized_severity_formatter_factory>());
  bl::register_formatter_factory("Severity", boost::make_shared<severity_formatter_factory>());
  bl::register_simple_filter_factory<bl::trivial::severity_level, char>("Severity");
 
  // Possible, longer output format. Currently unused.
  auto fmtStream =
      bl::expressions::stream
      << "(" << bl::expressions::attr<int>("Rank") << ") "
      << bl::expressions::format_date_time<boost::posix_time::ptime>("TimeStamp", "%H:%M:%S") << " "
      << bl::expressions::attr<std::string>("File") << ":"
      << bl::expressions::attr<int>("Line")
      << " [" << bl::expressions::attr<std::string>("Module") << "] in "
      << bl::expressions::attr<std::string>("Function") << ": "
      << bl::expressions::message;
 
  // Remove active preCICE sinks
  using sink_ptr = typename boost::shared_ptr<boost::log::sinks::sink>;
 
  static std::vector<sink_ptr> activeSinks;
  for (auto &sink : activeSinks) {
    boost::log::core::get()->remove_sink(sink);
    sink->flush();
    sink.reset();
  }
  activeSinks.clear();
 
  // If logging sinks are disabled, then we need to disable the default sink.
  // We do this by adding a NullSink.
  // We need to exit after the sink removal as the default sink exists before
  // the log configuration is parsed.
  if (auto noconfigs = std::none_of(configs.begin(), configs.end(), [](const auto &config) { return config.enabled; });
      !enabled && noconfigs) {
    auto sink = boost::make_shared<NullSink>();
    boost::log::core::get()->add_sink(sink);
    activeSinks.emplace_back(std::move(sink));
    return;
  }
 
  // Add the default config in case no sinks are configured
  if (configs.empty()) {
    configs.emplace_back();
  }
 
  // Create new sinks
  for (const auto &config : configs) {
    if (!config.enabled) {
      continue;
    }
 
    // Setup backend of sink
    boost::shared_ptr<StreamBackend> backend;
    if (config.type == "file")
      backend = boost::make_shared<StreamBackend>(boost::shared_ptr<std::ostream>(new std::ofstream(config.output)));
    if (config.type == "stream") {
      if (config.output == "stdout")
        backend = boost::make_shared<StreamBackend>(boost::shared_ptr<std::ostream>(&std::cout, boost::null_deleter()));
      if (config.output == "stderr")
        backend = boost::make_shared<StreamBackend>(boost::shared_ptr<std::ostream>(&std::cerr, boost::null_deleter()));
    }
    PRECICE_ASSERT(backend != nullptr, "The logging backend was not initialized properly. Check your log config.");
    backend->auto_flush(true);
 
    // Setup sink
    auto sink = boost::make_shared<boost::log::sinks::synchronous_sink<StreamBackend>>(backend);
    sink->set_formatter(boost::log::parse_formatter(config.format));
 
    if (config.filter.empty()) {
      sink->set_filter(boost::log::expressions::attr<bool>("preCICE") == true);
    } else {
      // We extend the filter here to filter all log entries not originating from preCICE.
      sink->set_filter(boost::log::parse_filter("%preCICE% & ( " + config.filter + " )"));
    }
 
    boost::log::core::get()->add_sink(sink);
    activeSinks.emplace_back(std::move(sink));
  }
}
 
void setupLogging(std::string const &logConfigFile)
{
  setupLogging(readLogConfFile(logConfigFile));
}
 
void setMPIRank(int const rank)
{
  getGlobalLoggingConfig().rank = rank;
}
 
void setParticipant(std::string const &participant)
{
  getGlobalLoggingConfig().participant = participant;
}
 
GlobalLoggingConfig &getGlobalLoggingConfig()
{
  static GlobalLoggingConfig instance;
  return instance;
}
 
void lockConf()
{
  getGlobalLoggingConfig().locked = true;
}
 
} // namespace precice::logging