SecurePlay/SecurePlayAuxServer/common/LHLog/LHLogInit.cpp

#include "LHLogInit.h"

#include "spdlog/spdlog.h"
#include "spdlog/sinks/stdout_color_sinks.h"
#include "LHLog_file_sink.h"


#include <string>
#include <filesystem>


/* 初始化spdlog，输入的是模组名称，可以为空
 * lhQLog是一个全局函数，在库LHQLog中定义
 * 这里获取可执行文件文件夹的路径，使用的是C++17新引入的特性
 * GCC9会在标准库中，GCC8.3需要手动链接库stdc++fs才可以使用
 */

void initLog(QString ModuleName, CLHQLogApi& lhQLog)
{

    try {
        /* 创建一个控制台sink */
        auto sink_consolse = std::make_shared<spdlog::sinks::stdout_color_sink_mt>();
        /* 创建一个文件sink，每天一个，00:00创建新的 */
        // auto sink_file = std::make_shared<spdlog::sinks::daily_file_sink_mt>("FlowChartLog/log.txt",0,0);
        /* 自定义的sink */
        std::filesystem::path execPath = std::filesystem::current_path();
    #if defined(Q_OS_WIN32)
        #if C_DEBUG
            std::string libName = execPath.string() + "/LHQLogd.dll";
        #elif C_RELEASE
            std::string libName = execPath.string() + "/LHQLog.dll";
        #endif
    #elif defined(Q_OS_LINUX)
        std::string libName = execPath.string() + "/libLHQLog.so";
    #endif
        /* 写入文件的sink */
        auto sink_custom = std::make_shared<spdlog::sinks::LHLog_file_sink_mt>(&lhQLog, QString(libName.c_str()), ModuleName);
        /* 给默认记录器用的sink */
        auto sink_default = std::make_shared<spdlog::sinks::stdout_color_sink_mt>();

        /* 修改输出格式 */
#if  defined (C_DEBUG)
        sink_consolse->set_pattern("%^[%Y-%m-%d %H:%M:%S:%e] [%l] [%n] [%s %#] %v%$");
        // sink_file->set_pattern("[%Y-%m-%d %H:%M:%S] [%^%n%$] [%^%l%$] %s %#: %v");
        sink_custom->set_pattern("%v");
        sink_default->set_pattern("%^[%Y-%m-%d %H:%M:%S:%e] [%l] [%s %#] %v%$");
#elif defined(C_RELEASE)
        sink_consolse->set_pattern("%^[%Y-%m-%d %H:%M:%S:%e] [%l] [%n] %v%$");
        // sink_file->set_pattern("[%Y-%m-%d %H:%M:%S] [%^%n%$] [%^%l%$] %s %#: %v");
        sink_custom->set_pattern("%v");
        sink_default->set_pattern("%^[%Y-%m-%d %H:%M:%S:%e] [%l] %v%$");
#endif

        /* 日志输出方向，终端和文件 */
        std::vector<spdlog::sink_ptr> sinks;
        sinks.push_back(sink_consolse);
        // sinks.push_back(sink_file);
        sinks.push_back(sink_custom);

        /* 设置默认日志记录器，设置成自定义的sink */
        spdlog::default_logger()->sinks().clear();
        spdlog::default_logger()->sinks().push_back(sink_default);
        spdlog::default_logger()->sinks().push_back(sink_custom);
        // auto vecSinks =  spdlog::default_logger()->sinks();
        // for(auto& sink : vecSinks)
        // {
        //     SPDLOG_INFO("sink name = {}", typeid(*sink).name());
        // }
        /* 创建一个标准输出 */
        auto logger_main = std::make_shared<spdlog::logger>("main",begin(sinks),end(sinks));
        /* 创建一个 FromSuperBrain Logger */
        auto logger_FromSuperBrain = std::make_shared<spdlog::logger>("FromSuperBrain",begin(sinks),end(sinks));
        /* 创建一个 SPAServer logger */
        auto logger_SPAServer = std::make_shared<spdlog::logger>("SPAServer",begin(sinks),end(sinks));
        /* 创建一个 FromRedis logger */
        auto logger_FromRedis = std::make_shared<spdlog::logger>("FromRedis",begin(sinks),end(sinks));
        /* 创建一个 FromWebAPI logger */
        auto logger_webapi = std::make_shared<spdlog::logger>("FromWebAPI",begin(sinks),end(sinks));
        /* 创建一个MQTT logger */
        auto logger_MQTT = std::make_shared<spdlog::logger>("MQTT",begin(sinks),end(sinks));


        /* 注册到注册表 */
        spdlog::register_logger(logger_main);
        spdlog::register_logger(logger_FromSuperBrain);
        spdlog::register_logger(logger_SPAServer);
        spdlog::register_logger(logger_FromRedis);
        spdlog::register_logger(logger_webapi);
        spdlog::register_logger(logger_MQTT);

        /* 设置spdlog输出级别，默认的估计不输出debug这个级别
         * 这是默认的设置，可以在外面单数设置输出方式
         * release模式下不输出console的日志输出
         * */
#if  defined (C_DEBUG)
        spdlog::set_level(spdlog::level::trace);
        spdlog::flush_on(spdlog::level::trace);
#elif defined(C_RELEASE)
        spdlog::set_level(spdlog::level::info);             /* 只输出info以上的输出 */
        spdlog::flush_on(spdlog::level::info);              /* 设置刷新等级 */
        // sink_consolse->set_level(spdlog::level::off);       /* 控制台不输出 */
        // sink_default->set_level(spdlog::level::off);        /* 默认sink也不输出 */
#endif

//    SPDLOG_LOGGER_DEBUG(logger_main,"******* 有行号且debug也能输出 *******");
    }
    catch (const spdlog::spdlog_ex& ex) {
        qDebug() << "Log initialization failed: " << ex.what() ;
    }
}


/* 修改日志输出等级，输入1表示 */
void changeLogLevel(ELogLevel level)
{
    if(ELogLevel::Log_Debug == level)
    {
        spdlog::set_level(spdlog::level::debug);
        spdlog::flush_on(spdlog::level::debug);
    }
    else if(ELogLevel::Log_Info == level)
    {
        spdlog::set_level(spdlog::level::info);
        spdlog::flush_on(spdlog::level::info);
        /* 获取写入文件的sink，release模式下不写入文件 */
        auto logger = spdlog::get("main");
        auto sinks = logger->sinks();
        sinks[1]->set_level(spdlog::level::info);
    }
}