71 lines
2.0 KiB
C++
71 lines
2.0 KiB
C++
#ifndef REXY_THREADPOOL_HPP
|
|
#define REXY_THREADPOOL_HPP
|
|
|
|
#include <vector>
|
|
#include <queue>
|
|
#include <condition_variable>
|
|
#include <mutex> //unique_lock, scoped_lock
|
|
#include <shared_mutex> //shared_mutex, shared_lock
|
|
#include <atomic> //atomic_bool
|
|
#include <memory> //shared_ptr
|
|
#include <future> //future, packaged_task
|
|
#include <functional> //function, bind
|
|
#include <utility> //move, forward
|
|
|
|
namespace rexy{
|
|
|
|
class threadpool
|
|
{
|
|
private:
|
|
using mutex_t = std::shared_mutex;
|
|
using write_lock_t = std::unique_lock<mutex_t>;
|
|
using read_lock_t = std::shared_lock<mutex_t>;
|
|
|
|
private:
|
|
mutable read_lock_t m_ctor_lock;
|
|
mutable std::condition_variable_any m_qcv;
|
|
mutable mutex_t m_qlk;
|
|
|
|
std::vector<std::thread> m_workers;
|
|
std::queue<std::function<void()>> m_jobs;
|
|
std::atomic_bool m_valid = true;
|
|
|
|
public:
|
|
explicit threadpool(int numthreads = std::thread::hardware_concurrency());
|
|
threadpool(const threadpool&);
|
|
threadpool(threadpool&&);
|
|
~threadpool(void);
|
|
|
|
threadpool& operator=(const threadpool&) = delete;
|
|
threadpool& operator=(threadpool&&) = delete;
|
|
|
|
template<class Func, class... Args>
|
|
auto add_job(Func&& f, Args&&... args) -> std::future<decltype(std::forward<Func>(f)(std::forward<Args>(args)...))>;
|
|
|
|
private:
|
|
void worker_loop(void);
|
|
};
|
|
|
|
template<class Func, class... Args>
|
|
auto threadpool::add_job(Func&& f, Args&&... args) -> std::future<decltype(std::forward<Func>(f)(std::forward<Args>(args)...))>{
|
|
using return_t = decltype(std::forward<Func>(f)(std::forward<Args>(args)...));
|
|
using task_t = std::packaged_task<return_t(void)>;
|
|
|
|
//shared pointer to a packaged task which takes no arguments in operator()
|
|
std::shared_ptr<task_t> task_ptr = std::make_shared<task_t>(std::bind(std::forward<Func>(f), std::forward<Args>(args)...));
|
|
|
|
{
|
|
write_lock_t lk(m_qlk);
|
|
//make a copy of the shared pointer by capturing by-value
|
|
m_jobs.emplace([task_ptr]{(*task_ptr)();});
|
|
}
|
|
|
|
//wakeup a worker to run the job
|
|
m_qcv.notify_one();
|
|
return task_ptr->get_future();
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|