First commit with a tested version of the thread pool
This commit is contained in:
1
include/ThreadPool.cpp
Normal file
1
include/ThreadPool.cpp
Normal file
@@ -0,0 +1 @@
|
||||
#include "ThreadPool.h"
|
98
include/ThreadPool.h
Normal file
98
include/ThreadPool.h
Normal file
@@ -0,0 +1,98 @@
|
||||
#ifndef THREADPOOL_H
|
||||
#define THREADPOOL_H
|
||||
|
||||
#include <atomic>
|
||||
#include <deque>
|
||||
#include <functional>
|
||||
#include <mutex>
|
||||
#include <thread>
|
||||
#include <vector>
|
||||
|
||||
class ThreadPool {
|
||||
int numOfThreads;
|
||||
std::deque<std::function<void()>> _tasks;
|
||||
std::mutex _queueLock;
|
||||
std::atomic<bool> stopped{false};
|
||||
bool startedAlready = false;
|
||||
std::condition_variable _queueCond;
|
||||
std::condition_variable _threadFinishedCond;
|
||||
std::atomic_int numOfThreadsRunning{0};
|
||||
std::vector<std::thread> _threads;
|
||||
|
||||
protected:
|
||||
inline void threadWorker();
|
||||
inline void joinAll();
|
||||
|
||||
public:
|
||||
inline ThreadPool();
|
||||
inline ~ThreadPool();
|
||||
inline void push(const std::function<void()> &task);
|
||||
inline void
|
||||
start(const int NumOfThreads = std::thread::hardware_concurrency());
|
||||
inline void finish();
|
||||
};
|
||||
|
||||
void ThreadPool::threadWorker() {
|
||||
while (true) {
|
||||
std::unique_lock<decltype(_queueLock)> lg(_queueLock);
|
||||
while (_tasks.empty() && !stopped.load()) {
|
||||
_queueCond.wait(lg);
|
||||
}
|
||||
if (!_tasks.empty()) {
|
||||
auto theTask = _tasks[0];
|
||||
_tasks.pop_front();
|
||||
lg.unlock();
|
||||
theTask();
|
||||
lg.lock();
|
||||
}
|
||||
if (_tasks.empty() && stopped.load()) {
|
||||
numOfThreadsRunning--;
|
||||
_threadFinishedCond.notify_one();
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void ThreadPool::joinAll() {
|
||||
for (long i = 0; i < numOfThreads; i++) {
|
||||
if (_threads[i].joinable())
|
||||
_threads[i].join();
|
||||
}
|
||||
}
|
||||
|
||||
ThreadPool::ThreadPool() {}
|
||||
|
||||
ThreadPool::~ThreadPool() { this->finish(); }
|
||||
|
||||
void ThreadPool::push(const std::function<void()> &task) {
|
||||
std::unique_lock<decltype(_queueLock)> lg(_queueLock);
|
||||
_tasks.push_back(task);
|
||||
lg.unlock();
|
||||
_queueCond.notify_one();
|
||||
}
|
||||
|
||||
void ThreadPool::start(const int NumOfThreads) {
|
||||
if (!startedAlready)
|
||||
startedAlready = true;
|
||||
else
|
||||
throw std::logic_error("You cannot start the thread pool multiple times.");
|
||||
numOfThreads = NumOfThreads;
|
||||
_threads.reserve(numOfThreads);
|
||||
for (long i = 0; i < numOfThreads; i++) {
|
||||
_threads.push_back(std::thread(std::bind(&ThreadPool::threadWorker, this)));
|
||||
numOfThreadsRunning++;
|
||||
}
|
||||
}
|
||||
|
||||
void ThreadPool::finish() {
|
||||
std::unique_lock<decltype(_queueLock)> lg(_queueLock);
|
||||
stopped.store(true);
|
||||
_queueCond.notify_all();
|
||||
while (numOfThreadsRunning > 0) {
|
||||
_threadFinishedCond.wait(lg);
|
||||
}
|
||||
lg.unlock();
|
||||
joinAll();
|
||||
}
|
||||
|
||||
#endif // THREADPOOL_H
|
Reference in New Issue
Block a user