118 lines
2.9 KiB
C++
Executable File
118 lines
2.9 KiB
C++
Executable File
#include "XNLogger.h"
|
|
|
|
XNLogger::XNLogger()
|
|
: consoleOutputEnabled{true, true, true, true}, fileOutputEnabled{true, true, true, true}
|
|
{
|
|
// 获取当前工作目录
|
|
std::filesystem::path currentDir = std::filesystem::current_path();
|
|
std::filesystem::path logDirPath = currentDir / "log";
|
|
|
|
// 创建 log 文件夹
|
|
if (!std::filesystem::exists(logDirPath)) {
|
|
std::filesystem::create_directories(logDirPath);
|
|
}
|
|
|
|
// 创建以当前日期和时间命名的日志文件
|
|
auto now = std::chrono::system_clock::now();
|
|
auto time = std::chrono::system_clock::to_time_t(now);
|
|
std::stringstream ss;
|
|
ss << std::put_time(std::localtime(&time), "%Y%m%d_%H%M%S");
|
|
std::string logFileName = "log_" + ss.str() + ".log";
|
|
|
|
logFilePath = (logDirPath / logFileName).string();
|
|
logFile.open(logFilePath, std::ios::app);
|
|
}
|
|
|
|
XNLogger::~XNLogger()
|
|
{
|
|
if (logFile.is_open()) {
|
|
logFile.close();
|
|
}
|
|
}
|
|
|
|
std::string XNLogger::getCurrentTimeString() const
|
|
{
|
|
auto now = std::chrono::system_clock::now();
|
|
auto time = 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), "%Y-%m-%d %H:%M:%S");
|
|
ss << '.' << std::setfill('0') << std::setw(3) << ms.count();
|
|
return ss.str();
|
|
}
|
|
|
|
void XNLogger::log(LogLevel level, const std::string &message)
|
|
{
|
|
std::lock_guard<std::mutex> locker(mutex);
|
|
std::string logMessage =
|
|
"[" + getCurrentTimeString() + "] [" + logLevelToString(level) + "] " + message;
|
|
|
|
std::string coloredMessage;
|
|
|
|
// 根据日志等级设置颜色
|
|
switch (level) {
|
|
case Debug:
|
|
coloredMessage = COLOR_DEBUG + logMessage + COLOR_RESET;
|
|
break;
|
|
case Info:
|
|
coloredMessage = COLOR_INFO + logMessage + COLOR_RESET;
|
|
break;
|
|
case Warning:
|
|
coloredMessage = COLOR_WARNING + logMessage + COLOR_RESET;
|
|
break;
|
|
case Error:
|
|
coloredMessage = COLOR_ERROR + logMessage + COLOR_RESET;
|
|
break;
|
|
default:
|
|
coloredMessage = logMessage; // 默认无颜色
|
|
break;
|
|
}
|
|
|
|
// 控制台输出
|
|
if (consoleOutputEnabled[level]) {
|
|
if (level == Time) {
|
|
// 如果是时间日志，则不换行，回退到该行开始
|
|
std::cout << coloredMessage << "\r" << std::flush;
|
|
} else {
|
|
std::cout << coloredMessage << std::endl;
|
|
}
|
|
}
|
|
|
|
// 文件输出
|
|
if (fileOutputEnabled[level] && logFile.is_open()) {
|
|
if (level == Time) {
|
|
logFile << logMessage << "\r" << std::flush;
|
|
} else {
|
|
logFile << logMessage << std::endl;
|
|
}
|
|
}
|
|
}
|
|
|
|
void XNLogger::enableConsoleOutput(LogLevel level, bool enable)
|
|
{
|
|
consoleOutputEnabled[level] = enable;
|
|
}
|
|
|
|
void XNLogger::enableFileOutput(LogLevel level, bool enable)
|
|
{
|
|
fileOutputEnabled[level] = enable;
|
|
}
|
|
|
|
std::string XNLogger::logLevelToString(LogLevel level) const
|
|
{
|
|
switch (level) {
|
|
case Debug:
|
|
return "DEBUG";
|
|
case Info:
|
|
return "INFO";
|
|
case Warning:
|
|
return "WARNING";
|
|
case Error:
|
|
return "ERROR";
|
|
case Time:
|
|
return "TIME";
|
|
default:
|
|
return "UNKNOWN";
|
|
}
|
|
} |