64 lines
2.2 KiB
C
64 lines
2.2 KiB
C
|
#ifndef HUMANUSlogger_H
|
|||
|
|
#define HUMANUSlogger_H
|
|||
|
|
|
|||
|
|
#include "spdlog/spdlog.h"
|
|||
|
|
#include "spdlog/sinks/stdout_color_sinks.h"
|
|||
|
|
#include "spdlog/sinks/basic_file_sink.h"
|
|||
|
|
#include "spdlog/sinks/rotating_file_sink.h"
|
|||
|
|
#include "spdlog/sinks/daily_file_sink.h"
|
|||
|
|
#include "spdlog/sinks/dist_sink.h"
|
|||
|
|
|
|||
|
|
#include <string>
|
|||
|
|
|
|||
|
|
namespace humanus {
|
|||
|
|
|
|||
|
|
static spdlog::level::level_enum _print_level = spdlog::level::info;
|
|||
|
|
|
|||
|
|
static std::shared_ptr<spdlog::logger> logger = spdlog::default_logger();
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 调整日志级别
|
|||
|
|
* @param print_level 控制台输出日志级别
|
|||
|
|
* @param logfile_level 文件记录日志级别
|
|||
|
|
* @param name 日志文件名前缀
|
|||
|
|
* @return 日志记录器实例
|
|||
|
|
*/
|
|||
|
|
std::shared_ptr<spdlog::logger> define_log_level(spdlog::level::level_enum print_level = spdlog::level::info,
|
|||
|
|
spdlog::level::level_enum logfile_level = spdlog::level::debug,
|
|||
|
|
std::string name = "") {
|
|||
|
|
_print_level = print_level;
|
|||
|
|
|
|||
|
|
auto current_date = std::chrono::system_clock::now();
|
|||
|
|
auto in_time_t = std::chrono::system_clock::to_time_t(current_date);
|
|||
|
|
|
|||
|
|
std::stringstream ss;
|
|||
|
|
std::tm tm_info = *std::localtime(&in_time_t);
|
|||
|
|
ss << std::put_time(&tm_info, "%Y%m%d");
|
|||
|
|
std::string formatted_date = ss.str(); // YYYYMMDD
|
|||
|
|
|
|||
|
|
std::string log_name = name.empty() ? formatted_date : name + "_" + formatted_date;
|
|||
|
|
std::string log_file_path = (PROJECT_ROOT / "logs" / (log_name + ".log")).string();
|
|||
|
|
|
|||
|
|
// 确保日志目录存在
|
|||
|
|
std::filesystem::create_directories((PROJECT_ROOT / "logs").string());
|
|||
|
|
|
|||
|
|
// 重置日志输出
|
|||
|
|
// 清除现有的sinks
|
|||
|
|
logger->sinks().clear();
|
|||
|
|
|
|||
|
|
// 添加标准错误输出sink，相当于Python中的sys.stderr
|
|||
|
|
auto stderr_sink = std::make_shared<spdlog::sinks::stderr_color_sink_mt>();
|
|||
|
|
stderr_sink->set_level(print_level);
|
|||
|
|
logger->sinks().push_back(stderr_sink);
|
|||
|
|
|
|||
|
|
// 添加文件sink，相当于Python中的PROJECT_ROOT / f"logs/{log_name}.log"
|
|||
|
|
auto file_sink = std::make_shared<spdlog::sinks::basic_file_sink_mt>(log_file_path, true);
|
|||
|
|
file_sink->set_level(logfile_level);
|
|||
|
|
logger->sinks().push_back(file_sink);
|
|||
|
|
|
|||
|
|
return logger;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
} // namespace humanus
|
|||
|
|
|
|||
|
|
#endif
|