-
Notifications
You must be signed in to change notification settings - Fork 0
/
scheduler.h
68 lines (48 loc) · 1.48 KB
/
scheduler.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
//
// Created by weitao on 7/1/16.
//
#pragma once
#include <memory>
#include <functional>
#include <deque>
#include "event/poller.h"
#include "timer/timer_queue.h"
#include "awakener.h"
namespace snow {
class scheduler {
public:
typedef std::function<void()> task_type;
typedef std::unique_ptr<task_type> task_type_ptr;
typedef std::function<void()> call_back_before_loop_type;
typedef std::function<void()> call_back_after_loop_type;
static scheduler &instance();
void run();
void post(task_type_ptr &task);
void spawn(task_type_ptr &task);
poller& get_poller() {
return m_poller;
}
timer_queue& get_timer_queue() {
return m_timer_queue;
}
void set_call_back_before_loop(call_back_before_loop_type cb) {
m_cb_before_loop = std::move(cb);
}
void set_call_back_after_loop(call_back_after_loop_type cb) {
m_cb_after_loop = std::move(cb);
}
private:
scheduler();
scheduler(const scheduler&) = delete;
void operator=(const scheduler&) = delete;
void wake_up();
private:
poller m_poller;
timer_queue m_timer_queue;
awakener m_awakener;
call_back_before_loop_type m_cb_before_loop;
call_back_after_loop_type m_cb_after_loop;
std::deque<std::unique_ptr<task_type>> m_task_queue;
bool m_running;
};
}