lokinet/llarp/util/threadpool.h

94 lines
2.1 KiB
C
Raw Normal View History

2018-01-27 01:18:10 +00:00
#ifndef LLARP_THREADPOOL_H
#define LLARP_THREADPOOL_H
#include <util/threading.hpp>
#include <util/threadpool.hpp>
#include <absl/base/thread_annotations.h>
#include <memory>
#include <queue>
struct llarp_threadpool
{
std::unique_ptr< llarp::thread::ThreadPool > impl;
mutable llarp::util::Mutex m_access; // protects jobs
std::queue< std::function< void(void) > > jobs GUARDED_BY(m_access);
llarp_threadpool(int workers, const char *name)
: impl(
std::make_unique< llarp::thread::ThreadPool >(workers, workers * 128))
{
(void)name;
}
llarp_threadpool()
{
}
size_t
size() const LOCKS_EXCLUDED(m_access)
{
absl::ReaderMutexLock l(&m_access);
return jobs.size();
}
};
2018-01-27 01:18:10 +00:00
struct llarp_threadpool *
llarp_init_threadpool(int workers, const char *name);
2018-06-06 12:46:26 +00:00
/// for single process mode
struct llarp_threadpool *
llarp_init_same_process_threadpool();
2018-08-18 14:01:21 +00:00
typedef bool (*setup_net_func)(void *, bool);
typedef void (*run_main_func)(void *);
/// for network isolation
struct llarp_threadpool *
2018-08-09 19:02:17 +00:00
llarp_init_isolated_net_threadpool(const char *name, setup_net_func setupNet,
run_main_func runMain, void *context);
void
llarp_free_threadpool(struct llarp_threadpool **tp);
2018-01-27 01:18:10 +00:00
2018-02-01 13:21:00 +00:00
typedef void (*llarp_thread_work_func)(void *);
2018-01-31 19:59:26 +00:00
2018-01-29 14:27:24 +00:00
/** job to be done in worker thread */
struct llarp_thread_job
{
2018-01-29 14:27:24 +00:00
/** user data to pass to work function */
void *user;
/** called in threadpool worker thread */
2018-01-31 19:59:26 +00:00
llarp_thread_work_func work;
2018-07-16 04:55:46 +00:00
#ifdef __cplusplus
2018-06-06 12:46:26 +00:00
llarp_thread_job(void *u, llarp_thread_work_func w) : user(u), work(w)
{
}
llarp_thread_job() : user(nullptr), work(nullptr)
{
}
2018-07-16 04:55:46 +00:00
#endif
2018-01-29 14:27:24 +00:00
};
2018-01-27 01:18:10 +00:00
2018-06-06 12:46:26 +00:00
/// for single process mode
void
llarp_threadpool_tick(struct llarp_threadpool *tp);
void
llarp_threadpool_queue_job(struct llarp_threadpool *tp,
struct llarp_thread_job j);
2018-11-19 16:40:49 +00:00
void
llarp_threadpool_start(struct llarp_threadpool *tp);
2018-01-29 14:27:24 +00:00
void
llarp_threadpool_stop(struct llarp_threadpool *tp);
void
llarp_threadpool_join(struct llarp_threadpool *tp);
2018-04-30 16:14:20 +00:00
void
llarp_threadpool_wait(struct llarp_threadpool *tp);
2018-01-27 01:18:10 +00:00
#endif