79 lines
2.0 KiB
C++
79 lines
2.0 KiB
C++
#pragma once
|
|
#include <condition_variable>
|
|
#include <functional>
|
|
#include <future>
|
|
#include <memory>
|
|
#include <mutex>
|
|
#include <queue>
|
|
#include <vector>
|
|
#include <thread>
|
|
|
|
namespace utils {
|
|
|
|
class ThreadPool {
|
|
public:
|
|
ThreadPool();
|
|
ThreadPool(std::uint32_t numThreads);
|
|
~ThreadPool();
|
|
|
|
void init(std::uint32_t numThreads);
|
|
void terminate();
|
|
|
|
void respawnWorker(std::uint32_t numThreads);
|
|
|
|
template <typename _Callable, typename... _Args>
|
|
requires(
|
|
!std::is_void_v<std::invoke_result_t<_Callable, ThreadPool*, _Args...>>)
|
|
int enqueueJob(_Callable&& __job,
|
|
std::invoke_result_t<_Callable, _Args...>& retVal,
|
|
_Args&&... __args) {
|
|
if (terminate_) {
|
|
spdlog::warn("Cannot run jobs on threads that terminating...");
|
|
return -1;
|
|
}
|
|
|
|
std::lock_guard<std::mutex> lock(jobQueueMutex);
|
|
auto boundFunc = [this, &retVal, __job,
|
|
... __args = std::move(__args)]() mutable {
|
|
retVal = __job(this, std::move(__args)...);
|
|
};
|
|
auto task = std::packaged_task<void()>(std::move(boundFunc));
|
|
jobs_.push(std::move(task));
|
|
jobQueueCV_.notify_one();
|
|
|
|
return 0;
|
|
}
|
|
template <typename _Callable, typename... _Args>
|
|
requires std::is_void_v<
|
|
std::invoke_result_t<_Callable, ThreadPool*, _Args...>>
|
|
int enqueueJob(_Callable&& __job, _Args&&... __args) {
|
|
if (terminate_) {
|
|
spdlog::warn("Cannot run jobs on threads that terminating...");
|
|
return -1;
|
|
}
|
|
|
|
std::lock_guard<std::mutex> lock(jobQueueMutex);
|
|
auto boundFunc = [this, __job, ... __args = std::move(__args)]() mutable {
|
|
__job(this, std::move(__args)...);
|
|
};
|
|
auto task = std::packaged_task<void()>(std::move(boundFunc));
|
|
jobs_.push(std::move(task));
|
|
jobQueueCV_.notify_one();
|
|
|
|
return 0;
|
|
}
|
|
|
|
int threadCount = 0;
|
|
|
|
private:
|
|
void* Worker();
|
|
|
|
std::condition_variable jobQueueCV_;
|
|
std::mutex jobQueueMutex;
|
|
std::queue<std::packaged_task<void()>> jobs_;
|
|
std::vector<std::thread> workers_;
|
|
bool terminate_ = false;
|
|
};
|
|
|
|
} // namespace Chattr
|