-
Notifications
You must be signed in to change notification settings - Fork 1
/
workqueueinsomniac.h
54 lines (41 loc) · 1.71 KB
/
workqueueinsomniac.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
#ifndef JWUTIL_WORKQUEUEINSOMNIAC_H
#define JWUTIL_WORKQUEUEINSOMNIAC_H
#include <chrono>
#include "workqueuebase.h"
namespace jw_util
{
template <unsigned int num_threads, typename... ArgTypes>
class WorkQueueInsomniac : public WorkQueueBase<WorkQueueInsomniac<num_threads, ArgTypes...>, num_threads, ArgTypes...>
{
typedef WorkQueueBase<WorkQueueInsomniac<num_threads, ArgTypes...>, num_threads, ArgTypes...> BaseType;
friend BaseType;
public:
using WorkQueueBase<WorkQueueInsomniac<num_threads, ArgTypes...>, num_threads, ArgTypes...>::WorkQueueBase;
void set_wakeup_worker(jw_util::MethodCallback<> worker)
{
wakeup_worker = worker;
}
template <typename DurationRep, typename DurationPeriod>
void set_wakeup_interval(std::chrono::duration<DurationRep, DurationPeriod> duration)
{
next_wakeup = std::chrono::steady_clock::now();
wakeup_interval = std::chrono::duration_cast<std::chrono::steady_clock::duration>(duration);
}
private:
jw_util::MethodCallback<> wakeup_worker;
std::chrono::steady_clock::time_point next_wakeup = std::chrono::steady_clock::time_point::max();
std::chrono::steady_clock::duration wakeup_interval = std::chrono::steady_clock::duration::max();
void wait(std::unique_lock<std::mutex> &lock)
{
std::cv_status status = BaseType::conditional_variable.wait_until(lock, next_wakeup);
if (status == std::cv_status::timeout)
{
std::chrono::steady_clock::time_point now = std::chrono::steady_clock::now();
next_wakeup += wakeup_interval;
if (next_wakeup < now) {next_wakeup = now;}
wakeup_worker.call();
}
}
};
}
#endif // JWUTIL_WORKQUEUEINSOMNIAC_H