blob: 4c1292bfd1bbe255143b95123eda5817b5725bc6 (
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
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
|
#include "common.hh"
#include <deque>
#include <mutex>
#include "io.hh"
#include "looper.hh"
#include "task_runner.hh"
#include "unique_pipe.hh"
namespace {
constexpr char kMessage = 1;
class TaskRunnerLooper : public TaskRunner {
public:
explicit TaskRunnerLooper(std::shared_ptr<Looper> looper)
: looper_(looper), pipe_(true, false) {
looper_->add(pipe_.reader(), Looper::EVENT_READ,
std::bind(&TaskRunnerLooper::run, this,
std::placeholders::_1));
}
~TaskRunnerLooper() override {
looper_->remove(pipe_.reader());
}
void post(std::function<void()> callback) override {
bool notify;
{
std::lock_guard<std::mutex> lock(mutex_);
notify = queue_.empty();
queue_.push_back(std::move(callback));
}
if (notify)
io::write_all(pipe_.writer(), &kMessage, 1);
}
private:
void run(uint8_t event) {
if (event & Looper::EVENT_READ) {
char in[1];
io::read_all(pipe_.reader(), &in, 1);
}
if (event & Looper::EVENT_ERROR) {
assert(false);
looper_->quit();
}
bool more;
do {
std::function<void()> callback;
{
std::lock_guard<std::mutex> lock(mutex_);
if (queue_.empty())
break;
callback = std::move(queue_.front());
queue_.pop_front();
more = !queue_.empty();
}
callback();
} while (more);
}
std::shared_ptr<Looper> looper_;
unique_pipe pipe_;
std::mutex mutex_;
std::deque<std::function<void()>> queue_;
};
} // namespace
std::unique_ptr<TaskRunner> TaskRunner::create(std::shared_ptr<Looper> looper) {
return std::make_unique<TaskRunnerLooper>(looper);
}
|
