blob: 7198b65fbe79da5731b9a72da8b60b6d1ece8fdd (
plain)
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
75
76
77
78
79
|
#pragma once
#include <condition_variable>
#include <mutex>
#include <queue>
#include <utility>
namespace base
{
namespace threads
{
template <typename T>
class ThreadSafeQueue
{
public:
ThreadSafeQueue() = default;
ThreadSafeQueue(ThreadSafeQueue const & other)
{
std::lock_guard<std::mutex> lk(other.m_mutex);
m_queue = other.m_queue;
}
void Push(T const & value)
{
{
std::lock_guard<std::mutex> lk(m_mutex);
m_queue.push(value);
}
m_cond.notify_one();
}
void Push(T && value)
{
{
std::lock_guard<std::mutex> lk(m_mutex);
m_queue.push(std::move(value));
}
m_cond.notify_one();
}
void WaitAndPop(T & value)
{
std::unique_lock<std::mutex> lk(m_mutex);
m_cond.wait(lk, [this]{ return !m_queue.empty(); });
value = std::move(m_queue.front());
m_queue.pop();
}
bool TryPop(T & value)
{
std::lock_guard<std::mutex> lk(m_mutex);
if (m_queue.empty())
return false;
value = std::move(m_queue.front());
m_queue.pop();
return true;
}
bool Empty() const
{
std::lock_guard<std::mutex> lk(m_mutex);
return m_queue.empty();
}
size_t Size() const
{
std::lock_guard<std::mutex> lk(m_mutex);
return m_queue.size();
}
private:
mutable std::mutex m_mutex;
std::queue<T> m_queue;
std::condition_variable m_cond;
};
} // namespace threads
} // namespace base
|
