Blame view

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