80 lines
1.8 KiB
C
80 lines
1.8 KiB
C
|
//
|
||
|
|
// Copyright(c) 2018 Gabi Melman.
|
||
|
|
// Distributed under the MIT License (http://opensource.org/licenses/MIT)
|
||
|
|
//
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "spdlog/details/null_mutex.h"
|
||
|
|
#include "spdlog/sinks/base_sink.h"
|
||
|
|
#include "spdlog/fmt/fmt.h"
|
||
|
|
#include <chrono>
|
||
|
|
#include <mutex>
|
||
|
|
#include <thread>
|
||
|
|
|
||
|
|
namespace spdlog {
|
||
|
|
namespace sinks {
|
||
|
|
|
||
|
|
template<class Mutex>
|
||
|
|
class test_sink : public base_sink<Mutex>
|
||
|
|
{
|
||
|
|
const size_t lines_to_save = 100;
|
||
|
|
|
||
|
|
public:
|
||
|
|
size_t msg_counter()
|
||
|
|
{
|
||
|
|
std::lock_guard<Mutex> lock(base_sink<Mutex>::mutex_);
|
||
|
|
return msg_counter_;
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t flush_counter()
|
||
|
|
{
|
||
|
|
std::lock_guard<Mutex> lock(base_sink<Mutex>::mutex_);
|
||
|
|
return flush_counter_;
|
||
|
|
}
|
||
|
|
|
||
|
|
void set_delay(std::chrono::milliseconds delay)
|
||
|
|
{
|
||
|
|
std::lock_guard<Mutex> lock(base_sink<Mutex>::mutex_);
|
||
|
|
delay_ = delay;
|
||
|
|
}
|
||
|
|
|
||
|
|
// return last output without the eol
|
||
|
|
std::vector<std::string> lines()
|
||
|
|
{
|
||
|
|
std::lock_guard<Mutex> lock(base_sink<Mutex>::mutex_);
|
||
|
|
return lines_;
|
||
|
|
}
|
||
|
|
|
||
|
|
protected:
|
||
|
|
void sink_it_(const details::log_msg &msg) override
|
||
|
|
{
|
||
|
|
memory_buf_t formatted;
|
||
|
|
base_sink<Mutex>::formatter_->format(msg, formatted);
|
||
|
|
// save the line without the eol
|
||
|
|
auto eol_len = strlen(details::os::default_eol);
|
||
|
|
if (lines_.size() < lines_to_save)
|
||
|
|
{
|
||
|
|
lines_.emplace_back(formatted.begin(), formatted.end() - eol_len);
|
||
|
|
}
|
||
|
|
msg_counter_++;
|
||
|
|
std::this_thread::sleep_for(delay_);
|
||
|
|
}
|
||
|
|
|
||
|
|
void flush_() override
|
||
|
|
{
|
||
|
|
flush_counter_++;
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t msg_counter_{0};
|
||
|
|
size_t flush_counter_{0};
|
||
|
|
std::chrono::milliseconds delay_{std::chrono::milliseconds::zero()};
|
||
|
|
std::vector<std::string> lines_;
|
||
|
|
};
|
||
|
|
|
||
|
|
using test_sink_mt = test_sink<std::mutex>;
|
||
|
|
using test_sink_st = test_sink<details::null_mutex>;
|
||
|
|
|
||
|
|
} // namespace sinks
|
||
|
|
} // namespace spdlog
|