logger.hpp

Go to the documentation of this file.

/*
  _                                       _
 | |    ___   __ _  __ _  ___ _ __    ___| | __ _ ___ ___
 | |   / _ \ / _` |/ _` |/ _ \ '__|  / __| |/ _` / __/ __|
 | |__| (_) | (_| | (_| |  __/ |    | (__| | (_| \__ \__ \
 |_____\___/ \__, |\__, |\___|_|     \___|_|\__,_|___/___/
             |___/ |___/
 
This class subscribes to all messages published by the broker and logs them to a
MongoDB instance.
 
Author(s): Paolo Bosetti
*/
 
#ifndef LOGGER_HPP
#define LOGGER_HPP
 
 
#include "mads.hpp"
#include "agent.hpp"
#include <memory>
 
namespace Mads {
 
class Logger : public Agent {
public:
  Logger(std::string name, std::string settings_path);
 
  ~Logger();
 
  Logger(const Logger &) = delete;
  Logger &operator=(const Logger &) = delete;
 
  static bool has_mongo_support() noexcept;
 
  void set_mongo(bool enabled = true, string uri = "");
 
  void set_file(string filename, bool array = false);
 
  void set_file(bool enabled = false);
 
  void open_db();
 
  void close_db();
 
  void info(ostream &out = cout) override;
 
  void register_event(event_type event);
 
  void log(tuple<string, string> *message = nullptr);
 
  void log(message_type type);
 
  string truncated_message(const string &message);
 
public:
  bool paused = false; // Whether the logger is paused or not
 
private:
  void load_settings() override;
 
  void connect_to_db();
  void open_log_file(string filename, bool array = false);
  void close_log_file();
  void log_to_mongo(tuple<string, string> *message = nullptr);
  void log_doc_to_mongo(const std::string &topic, const nlohmann::json &message);
  void log_blob_to_mongo();
  void log_to_file(tuple<string, string> *message = nullptr);
 
  struct Impl;
  std::unique_ptr<Impl> _impl;
};
 
} // namespace Mads
 
#endif // LOGGER_HPP