mirror of
https://github.com/gabime/spdlog.git
synced 2024-12-25 10:01:33 +08:00
Fixed #1680
This commit is contained in:
parent
05d5546eb1
commit
c7613f3e91
@ -1,52 +0,0 @@
|
||||
// Copyright(c) 2015-present, Gabi Melman & spdlog contributors.
|
||||
// Distributed under the MIT License (http://opensource.org/licenses/MIT)
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <spdlog/logger.h>
|
||||
#include <string>
|
||||
#include <unordered_map>
|
||||
|
||||
namespace spdlog {
|
||||
namespace cfg {
|
||||
class log_levels
|
||||
{
|
||||
std::unordered_map<std::string, spdlog::level::level_enum> levels_;
|
||||
spdlog::level::level_enum global_level_ = level::info;
|
||||
|
||||
public:
|
||||
void set(const std::string &logger_name, level::level_enum lvl)
|
||||
{
|
||||
levels_[logger_name] = lvl;
|
||||
}
|
||||
|
||||
void set_global_level(level::level_enum lvl)
|
||||
{
|
||||
global_level_ = lvl;
|
||||
}
|
||||
|
||||
level::level_enum default_level()
|
||||
{
|
||||
return global_level_;
|
||||
}
|
||||
|
||||
// update log level of given logger if it appears in the config list or if default level is set
|
||||
// return true if updated
|
||||
bool update_logger_level(spdlog::logger &logger)
|
||||
{
|
||||
// if logger was not configured, check if global log level was configured
|
||||
auto it = levels_.find(logger.name());
|
||||
if (it == levels_.end())
|
||||
{
|
||||
it = levels_.find("*");
|
||||
}
|
||||
if (it != levels_.end())
|
||||
{
|
||||
logger.set_level(it->second);
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
};
|
||||
} // namespace cfg
|
||||
} // namespace spdlog
|
@ -57,6 +57,27 @@ SPDLOG_INLINE void registry::register_logger(std::shared_ptr<logger> new_logger)
|
||||
register_logger_(std::move(new_logger));
|
||||
}
|
||||
|
||||
// set level if this logger was configured if the cfg_levels_
|
||||
// return true if found and set
|
||||
SPDLOG_INLINE bool registry::set_level_from_cfg_(logger *logger)
|
||||
{
|
||||
if (cfg_levels_.empty())
|
||||
{
|
||||
return false;
|
||||
}
|
||||
auto cfg_level_it = cfg_levels_.find(logger->name());
|
||||
if (cfg_level_it == cfg_levels_.end())
|
||||
{
|
||||
// if logger name not found, set it anyway if "*" exists (i.e. all loggers)
|
||||
cfg_level_it = cfg_levels_.find(("*"));
|
||||
}
|
||||
if (cfg_level_it != cfg_levels_.end())
|
||||
{
|
||||
logger->set_level(cfg_level_it->second);
|
||||
}
|
||||
return cfg_level_it != cfg_levels_.end();
|
||||
}
|
||||
|
||||
SPDLOG_INLINE void registry::initialize_logger(std::shared_ptr<logger> new_logger)
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(logger_map_mutex_);
|
||||
@ -66,17 +87,8 @@ SPDLOG_INLINE void registry::initialize_logger(std::shared_ptr<logger> new_logge
|
||||
{
|
||||
new_logger->set_error_handler(err_handler_);
|
||||
}
|
||||
// check if log level for this logger name was already configured
|
||||
auto cfg_level_it = cfg_levels_.find(new_logger->name());
|
||||
if (cfg_level_it == cfg_levels_.end())
|
||||
{
|
||||
cfg_level_it = cfg_levels_.find(("*"));
|
||||
}
|
||||
if (cfg_level_it != cfg_levels_.end())
|
||||
{
|
||||
new_logger->set_level(cfg_level_it->second);
|
||||
}
|
||||
else
|
||||
|
||||
if (!set_level_from_cfg_(new_logger.get()))
|
||||
{
|
||||
new_logger->set_level(global_log_level_);
|
||||
}
|
||||
@ -281,20 +293,9 @@ SPDLOG_INLINE void registry::set_levels(std::unordered_map<std::string, spdlog::
|
||||
std::lock_guard<std::mutex> lock(logger_map_mutex_);
|
||||
cfg_levels_ = std::move(levels);
|
||||
|
||||
// for each logger update level according to given map
|
||||
// "*" entry means all loggers
|
||||
auto global_level_it = cfg_levels_.find("*");
|
||||
for (auto &logger : loggers_)
|
||||
{
|
||||
auto cfg_entry = cfg_levels_.find(logger.first);
|
||||
if (cfg_entry != cfg_levels_.end())
|
||||
{
|
||||
logger.second->set_level(cfg_entry->second);
|
||||
}
|
||||
else if (global_level_it != cfg_levels_.end())
|
||||
{
|
||||
logger.second->set_level(global_level_it->second);
|
||||
}
|
||||
set_level_from_cfg_(logger.second.get());
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -89,6 +89,7 @@ private:
|
||||
|
||||
void throw_if_exists_(const std::string &logger_name);
|
||||
void register_logger_(std::shared_ptr<logger> new_logger);
|
||||
bool set_level_from_cfg_(logger *logger);
|
||||
std::mutex logger_map_mutex_, flusher_mutex_;
|
||||
std::recursive_mutex tp_mutex_;
|
||||
std::unordered_map<std::string, std::shared_ptr<logger>> loggers_;
|
||||
|
Loading…
Reference in New Issue
Block a user