blob: f0dc9af050021335706968d84a3e657620247da0 (
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
|
#include "base/worker_thread.hpp"
using namespace std;
namespace base
{
WorkerThread::WorkerThread()
{
m_thread = threads::SimpleThread(&WorkerThread::ProcessTasks, this);
}
WorkerThread::~WorkerThread()
{
ASSERT(m_checker.CalledOnOriginalThread(), ());
Shutdown(Exit::SkipPending);
m_thread.join();
}
void WorkerThread::ProcessTasks()
{
queue<Task> pending;
unique_lock<mutex> lk(m_mu, defer_lock);
while (true)
{
Task task;
{
lk.lock();
m_cv.wait(lk, [this]() { return m_shutdown || !m_queue.empty(); });
if (m_shutdown)
{
switch (m_exit)
{
case Exit::ExecPending:
CHECK(pending.empty(), ());
m_queue.swap(pending);
break;
case Exit::SkipPending: break;
}
break;
}
CHECK(!m_queue.empty(), ());
task = move(m_queue.front());
m_queue.pop();
lk.unlock();
}
task();
}
while (!pending.empty())
{
pending.front()();
pending.pop();
}
}
bool WorkerThread::Shutdown(Exit e)
{
lock_guard<mutex> lk(m_mu);
if (m_shutdown)
return false;
m_shutdown = true;
m_exit = e;
m_cv.notify_one();
return true;
}
} // namespace base
|