mirror of
https://github.com/gabime/spdlog.git
synced 2024-11-15 08:25:43 +08:00
Add consolute mutex support to wincolor sink
This commit is contained in:
parent
bfe2116191
commit
3c4b1ceb36
@ -17,7 +17,7 @@ namespace spdlog {
|
||||
namespace sinks {
|
||||
template <typename Mutex>
|
||||
SPDLOG_INLINE wincolor_sink<Mutex>::wincolor_sink(void *out_handle, color_mode mode)
|
||||
: out_handle_(out_handle) {
|
||||
: out_handle_(out_handle), console_mutex_(nullptr) {
|
||||
set_color_mode_impl(mode);
|
||||
// set level colors
|
||||
colors_[level::trace] = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE; // white
|
||||
@ -45,6 +45,12 @@ void SPDLOG_INLINE wincolor_sink<Mutex>::set_color(level::level_enum level,
|
||||
colors_[static_cast<size_t>(level)] = color;
|
||||
}
|
||||
|
||||
template <typename Mutex>
|
||||
void SPDLOG_INLINE wincolor_sink<Mutex>::set_console_mutex(std::mutex* mutex) {
|
||||
std::lock_guard<Mutex> lock(base_t::mutex_);
|
||||
console_mutex_ = mutex;
|
||||
}
|
||||
|
||||
template <typename Mutex>
|
||||
void SPDLOG_INLINE wincolor_sink<Mutex>::sink_it_(const details::log_msg &msg) {
|
||||
if (out_handle_ == nullptr || out_handle_ == INVALID_HANDLE_VALUE) {
|
||||
@ -54,6 +60,10 @@ void SPDLOG_INLINE wincolor_sink<Mutex>::sink_it_(const details::log_msg &msg) {
|
||||
msg.color_range_end = 0;
|
||||
memory_buf_t formatted;
|
||||
base_t::formatter_->format(msg, formatted);
|
||||
|
||||
auto console_lock = console_mutex_ != nullptr ? std::unique_lock<std::mutex>(*console_mutex_)
|
||||
: std::unique_lock<std::mutex>();
|
||||
|
||||
if (should_do_colors_ && msg.color_range_end > msg.color_range_start) {
|
||||
// before color range
|
||||
print_range_(formatted, 0, msg.color_range_start);
|
||||
|
@ -30,11 +30,13 @@ public:
|
||||
// change the color for the given level
|
||||
void set_color(level::level_enum level, std::uint16_t color);
|
||||
void set_color_mode(color_mode mode);
|
||||
void set_console_mutex(std::mutex *console_mutex);
|
||||
|
||||
protected:
|
||||
void *out_handle_;
|
||||
bool should_do_colors_;
|
||||
std::array<std::uint16_t, level::n_levels> colors_;
|
||||
std::mutex *console_mutex_;
|
||||
|
||||
void sink_it_(const details::log_msg &msg) final override;
|
||||
void flush_() final override;
|
||||
|
Loading…
Reference in New Issue
Block a user