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
|
#pragma once
#include <condition_variable>
template<class T>
class Condition
{
public:
std::mutex m_mutex;
std::condition_variable m_condition;
~Condition()
{
m_condition.notify_all();
}
void notify(std::unique_ptr<T> data) noexcept
{
{
std::lock_guard<std::mutex> eventLock(m_mutex);
any.swap(data);
}
// wake up one waiter
m_condition.notify_one();
};
std::unique_ptr<T> wait(unsigned timeout = 0)
{
std::unique_lock<std::mutex> ul(m_mutex);
if (!timeout)
{
m_condition.wait(
ul,
[&]()
{
if (!any)
{
return false;
}
return true;
}
);
}
else
{
if (!any)
{
std::cv_status status = m_condition.wait_for(ul, std::chrono::milliseconds(timeout));
if (status == std::cv_status::timeout)
{
return {};
}
}
}
return std::unique_ptr<T>(any.release());
}
private:
std::unique_ptr<T> any;
};
|
