blob: a5f550041d36fa8a302a1324efbbfd048271d896 (
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
|
#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;
};
|