Commit c7613f3e authored by gabime's avatar gabime

Fixed #1680

parent 05d5546e
// 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,26 +57,38 @@ SPDLOG_INLINE void registry::register_logger(std::shared_ptr<logger> new_logger) ...@@ -57,26 +57,38 @@ SPDLOG_INLINE void registry::register_logger(std::shared_ptr<logger> new_logger)
register_logger_(std::move(new_logger)); register_logger_(std::move(new_logger));
} }
SPDLOG_INLINE void registry::initialize_logger(std::shared_ptr<logger> 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)
{ {
std::lock_guard<std::mutex> lock(logger_map_mutex_); if (cfg_levels_.empty())
new_logger->set_formatter(formatter_->clone());
if (err_handler_)
{ {
new_logger->set_error_handler(err_handler_); return false;
} }
// check if log level for this logger name was already configured auto cfg_level_it = cfg_levels_.find(logger->name());
auto cfg_level_it = cfg_levels_.find(new_logger->name());
if (cfg_level_it == cfg_levels_.end()) 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(("*")); cfg_level_it = cfg_levels_.find(("*"));
} }
if (cfg_level_it != cfg_levels_.end()) if (cfg_level_it != cfg_levels_.end())
{ {
new_logger->set_level(cfg_level_it->second); 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_);
new_logger->set_formatter(formatter_->clone());
if (err_handler_)
{
new_logger->set_error_handler(err_handler_);
} }
else
if (!set_level_from_cfg_(new_logger.get()))
{ {
new_logger->set_level(global_log_level_); new_logger->set_level(global_log_level_);
} }
...@@ -281,20 +293,9 @@ SPDLOG_INLINE void registry::set_levels(std::unordered_map<std::string, spdlog:: ...@@ -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_); std::lock_guard<std::mutex> lock(logger_map_mutex_);
cfg_levels_ = std::move(levels); 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_) for (auto &logger : loggers_)
{ {
auto cfg_entry = cfg_levels_.find(logger.first); set_level_from_cfg_(logger.second.get());
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);
}
} }
} }
......
...@@ -89,6 +89,7 @@ private: ...@@ -89,6 +89,7 @@ private:
void throw_if_exists_(const std::string &logger_name); void throw_if_exists_(const std::string &logger_name);
void register_logger_(std::shared_ptr<logger> new_logger); void register_logger_(std::shared_ptr<logger> new_logger);
bool set_level_from_cfg_(logger *logger);
std::mutex logger_map_mutex_, flusher_mutex_; std::mutex logger_map_mutex_, flusher_mutex_;
std::recursive_mutex tp_mutex_; std::recursive_mutex tp_mutex_;
std::unordered_map<std::string, std::shared_ptr<logger>> loggers_; std::unordered_map<std::string, std::shared_ptr<logger>> loggers_;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment