mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
2.0 KiB
C++
97 lines
2.0 KiB
C++
#include "threadpool.hpp"
|
|
#include <iostream>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace thread
|
|
{
|
|
Pool::Pool(size_t workers)
|
|
{
|
|
stop.store(true);
|
|
while(workers--)
|
|
{
|
|
threads.emplace_back(
|
|
[this]
|
|
{
|
|
for(;;)
|
|
{
|
|
llarp_thread_job job;
|
|
{
|
|
lock_t lock(this->queue_mutex);
|
|
this->condition.wait(lock, [this]{ return this->stop || !this->jobs.empty(); });
|
|
if(this->stop && this->jobs.empty())
|
|
return;
|
|
job = std::move(this->jobs.front());
|
|
this->jobs.pop();
|
|
}
|
|
// do work
|
|
job.work(job.user);
|
|
// inform result if needed
|
|
if(job.result && job.result->loop)
|
|
if(!llarp_ev_async(job.result->loop, *job.result))
|
|
{
|
|
std::cerr << "failed to queue result in thread worker" << std::endl;
|
|
}
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
void Pool::Join()
|
|
{
|
|
{
|
|
lock_t lock(queue_mutex);
|
|
stop.store(true);
|
|
}
|
|
condition.notify_all();
|
|
for(auto & t : threads)
|
|
t.join();
|
|
}
|
|
|
|
void Pool::QueueJob(llarp_thread_job job)
|
|
{
|
|
{
|
|
lock_t lock(queue_mutex);
|
|
|
|
// don't allow enqueueing after stopping the pool
|
|
if(stop)
|
|
throw std::runtime_error("enqueue on stopped ThreadPool");
|
|
|
|
jobs.emplace(job);
|
|
}
|
|
condition.notify_one();
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
struct llarp_threadpool
|
|
{
|
|
llarp::thread::Pool impl;
|
|
|
|
llarp_threadpool(int workers) : impl(workers) {}
|
|
};
|
|
|
|
extern "C" {
|
|
|
|
struct llarp_threadpool * llarp_init_threadpool(int workers)
|
|
{
|
|
if(workers > 0)
|
|
return new llarp_threadpool(workers);
|
|
else
|
|
return nullptr;
|
|
}
|
|
|
|
void llarp_threadpool_join(struct llarp_threadpool * pool)
|
|
{
|
|
pool->impl.Join();
|
|
}
|
|
|
|
void llarp_free_threadpool(struct llarp_threadpool ** pool)
|
|
{
|
|
delete *pool;
|
|
*pool = nullptr;
|
|
}
|
|
|
|
}
|