lokinet/llarp/util/threadpool.cpp
Thomas Winget 38fd0552d3 Adds Link-Relay Status Messages
Success case:
  - the path endpoint creates and sends a LR_StatusMessage upon
    successful path creation

Failure case:
  - an intermediate hop creates and sends a LR_StatusMessage upon
    failure to forward the path to the next hop for any reason

Both cases:
  - transit hops receive LR_StatusMessages and add a frame
    to them reflecting their "status" with respect to that path
  - the path creator receives LR_StatusMessages and decrypts/parses
    the LR_StatusRecord frames from the path hops.  If all is good,
    the Path does as it would when receiving a PathConfirmMessage.
    If not, the Path marks the new path as failed.

LR_StatusMessage is now used/sent in place of PathConfirmMessage
2019-07-25 17:46:56 -04:00

114 lines
2.1 KiB
C++

#include <util/logger.hpp>
#include <util/time.hpp>
#include <util/threadpool.h>
#include <util/thread_pool.hpp>
#include <cstring>
#include <functional>
#include <queue>
struct llarp_threadpool *
llarp_init_threadpool(int workers, const char *name)
{
if(workers <= 0)
workers = 1;
return new llarp_threadpool(workers, name);
}
struct llarp_threadpool *
llarp_init_same_process_threadpool()
{
return new llarp_threadpool();
}
void
llarp_threadpool_join(struct llarp_threadpool *pool)
{
llarp::LogDebug("threadpool join");
if(pool->impl)
pool->impl->drain();
}
void
llarp_threadpool_start(struct llarp_threadpool *pool)
{
if(pool->impl)
pool->impl->start();
}
void
llarp_threadpool_stop(struct llarp_threadpool *pool)
{
llarp::LogDebug("threadpool stop");
if(pool->impl)
pool->impl->stop();
if(pool->jobs)
pool->jobs->disable();
}
void
llarp_threadpool_wait(struct llarp_threadpool *pool)
{
llarp::LogDebug("threadpool wait");
if(pool->impl)
{
pool->impl->drain();
}
}
void
llarp_threadpool_queue_job(struct llarp_threadpool *pool,
struct llarp_thread_job job)
{
return llarp_threadpool_queue_job(pool, (std::bind(job.work, job.user)));
}
void
llarp_threadpool_queue_job(struct llarp_threadpool *pool,
std::function< void() > func)
{
if(pool->impl)
{
while(!pool->impl->tryAddJob(func))
{
std::this_thread::sleep_for(std::chrono::microseconds(1000));
}
}
else
{
// single threaded mode
while(pool->jobs->tryPushBack(func) != 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));
}
}
}
void
llarp_threadpool_tick(struct llarp_threadpool *pool)
{
while(pool->size())
{
auto job = pool->jobs->tryPopFront();
if(job)
{
(*job)();
}
}
}
void
llarp_free_threadpool(struct llarp_threadpool **pool)
{
if(*pool)
{
delete *pool;
}
*pool = nullptr;
}