-
Notifications
You must be signed in to change notification settings - Fork 1
/
runloop.cpp
104 lines (83 loc) · 1.71 KB
/
runloop.cpp
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
#include <runloop.hpp>
#include <unistd.h>
#include <fcntl.h>
class SpinLock {
public:
class Guard {
public:
Guard(std::atomic_bool &locked) : _locked(locked) {
while (_locked) {
// Spin!
}
_locked = true;
}
~Guard() {
_locked = false;
}
private:
std::atomic_bool &_locked;
};
inline Guard acquire() {
return {_locked};
}
private:
std::atomic_bool _locked{false};
};
class Runloop::Impl {
public:
Impl() {
pipe(_pipe);
fcntl(_pipe[0], F_SETFL, O_NONBLOCK);
}
void wait() {
char c;
while (read(_pipe[0], &c, 1) > 0);
}
void wake() {
write(_pipe[1], "*", 1);
}
SpinLock &lock() {
return _lock;
}
private:
int _pipe[2]{0};
SpinLock _lock;
};
Runloop::Runloop() : _impl(std::make_unique<Runloop::Impl>()) {
current().set(this);
}
Runloop::~Runloop() {
current().set(nullptr);
}
void Runloop::run() {
_running = true;
std::queue<std::shared_ptr<WorkTask>> tmp;
std::shared_ptr<WorkTask> task;
while (_running) {
{
auto guard = _impl->lock().acquire();
tmp.swap(_queue);
}
while (!tmp.empty()) {
if ((task = tmp.front())) {
(*task)();
}
tmp.pop();
}
_impl->wait();
}
}
void Runloop::stop() {
invoke([this] {
_running = false;
wake();
});
}
void Runloop::invoke(WorkTask &&fn) {
auto guard = _impl->lock().acquire();
_queue.push(std::make_shared<WorkTask>(std::move(fn)));
wake();
}
void Runloop::wake() {
_impl->wake();
}