feat(probe): probe logging module first implimentation

This commit is contained in:
2025-02-21 10:30:59 -05:00
parent 31bc77159b
commit 95f943b45a
3 changed files with 198 additions and 0 deletions

22
src/probe/meson.build Normal file
View File

@@ -0,0 +1,22 @@
# Define the library
logger_sources = files(
'private/logger.cpp',
)
logger_headers = files(
'public/logger.h'
)
# Define the liblogger library so it can be linked against by other parts of the build system
liblogger = library('logger',
logger_sources,
include_directories: include_directories('public'),
cpp_args: ['-fvisibility=default'],
install : true)
logger_dep = declare_dependency(
include_directories: include_directories('public'),
link_with: liblogger,
)
# Make headers accessible
install_headers(logger_headers, subdir : '4DSSE/logger')

View File

@@ -0,0 +1,104 @@
#include <iostream>
#include <string>
#include <fstream>
#include <sstream>
#include <queue>
#include <mutex>
#include <condition_variable>
#include <chrono>
#include <thread>
#include <atomic>
#include <unordered_map>
#include <iomanip>
#include "logger.h"
Record makeRecord(const std::string& message, const std::string& context, DiagnosticLevel level) {
return {message, context, std::chrono::system_clock::now(), level};
}
std::string timestepToString(const std::chrono::time_point<std::chrono::system_clock>& timestamp) {
std::time_t timeT = std::chrono::system_clock::to_time_t(timestamp);
std::tm tmStruct = *std::localtime(&timeT);
std::ostringstream oss;
oss << std::put_time(&tmStruct, "%Y-%m-%d %H:%M:%S");
return oss.str();
}
std::unordered_map<std::string, Logger*> Logger::instances;
std::mutex Logger::mapMutex;
Logger::Logger(const std::string& filename) : filename(filename), logFile(filename, std::ios::app) {
logThread = std::thread(&Logger::processesLogs, this);
}
void Logger::processesLogs() {
while (running || !logQueue.empty()) {
std::unique_lock<std::mutex> lock(queueMutex);
queueCondition.wait(lock, [this] { return !logQueue.empty() || !running; });
while (!logQueue.empty()) {
logFile << logQueue.front() << std::endl;
logQueue.pop();
}
}
}
std::string Logger::levelAsString(DiagnosticLevel level) {
switch (level) {
case DiagnosticLevel::DEBUG:
return "DEBUG";
case DiagnosticLevel::INFO:
return "INFO";
case DiagnosticLevel::WARNING:
return "WARNING";
case DiagnosticLevel::ERROR:
return "ERROR";
case DiagnosticLevel::CRITICAL:
return "CRITICAL";
case DiagnosticLevel::NONE:
return "NONE";
}
return "";
}
Logger& Logger::getInstance(const std::string& filename) {
std::lock_guard<std::mutex> lock(mapMutex);
auto it = instances.find(filename);
if (it == instances.end()) {
it = instances.emplace(filename, new Logger(filename)).first;
}
return *it->second;
}
void Logger::log(const Record& record) {
std::stringstream ss;
ss << levelAsString(record.level) << " @ " << timestepToString(record.timestamp) << " [" << record.context << "] :" << record.message;
std::lock_guard<std::mutex> lock(queueMutex);
logQueue.push(ss.str());
queueCondition.notify_one();
}
void Logger::log(const std::string& message, const std::string& context, DiagnosticLevel level) {
log(makeRecord(message, context, level));
}
void Logger::log(const std::string& message, DiagnosticLevel level) {
log(makeRecord(message, "", level));
}
void Logger::log(const std::string& message) {
log(makeRecord(message, "", DiagnosticLevel::INFO));
}
Logger::~Logger() {
{
std::lock_guard<std::mutex> lock(queueMutex);
running = false;
}
queueCondition.notify_one();
if (logThread.joinable()) {
logThread.join();
}
std::this_thread::sleep_for(std::chrono::milliseconds(100)); // wait for the log thread to finish writing
}

72
src/probe/public/logger.h Normal file
View File

@@ -0,0 +1,72 @@
#ifndef LOGGER_H
#define LOGGER_H
#include <iostream>
#include <string>
#include <fstream>
#include <sstream>
#include <queue>
#include <mutex>
#include <condition_variable>
#include <chrono>
#include <thread>
#include <atomic>
#include <unordered_map>
enum class DiagnosticLevel {
DEBUG,
INFO,
WARNING,
ERROR,
CRITICAL,
NONE
};
struct Record {
std::string message;
std::string context;
std::chrono::time_point<std::chrono::system_clock> timestamp;
DiagnosticLevel level;
};
Record makeRecord(const std::string& message, const std::string& context, DiagnosticLevel level);
std::string timestepToString(const std::chrono::time_point<std::chrono::system_clock>& timestamp);
class Logger {
private:
static std::unordered_map<std::string, Logger*> instances;
static std::mutex mapMutex;
std::queue<std::string> logQueue;
std::mutex queueMutex;
std::condition_variable queueCondition;
std::thread logThread;
std::atomic<bool> running;
std::string filename;
std::ofstream logFile;
Logger(const std::string& filename);
Logger(const Logger&) = delete;
Logger& operator=(const Logger&) = delete;
void processesLogs();
std::string levelAsString(DiagnosticLevel level);
public:
static Logger& getInstance(const std::string& filename);
void log(const Record& record);
void log(const std::string& message, const std::string& context, DiagnosticLevel level);
void log(const std::string& message, DiagnosticLevel level);
void log(const std::string& message);
~Logger();
};
#endif