This commit is contained in:
gabime
2018-07-26 21:09:40 +03:00
parent ef5e7af68a
commit 420b17ae65
2 changed files with 24 additions and 20 deletions

View File

@@ -68,24 +68,14 @@ public:
void set_tp(std::shared_ptr<thread_pool> tp)
{
std::lock_guard<std::mutex> lock(tp_mutex_);
std::lock_guard<std::recursive_mutex> lock(tp_mutex_);
tp_ = std::move(tp);
}
// create tp only if not exists already
std::shared_ptr<thread_pool> create_tp_once(size_t queue_size, size_t n_threads)
{
std::lock_guard<std::mutex> lock(tp_mutex_);
if (tp_ == nullptr)
{
tp_ = std::make_shared<details::thread_pool>(queue_size, n_threads);
}
return tp_;
}
std::shared_ptr<thread_pool> get_tp()
{
std::lock_guard<std::mutex> lock(tp_mutex_);
std::lock_guard<std::recursive_mutex> lock(tp_mutex_);
return tp_;
}
@@ -178,16 +168,22 @@ public:
drop_all();
{
std::lock_guard<std::mutex> lock(tp_mutex_);
std::lock_guard<std::recursive_mutex> lock(tp_mutex_);
tp_.reset();
}
}
std::recursive_mutex &tp_mutex()
{
return tp_mutex_;
}
static registry &instance()
{
static registry s_instance;
return s_instance;
}
private:
registry()
@@ -210,7 +206,7 @@ private:
}
std::mutex logger_map_mutex_, flusher_mutex_;
std::mutex tp_mutex_;
std::recursive_mutex tp_mutex_;
std::unordered_map<std::string, std::shared_ptr<logger>> loggers_;
std::unique_ptr<formatter> formatter_;
level::level_enum level_ = level::info;