logger.hpp 3.33 KB
Newer Older
1 2
#pragma once

3 4 5 6 7 8 9 10 11

/**
 * Singleton used in logger macros to write into files or stdout, with
 * various levels of severity.
 * Only the macros should be used.
 * @class Logger
 */

#include <memory>
12
#include <string>
13 14
#include <iostream>
#include <fstream>
15
#include <sstream>
16 17 18 19 20 21

#define debug_lvl 0
#define info_lvl 1
#define warning_lvl 2
#define error_lvl 3

louiz’'s avatar
louiz’ committed
22
#include "biboumi.h"
23
#ifdef SYSTEMD_FOUND
24
#define SD_JOURNAL_SUPPRESS_LOCATION
25
# include <systemd/sd-daemon.h>
26
# include <systemd/sd-journal.h>
27 28 29 30 31
#else
# define SD_DEBUG    "[DEBUG]: "
# define SD_INFO     "[INFO]: "
# define SD_WARNING  "[WARNING]: "
# define SD_ERR      "[ERROR]: "
32 33 34 35
# define LOG_ERR 3
# define LOG_WARNING 4
# define LOG_INFO 6
# define LOG_DEBUG 7
36 37 38 39 40 41 42 43
#endif

// Macro defined to get the filename instead of the full path. But if it is
// not properly defined by the build system, we fallback to __FILE__
#ifndef __FILENAME__
# define __FILENAME__ __FILE__
#endif

44

45
/**
46 47
 * A buffer, used to construct an ostream that does nothing
 * when we output data in it
48
 */
49
class NullBuffer: public std::streambuf
50
{
51 52
 public:
  int overflow(int c) { return c; }
53 54 55 56 57 58 59 60 61 62
};

class Logger
{
public:
  static std::unique_ptr<Logger>& instance();
  std::ostream& get_stream(const int);
  Logger(const int log_level, const std::string& log_file);
  Logger(const int log_level);

louiz’'s avatar
louiz’ committed
63 64 65 66
  Logger(const Logger&) = delete;
  Logger& operator=(const Logger&) = delete;
  Logger(Logger&&) = delete;
  Logger& operator=(Logger&&) = delete;
67

68 69 70 71 72 73 74 75 76
#ifdef SYSTEMD_FOUND
  bool use_stdout() const
  {
    return this->stream.rdbuf() == std::cout.rdbuf();
  }

  bool use_systemd{false};
#endif

louiz’'s avatar
louiz’ committed
77
private:
78
  const int log_level;
79
  std::ofstream ofstream{};
80
  std::ostream stream;
81 82 83

  NullBuffer null_buffer;
  std::ostream null_stream;
84 85
};

86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101
namespace logging_details
{
  template <typename T>
  void log(std::ostream& os, const T& arg)
  {
    os << arg << std::endl;
  }

  template <typename T, typename... U>
  void log(std::ostream& os, const T& first, U&&... rest)
  {
    os << first;
    log(os, std::forward<U>(rest)...);
  }

  template <typename... U>
102
  void do_logging(const int level, int syslog_level, const char* src_file, int line, U&&... args)
103
  {
104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
  #ifdef SYSTEMD_FOUND
    if (Logger::instance()->use_systemd)
      {
        (void)level;
        std::ostringstream os;
        log(os, std::forward<U>(args)...);
        sd_journal_send("MESSAGE=%s", os.str().data(),
                        "PRIORITY=%i", syslog_level,
                        "CODE_FILE=%s", src_file,
                        "CODE_LINE=%i", line,
                        nullptr);
      }
    else
      {
  #endif
119
        (void)syslog_level;
120 121 122 123 124 125 126
        static const char* priority_names[] = {"DEBUG", "INFO", "WARNING", "ERROR"};
        auto& os = Logger::instance()->get_stream(level);
        os << '[' << priority_names[level] << "]: " << src_file << ':' << line << ":\t";
        log(os, std::forward<U>(args)...);
#ifdef SYSTEMD_FOUND
      }
#endif
127 128 129
  }
}

130
#define log_debug(...) logging_details::do_logging(debug_lvl, LOG_DEBUG, __FILENAME__, __LINE__, __VA_ARGS__)
131

132
#define log_info(...) logging_details::do_logging(info_lvl, LOG_INFO, __FILENAME__, __LINE__, __VA_ARGS__)
133

134
#define log_warning(...) logging_details::do_logging(warning_lvl, LOG_WARNING, __FILENAME__, __LINE__, __VA_ARGS__)
135

136
#define log_error(...) logging_details::do_logging(error_lvl, LOG_ERR, __FILENAME__, __LINE__, __VA_ARGS__)