-
Notifications
You must be signed in to change notification settings - Fork 11
Expand file tree
/
Copy pathKThreadPool.cpp
More file actions
executable file
·75 lines (64 loc) · 1.64 KB
/
Copy pathKThreadPool.cpp
File metadata and controls
executable file
·75 lines (64 loc) · 1.64 KB
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 "KThreadPool.h"
#include "webserver/utils/KAsyncLogger.h"
#include <assert.h>
ThreadPool::ThreadPool(const std::string &name)
: name_(name), running_(false) {}
ThreadPool::~ThreadPool() {
if (running_) {
stop();
}
}
void ThreadPool::start(int numThreads) {
assert(threads_.empty());
running_ = true;
threads_.reserve(numThreads);
for (int i = 0; i < numThreads; i++) {
// 注意这里是以右值的方式push_back
threads_.push_back(std::thread(std::bind(&ThreadPool::runInThread, this)));
}
}
void ThreadPool::stop() {
running_ = false;
cond_.notify_all();
for (std::thread &th : threads_) {
th.join();
}
}
void ThreadPool::run(const Task &task) {
if (threads_.empty()) {
task();
} else {
std::unique_lock<std::mutex> lock(mutex_);
queue_.push_back(task);
cond_.notify_one();
}
}
ThreadPool::Task ThreadPool::take() {
std::unique_lock<std::mutex> lock(mutex_);
this->cond_.wait(
lock, [this] { return (!this->queue_.empty()) || (!this->running_); });
Task task;
if (!this->queue_.empty()) {
task = queue_.front();
this->queue_.pop_front();
}
return task;
}
void ThreadPool::runInThread() {
try {
while (running_) {
Task task(this->take());
if (task) {
task();
}
}
} catch (const std::exception &ex) {
KBACK_LOG_ERROR("exception caught in ThreadPool %s", name_.c_str());
KBACK_LOG_ERROR("reason: %s", ex.what());
std::abort();
} catch (...) {
KBACK_LOG_ERROR("unknown exception caught in ThreadPool %s",
name_.c_str());
abort();
}
}