LoggerBase.h

#pragma once

#include "quill/core/Attributes.h"
#include "quill/core/Common.h"
#include "quill/core/LogLevel.h"
#include "quill/core/PatternFormatterOptions.h"
#include "quill/core/QuillError.h"
#include "quill/core/ThreadContextManager.h"

#include <atomic>
#include <cassert>
#include <memory>
#include <string>
#include <vector>

QUILL_BEGIN_NAMESPACE

class Sink;
class PatternFormatter;
class UserClockSource;

namespace detail
{
class BackendWorker;
class BacktraceStorage;
class LoggerManager;

/***/
class LoggerBase
{
public:
  /***/
  LoggerBase(std::string logger_name, std::vector<std::shared_ptr<Sink>> sinks,
             PatternFormatterOptions pattern_formatter_options, ClockSourceType clock_source,
             UserClockSource* user_clock)
    : _logger_name(static_cast<std::string&&>(logger_name)),
      _user_clock(user_clock),
      _clock_source(clock_source),
      _pattern_formatter_options(static_cast<PatternFormatterOptions&&>(pattern_formatter_options))
  {
#ifndef NDEBUG
    for (auto const& sink : sinks)
    {
      assert(sink && "sink pointer is nullptr");
    }
#endif

    this->_sinks = static_cast<std::vector<std::shared_ptr<Sink>>&&>(sinks);
  }

  /***/
  LoggerBase(LoggerBase const&) = delete;
  LoggerBase& operator=(LoggerBase const&) = delete;

  /***/
  virtual ~LoggerBase() = default;

  QUILL_NODISCARD std::string const& get_logger_name() const noexcept { return _logger_name; }

  QUILL_NODISCARD UserClockSource* get_user_clock_source() const noexcept { return _user_clock; }

  QUILL_NODISCARD ClockSourceType get_clock_source_type() const noexcept { return _clock_source; }

  QUILL_NODISCARD PatternFormatterOptions const& get_pattern_formatter_options() const noexcept
  {
    return _pattern_formatter_options;
  }

  QUILL_NODISCARD std::vector<std::shared_ptr<Sink>> const& get_sinks() const noexcept
  {
    return _sinks;
  }

  void mark_invalid() { _valid.store(false, std::memory_order_release); }

  QUILL_NODISCARD bool is_valid_logger() const noexcept
  {
    return _valid.load(std::memory_order_acquire);
  }

  QUILL_NODISCARD LogLevel get_log_level() const noexcept
  {
    return _log_level.load(std::memory_order_relaxed);
  }

  void set_log_level(LogLevel new_log_level)
  {
    if (QUILL_UNLIKELY(new_log_level == LogLevel::Backtrace))
    {
      QUILL_THROW(QuillError{"LogLevel::Backtrace is only used internally. Please don't use it."});
    }

    _log_level.store(new_log_level, std::memory_order_relaxed);
  }

  void set_immediate_flush(uint32_t flush_every_n_messages = 1)
  {
    _message_flush_threshold.store(flush_every_n_messages, std::memory_order_relaxed);
  }

  template <LogLevel log_statement_level>
  QUILL_NODISCARD QUILL_ATTRIBUTE_HOT bool should_log_statement() const noexcept
  {
    return log_statement_level >= get_log_level();
  }

  QUILL_NODISCARD QUILL_ATTRIBUTE_HOT bool should_log_statement(LogLevel log_statement_level) const noexcept
  {
    return log_statement_level >= get_log_level();
  }

protected:
  friend class BackendWorker;
  friend class LoggerManager;

  static inline QUILL_THREAD_LOCAL ThreadContext* _thread_context = nullptr; /* Set and accessed by the frontend */

  // -- frontend access BEGIN --
  std::string _logger_name; /* Set by the frontend once, accessed by the frontend AND backend */
  UserClockSource* _user_clock{nullptr}; /* A non-owned pointer to a custom timestamp clock, valid only when provided. used by frontend only */
  std::atomic<uint32_t> _message_flush_threshold{0};   /* used by frontend only */
  std::atomic<uint32_t> _messages_since_last_flush{0}; /* used by frontend only */
  ClockSourceType _clock_source; /* Set by the frontend and accessed by the frontend AND backend */
  std::atomic<LogLevel> _log_level{LogLevel::Info}; /* used by frontend only */
  std::atomic<LogLevel> _backtrace_flush_level{LogLevel::None}; 
  std::atomic<bool> _valid{true}; /* Updated by the frontend at any time, accessed by the backend */
  // -- frontend access END --

  PatternFormatterOptions _pattern_formatter_options; /* Set by the frontend once and accessed by the backend to initialise PatternFormatter */
  std::vector<std::shared_ptr<Sink>> _sinks; /* Set by the frontend once and accessed by the backend */
  std::shared_ptr<PatternFormatter> _pattern_formatter; /* The backend thread will set this once, we never access it on the frontend */
  std::shared_ptr<BacktraceStorage> _backtrace_storage; /* The backend thread will construct this, we never access it on the frontend */
};
} // namespace detail

QUILL_END_NAMESPACE