-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtsqueue.h
113 lines (99 loc) · 1.78 KB
/
tsqueue.h
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#include <mutex>
#include <deque>
template<typename T>
class tsqueue
{
public:
tsqueue() = default;
tsqueue(const tsqueue<T>&) = delete;
virtual ~tsqueue() { clear(); }
public:
const T& front()
{
std::lock_guard<std::mutex> lock(mutex_);
return queue_.front();
}
const T& back()
{
std::lock_guard<std::mutex> lock(mutex_);
return queue_.back();
}
T pop_front()
{
std::lock_guard<std::mutex> lock(mutex_);
auto t = std::move(queue_.front());
queue_.pop_front();
return t;
}
T pop_back()
{
std::lock_guard<std::mutex> lock(mutex_);
auto t = std::move(queue_.back());
queue_.pop_back();
return t;
}
void push_back(const T& item)
{
{
std::lock_guard<std::mutex> lock(mutex_);
queue_.emplace_back(std::move(item));
}
std::unique_lock<std::mutex> ul(cv_mutex_);
cv_.notify_one();
}
void push_front(const T& item)
{
{
std::lock_guard<std::mutex> lock(mutex_);
queue_.emplace_front(std::move(item));
}
std::unique_lock<std::mutex> ul(cv_mutex_);
cv_.notify_one();
}
bool empty()
{
std::lock_guard<std::mutex> lock(mutex_);
return queue_.empty();
}
size_t count()
{
std::lock_guard<std::mutex> lock(mutex_);
return queue_.size();
}
void clear()
{
std::lock_guard<std::mutex> lock(mutex_);
queue_.clear();
}
int wait(int timeout = 0)
{
int ret = 1;
if (timeout == 0)
{
while (empty())
{
std::unique_lock<std::mutex> ul(cv_mutex_);
cv_.wait(ul);
}
}
else
{
std::unique_lock<std::mutex> ul(cv_mutex_);
auto r = cv_.wait_for(ul, std::chrono::milliseconds(timeout), [this]{return !empty();});
if (!r)
{
ret = 0;
}
else
{
ret = 1;
}
}
return ret;
}
protected:
std::mutex mutex_;
std::deque<T> queue_;
std::condition_variable cv_;
std::mutex cv_mutex_;
};