93 lines
2.7 KiB
C++
93 lines
2.7 KiB
C++
/**
|
|
This file is a part of rexy's general purpose library
|
|
Copyright (C) 2020 rexy712
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#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
|
|
|
|
#include "rexy.hpp"
|
|
|
|
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;
|
|
|
|
void invalidate(void);
|
|
|
|
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
|