blob: 3a5ab3c081c457938797139f000ed54b809479f2 (
plain) (
blame)
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
|
#ifndef BUFFER_QUEUE_HPP
#define BUFFER_QUEUE_HPP
#include "QueueItem.hpp"
#include "concurrentqueue.h"
#include <atomic>
#include <vector>
#include <memory>
#include <condition_variable>
#include <chrono>
class BufferQueue
{
public:
using ProducerToken = moodycamel::ProducerToken;
using ConsumerToken = moodycamel::ConsumerToken;
private:
moodycamel::ConcurrentQueue<QueueItem> m_queue;
public:
explicit BufferQueue(size_t capacity, size_t maxExplicitProducers);
ProducerToken createProducerToken() { return ProducerToken(m_queue); }
ConsumerToken createConsumerToken() { return ConsumerToken(m_queue); }
bool enqueueBlocking(QueueItem item,
ProducerToken &token,
std::chrono::milliseconds timeout = std::chrono::milliseconds::max());
bool enqueueBatchBlocking(std::vector<QueueItem> items,
ProducerToken &token,
std::chrono::milliseconds timeout = std::chrono::milliseconds::max());
bool tryDequeue(QueueItem &item, ConsumerToken &token);
size_t tryDequeueBatch(std::vector<QueueItem> &items, size_t maxItems, ConsumerToken &token);
bool flush();
size_t size() const;
// delete copy/move
BufferQueue(const BufferQueue &) = delete;
BufferQueue &operator=(const BufferQueue &) = delete;
BufferQueue(BufferQueue &&) = delete;
BufferQueue &operator=(BufferQueue &&) = delete;
private:
bool enqueue(QueueItem item, ProducerToken &token);
bool enqueueBatch(std::vector<QueueItem> items, ProducerToken &token);
};
#endif
|