log_queue.h

Go to the documentation of this file.

#ifndef SNIFF_LOG_QUEUE
#define SNIFF_LOG_QUEUE
 
#include "proto/service.pb.h"
#include <condition_variable>
#include <mutex>
#include <vector>
 
namespace yarilo {
 
class LogQueue {
public:
  LogQueue(uint64_t queueMaxSize) : queueMaxSize(queueMaxSize), stopped(false) {
    queue.reserve(queueMaxSize);
  }
 
  bool insert(proto::LogEntry *item) {
    {
      std::unique_lock<std::mutex> lock(queueMutex);
      if (stopped)
        return false;
 
      if (queue.size() >= queueMaxSize)
        queue.back() = item;
      else
        queue.emplace_back(item);
    }
 
    return true;
  }
 
  bool fetch_all(std::vector<proto::LogEntry *> &refFetchedItems) {
    {
      std::unique_lock<std::mutex> lock(queueMutex);
      if (stopped)
        return false;
 
      refFetchedItems = std::move(queue);
      queue.clear();
    }
 
    cvInsert.notify_all();
    return true;
  }
 
  void stop() {
    {
      std::unique_lock<std::mutex> lock(queueMutex);
      stopped = true;
    }
 
    cvInsert.notify_all();
  }
 
  bool is_stopped() { return stopped; }
 
private:
  std::vector<proto::LogEntry *> queue;
  uint64_t queueMaxSize;
  std::mutex queueMutex;
  std::condition_variable cvInsert;
  bool stopped;
};
 
} // namespace yarilo
 
#endif // SNIFF_LOG_QUEUE