ringbuffer_sink.h 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. // Copyright(c) 2015-present, Gabi Melman & spdlog contributors.
  2. // Distributed under the MIT License (http://opensource.org/licenses/MIT)
  3. #pragma once
  4. #include "spdlog/details/circular_q.h"
  5. #include "spdlog/details/log_msg_buffer.h"
  6. #include "spdlog/details/null_mutex.h"
  7. #include "spdlog/sinks/base_sink.h"
  8. #include <mutex>
  9. #include <string>
  10. #include <vector>
  11. namespace spdlog {
  12. namespace sinks {
  13. /*
  14. * Ring buffer sink
  15. */
  16. template <typename Mutex>
  17. class ringbuffer_sink final : public base_sink<Mutex> {
  18. public:
  19. explicit ringbuffer_sink(size_t n_items)
  20. : q_{n_items} {}
  21. std::vector<details::log_msg_buffer> last_raw(size_t lim = 0) {
  22. std::lock_guard<Mutex> lock(base_sink<Mutex>::mutex_);
  23. auto items_available = q_.size();
  24. auto n_items = lim > 0 ? (std::min)(lim, items_available) : items_available;
  25. std::vector<details::log_msg_buffer> ret;
  26. ret.reserve(n_items);
  27. for (size_t i = (items_available - n_items); i < items_available; i++) {
  28. ret.push_back(q_.at(i));
  29. }
  30. return ret;
  31. }
  32. std::vector<std::string> last_formatted(size_t lim = 0) {
  33. std::lock_guard<Mutex> lock(base_sink<Mutex>::mutex_);
  34. auto items_available = q_.size();
  35. auto n_items = lim > 0 ? (std::min)(lim, items_available) : items_available;
  36. std::vector<std::string> ret;
  37. ret.reserve(n_items);
  38. for (size_t i = (items_available - n_items); i < items_available; i++) {
  39. memory_buf_t formatted;
  40. base_sink<Mutex>::formatter_->format(q_.at(i), formatted);
  41. ret.push_back(SPDLOG_BUF_TO_STRING(formatted));
  42. }
  43. return ret;
  44. }
  45. protected:
  46. void sink_it_(const details::log_msg &msg) override {
  47. q_.push_back(details::log_msg_buffer{msg});
  48. }
  49. void flush_() override {}
  50. private:
  51. details::circular_q<details::log_msg_buffer> q_;
  52. };
  53. using ringbuffer_sink_mt = ringbuffer_sink<std::mutex>;
  54. using ringbuffer_sink_st = ringbuffer_sink<details::null_mutex>;
  55. } // namespace sinks
  56. } // namespace spdlog