123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869 |
- //
- // 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
|