forked from OctaForge/libostd
revert the threadpool split (wasn't used in the end)
This commit is contained in:
parent
c120f49634
commit
307123fd35
|
@ -20,86 +20,6 @@
|
||||||
namespace ostd {
|
namespace ostd {
|
||||||
|
|
||||||
namespace detail {
|
namespace detail {
|
||||||
/* can be used as a base for any custom thread pool, internal */
|
|
||||||
struct thread_pool_base {
|
|
||||||
virtual ~thread_pool_base() {
|
|
||||||
destroy();
|
|
||||||
}
|
|
||||||
|
|
||||||
void destroy() {
|
|
||||||
{
|
|
||||||
std::lock_guard<std::mutex> l{p_lock};
|
|
||||||
if (!p_running) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
p_running = false;
|
|
||||||
}
|
|
||||||
p_cond.notify_all();
|
|
||||||
for (auto &tid: p_thrs) {
|
|
||||||
tid.join();
|
|
||||||
p_cond.notify_all();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
protected:
|
|
||||||
thread_pool_base() {}
|
|
||||||
|
|
||||||
thread_pool_base(thread_pool_base const &) = delete;
|
|
||||||
thread_pool_base(thread_pool_base &&) = delete;
|
|
||||||
thread_pool_base &operator=(thread_pool_base const &) = delete;
|
|
||||||
thread_pool_base &operator=(thread_pool_base &&) = delete;
|
|
||||||
|
|
||||||
template<typename B>
|
|
||||||
void start(size_t size) {
|
|
||||||
p_running = true;
|
|
||||||
auto tf = [this]() {
|
|
||||||
thread_run<B>();
|
|
||||||
};
|
|
||||||
for (size_t i = 0; i < size; ++i) {
|
|
||||||
std::thread tid{tf};
|
|
||||||
if (!tid.joinable()) {
|
|
||||||
throw std::runtime_error{"thread_pool worker failed"};
|
|
||||||
}
|
|
||||||
p_thrs.push_back(std::move(tid));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template<typename F>
|
|
||||||
void push_task(F &&func) {
|
|
||||||
{
|
|
||||||
std::lock_guard<std::mutex> l{this->p_lock};
|
|
||||||
if (!p_running) {
|
|
||||||
throw std::runtime_error{"push on stopped thread_pool"};
|
|
||||||
}
|
|
||||||
func();
|
|
||||||
}
|
|
||||||
p_cond.notify_one();
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
template<typename B>
|
|
||||||
void thread_run() {
|
|
||||||
B &self = *static_cast<B *>(this);
|
|
||||||
for (;;) {
|
|
||||||
std::unique_lock<std::mutex> l{p_lock};
|
|
||||||
while (p_running && self.empty()) {
|
|
||||||
p_cond.wait(l);
|
|
||||||
}
|
|
||||||
if (!p_running && self.empty()) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
self.task_run(l);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
std::condition_variable p_cond;
|
|
||||||
std::mutex p_lock;
|
|
||||||
std::vector<std::thread> p_thrs;
|
|
||||||
bool p_running = false;
|
|
||||||
};
|
|
||||||
|
|
||||||
/* regular thread pool task, as lightweight as possible */
|
|
||||||
|
|
||||||
struct tpool_func_base {
|
struct tpool_func_base {
|
||||||
tpool_func_base() {}
|
tpool_func_base() {}
|
||||||
virtual ~tpool_func_base() {}
|
virtual ~tpool_func_base() {}
|
||||||
|
@ -167,16 +87,38 @@ namespace detail {
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
struct thread_pool: detail::thread_pool_base {
|
struct thread_pool {
|
||||||
private:
|
|
||||||
friend struct detail::thread_pool_base;
|
|
||||||
using base_t = detail::thread_pool_base;
|
|
||||||
|
|
||||||
public:
|
|
||||||
thread_pool(): base_t() {}
|
|
||||||
|
|
||||||
void start(size_t size = std::thread::hardware_concurrency()) {
|
void start(size_t size = std::thread::hardware_concurrency()) {
|
||||||
base_t::template start<thread_pool>(size);
|
p_running = true;
|
||||||
|
auto tf = [this]() {
|
||||||
|
thread_run();
|
||||||
|
};
|
||||||
|
for (size_t i = 0; i < size; ++i) {
|
||||||
|
std::thread tid{tf};
|
||||||
|
if (!tid.joinable()) {
|
||||||
|
throw std::runtime_error{"thread_pool worker failed"};
|
||||||
|
}
|
||||||
|
p_thrs.push_back(std::move(tid));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
~thread_pool() {
|
||||||
|
destroy();
|
||||||
|
}
|
||||||
|
|
||||||
|
void destroy() {
|
||||||
|
{
|
||||||
|
std::lock_guard<std::mutex> l{p_lock};
|
||||||
|
if (!p_running) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
p_running = false;
|
||||||
|
}
|
||||||
|
p_cond.notify_all();
|
||||||
|
for (auto &tid: p_thrs) {
|
||||||
|
tid.join();
|
||||||
|
p_cond.notify_all();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename F, typename ...A>
|
template<typename F, typename ...A>
|
||||||
|
@ -193,25 +135,39 @@ public:
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
auto ret = t.get_future();
|
auto ret = t.get_future();
|
||||||
this->push_task([this, &t]() {
|
{
|
||||||
|
std::lock_guard<std::mutex> l{p_lock};
|
||||||
|
if (!p_running) {
|
||||||
|
throw std::runtime_error{"push on stopped thread_pool"};
|
||||||
|
}
|
||||||
p_tasks.emplace(std::move(t));
|
p_tasks.emplace(std::move(t));
|
||||||
});
|
}
|
||||||
|
p_cond.notify_one();
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
bool empty() const {
|
void thread_run() {
|
||||||
return p_tasks.empty();
|
for (;;) {
|
||||||
}
|
std::unique_lock<std::mutex> l{p_lock};
|
||||||
|
while (p_running && p_tasks.empty()) {
|
||||||
void task_run(std::unique_lock<std::mutex> &l) {
|
p_cond.wait(l);
|
||||||
auto t{std::move(p_tasks.front())};
|
}
|
||||||
p_tasks.pop();
|
if (!p_running && p_tasks.empty()) {
|
||||||
l.unlock();
|
return;
|
||||||
t();
|
}
|
||||||
|
auto t{std::move(p_tasks.front())};
|
||||||
|
p_tasks.pop();
|
||||||
|
l.unlock();
|
||||||
|
t();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::condition_variable p_cond;
|
||||||
|
std::mutex p_lock;
|
||||||
|
std::vector<std::thread> p_thrs;
|
||||||
std::queue<detail::tpool_func> p_tasks;
|
std::queue<detail::tpool_func> p_tasks;
|
||||||
|
bool p_running = false;
|
||||||
};
|
};
|
||||||
|
|
||||||
} /* namespace ostd */
|
} /* namespace ostd */
|
||||||
|
|
Loading…
Reference in a new issue