lokinet/llarp/util/threadpool.cpp

107 lines
1.9 KiB
C++
Raw Normal View History

#include <util/logger.hpp>
#include <util/threadpool.hpp>
#include <util/time.hpp>
2018-06-06 12:46:26 +00:00
#include <cstring>
#include <functional>
2018-06-06 12:46:26 +00:00
#include <queue>
struct llarp_threadpool *
llarp_init_threadpool(int workers, const char *name)
{
2018-09-07 20:48:52 +00:00
if(workers <= 0)
workers = 1;
2018-11-19 11:56:40 +00:00
return new llarp_threadpool(workers, name);
2018-01-29 14:27:24 +00:00
}
2018-01-29 14:19:00 +00:00
2018-06-06 12:46:26 +00:00
struct llarp_threadpool *
llarp_init_same_process_threadpool()
{
return new llarp_threadpool();
}
void
llarp_threadpool_join(struct llarp_threadpool *pool)
{
llarp::LogDebug("threadpool join");
2018-06-06 12:46:26 +00:00
if(pool->impl)
2018-11-19 11:56:40 +00:00
pool->impl->drain();
}
2018-01-29 14:19:00 +00:00
void
2018-11-19 11:56:40 +00:00
llarp_threadpool_start(struct llarp_threadpool *pool)
{
if(pool->impl)
pool->impl->start();
2018-02-01 13:21:00 +00:00
}
void
llarp_threadpool_stop(struct llarp_threadpool *pool)
{
llarp::LogDebug("threadpool stop");
2018-06-06 12:46:26 +00:00
if(pool->impl)
2018-11-19 11:56:40 +00:00
pool->impl->stop();
if(pool->jobs)
pool->jobs->disable();
}
2018-04-30 14:57:13 +00:00
void
llarp_threadpool_wait(struct llarp_threadpool *pool)
{
llarp::LogDebug("threadpool wait");
2018-06-06 12:46:26 +00:00
if(pool->impl)
2018-04-30 14:57:13 +00:00
{
2018-11-19 11:56:40 +00:00
pool->impl->drain();
2018-04-30 14:57:13 +00:00
}
}
void
llarp_threadpool_queue_job(struct llarp_threadpool *pool,
struct llarp_thread_job job)
{
2018-06-06 12:46:26 +00:00
if(pool->impl)
2019-05-15 15:54:26 +00:00
{
while(!pool->impl->tryAddJob(std::bind(job.work, job.user)))
{
2019-05-15 22:03:24 +00:00
std::this_thread::sleep_for(std::chrono::microseconds(1000));
2019-05-15 15:54:26 +00:00
}
}
2018-08-30 18:48:43 +00:00
else
{
2018-08-30 18:48:43 +00:00
// single threaded mode
while(pool->jobs->tryPushBack(std::bind(job.work, job.user))
!= llarp::thread::QueueReturn::Success)
{
if(!pool->jobs->enabled())
return;
if(::getpid() == pool->callingPID)
llarp_threadpool_tick(pool);
else
std::this_thread::sleep_for(std::chrono::microseconds(1000));
}
}
2018-06-06 12:46:26 +00:00
}
void
llarp_threadpool_tick(struct llarp_threadpool *pool)
{
while(pool->size())
2018-06-06 12:46:26 +00:00
{
auto job = pool->jobs->tryPopFront();
2018-12-27 19:10:38 +00:00
if(job)
{
(*job)();
}
2018-06-06 12:46:26 +00:00
}
2018-01-31 19:59:26 +00:00
}
void
llarp_free_threadpool(struct llarp_threadpool **pool)
{
if(*pool)
{
delete *pool;
}
2018-01-29 14:27:24 +00:00
*pool = nullptr;
}