1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#include "logger.h"
#include <iostream>
#include <fstream>
#include <chrono>
#include <iomanip>
#include <ctime>
// Global logger instance
Logger g_logger;
Logger::Logger()
: current_level(LogLevel::INFO)
, console_enabled(true)
, file_enabled(false) {
}
Logger::~Logger() {
if (file_stream) {
file_stream->flush();
}
}
void Logger::set_level(LogLevel level) {
current_level = level;
}
void Logger::debug(const std::string& message) {
log(LogLevel::DEBUG, message);
}
void Logger::info(const std::string& message) {
log(LogLevel::INFO, message);
}
void Logger::warning(const std::string& message) {
log(LogLevel::WARNING, message);
}
void Logger::error(const std::string& message) {
log(LogLevel::ERROR, message);
}
void Logger::fatal(const std::string& message) {
log(LogLevel::FATAL, message);
}
void Logger::log(LogLevel level, const std::string& message) {
if (level >= current_level) {
write_log(level, message);
}
}
void Logger::set_output_file(const std::string& filename) {
output_filename = filename;
if (file_enabled && !filename.empty()) {
file_stream = std::make_unique<std::ofstream>(filename, std::ios::app);
if (!file_stream->good()) {
std::cerr << "Failed to open log file: " << filename << std::endl;
file_stream.reset();
}
}
}
void Logger::enable_console(bool enable) {
console_enabled = enable;
}
void Logger::enable_file(bool enable) {
file_enabled = enable;
if (enable && !output_filename.empty()) {
set_output_file(output_filename);
} else if (!enable) {
file_stream.reset();
}
}
std::string Logger::level_to_string(LogLevel level) {
switch (level) {
case LogLevel::DEBUG: return "DEBUG";
case LogLevel::INFO: return "INFO";
case LogLevel::WARNING: return "WARNING";
case LogLevel::ERROR: return "ERROR";
case LogLevel::FATAL: return "FATAL";
default: return "UNKNOWN";
}
}
std::string Logger::get_timestamp() {
auto now = std::chrono::system_clock::now();
auto time_t = std::chrono::system_clock::to_time_t(now);
auto ms = std::chrono::duration_cast<std::chrono::milliseconds>(
now.time_since_epoch()) % 1000;
std::stringstream ss;
ss << std::put_time(std::localtime(&time_t), "%Y-%m-%d %H:%M:%S");
ss << '.' << std::setfill('0') << std::setw(3) << ms.count();
return ss.str();
}
void Logger::write_log(LogLevel level, const std::string& message) {
std::string timestamp = get_timestamp();
std::string level_str = level_to_string(level);
std::string log_entry = "[" + timestamp + "] [" + level_str + "] " + message + "\n";
if (console_enabled) {
if (level == LogLevel::ERROR || level == LogLevel::FATAL) {
std::cerr << log_entry;
} else {
std::cout << log_entry;
}
}
if (file_enabled && file_stream && file_stream->good()) {
*file_stream << log_entry;
file_stream->flush();
}
}
// LogStream implementation
LogStream::LogStream(Logger& logger, LogLevel level)
: logger(logger), level(level) {
}
LogStream::~LogStream() {
logger.log(level, stream.str());
}
|
