RMMLog.h

#pragma once
 
#include <sys/stat.h>
 
#include <atomic>
#include <chrono>
#include <cstdarg>
#include <cstring>
#include <fstream>
#include <iostream>
#include <map>
#include <mutex>
#include <thread>
#include <vector>
 
#include "RMMParams.h"
#include "RMMUtils.h"
 
#define MAX_CHAR_LOG_MESSAGE 1024
#define TIMESTAMP_SIZE 32
#define SEPARATORS_SIZE 5
#define MAX_FINAL_MESSAGE_SIZE (MAX_CHAR_LOG_MESSAGE + TIMESTAMP_SIZE + SEPARATORS_SIZE)
 
#define RMMLOG_DEBUG_MASK 1
#define RMMLOG_INFO_MASK 2
#define RMMLOG_WARN_MASK 4
#define RMMLOG_ERROR_MASK 8
#define RMMLOG_CRITICAL_MASK 16

enum class LogMask : uint8_t { INFO, DEBUG, WARN, ERROR, CRITICAL };

struct LogMessage {
  const char *sender;   
  const char *message;  
  int mask;             
};

class RMMLog {
 public:
  RMMLog(const char *name);
  ~RMMLog();
 
  void debug(const char *sender, const char *printer...);
  void info(const char *sender, const char *printer...);
  void warn(const char *sender, const char *printer...);
  void error(const char *sender, const char *printer...);
  void critical(const char *sender, const char *printer...);
  void logRelay(std::string logLevel, FILE *out);
  void getTimestamp(char message[MAX_CHAR_LOG_MESSAGE]);
  void getLogMask(int *mask);
  void getLogMask(char *mask);
  rmmStatus setLogMask(int mask);
  void setLogTimeout(int milliseconds);
  void setFileWriteTimeout(int milliseconds);
  int getFailedWrites();
  rmmStatus logMsg(LogMask mask, const char *sender, const char *message...);
  rmmStatus putMsg(std::string msg);
  rmmStatus putInfo(const char *sender, std::string msg);
  rmmStatus putMsg(const char *message...);
  rmmStatus putInfo(const char *sender, const char *msg...);
  rmmStatus getMsg(std::string *msg);
  rmmStatus getLogFilePath(std::string *filePath);
  rmmStatus getLogFilePath(std::string *path, std::string *name);
  rmmStatus setLogFilePath(std::string filePath, bool useName = false);
  rmmStatus setLogFilePath(std::string path, std::string name);
  rmmStatus setWriteToFile(bool option, char mode = 'w');
 
 private:
  void log(FILE *out, const char *msg);
  // Used to define which log will print which message.
  FILE *debug_log = NULL;
  FILE *info_log = NULL;
  FILE *warn_log = NULL;
  FILE *err_log = NULL;
  FILE *crit_log = NULL;
  std::string file_name;
  std::string name;
 
  std::string message;
  std::atomic_uint failedFileWrites = {0};
  std::atomic_uint failedLogs = {0};
  std::ofstream logFile;
  std::chrono::milliseconds max_log_timeout_ms = std::chrono::milliseconds(10);
  std::chrono::milliseconds max_file_write_timeout_ms = std::chrono::milliseconds(10);
  int log_mask = 28;  // Default is Warn, Error and Critical. Can be changed dynamically.
  std::atomic_bool writeToFile = {false};
};