54 lines
1.3 KiB
C++
54 lines
1.3 KiB
C++
#include <llmr/util/threadpool.hpp>
|
|
#include <llmr/util/std.hpp>
|
|
#include <thread>
|
|
|
|
using namespace llmr::util;
|
|
|
|
std::unique_ptr<Threadpool> llmr::util::threadpool = std::make_unique<Threadpool>(std::thread::hardware_concurrency());
|
|
|
|
Threadpool::Threadpool(int max_workers)
|
|
: max_workers(max_workers) {
|
|
}
|
|
|
|
void Threadpool::add(Callback callback, void *data) {
|
|
if (worker_count < max_workers) {
|
|
worker_count++;
|
|
workers.emplace_front(*this);
|
|
}
|
|
|
|
pthread_mutex_lock(&mutex);
|
|
tasks.push(std::make_pair(callback, data));
|
|
pthread_mutex_unlock(&mutex);
|
|
pthread_cond_signal(&condition);
|
|
}
|
|
|
|
Threadpool::Worker::Worker(Threadpool& pool)
|
|
: pool(pool) {
|
|
pthread_create(&thread, nullptr, loop, (void *)this);
|
|
}
|
|
|
|
Threadpool::Worker::~Worker() {
|
|
pthread_cancel(thread);
|
|
}
|
|
|
|
|
|
void *Threadpool::Worker::loop(void *ptr) {
|
|
Worker *worker = static_cast<Worker *>(ptr);
|
|
Threadpool& pool = worker->pool;
|
|
|
|
pthread_mutex_lock(&pool.mutex);
|
|
while (true) {
|
|
if (pool.tasks.size()) {
|
|
Threadpool::Task task = pool.tasks.front();
|
|
pool.tasks.pop();
|
|
pthread_mutex_unlock(&pool.mutex);
|
|
task.first(task.second);
|
|
pthread_mutex_lock(&pool.mutex);
|
|
} else {
|
|
pthread_cond_wait(&pool.condition, &pool.mutex);
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|