samerafach/StdThreadPool
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Removed atomicity (not necessary) and renamed some variables.

Browse Source
This commit is contained in:
Samer Afach 2017-06-06 19:31:23 +02:00
parent 0e45acf8ce
commit 375d5a38b1
1 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
include/ThreadPool.h
View File

@ -1,7 +1,6 @@
#ifndef THREADPOOL_H
#define THREADPOOL_H
#include <atomic>
#include <deque>
#include <functional>
#include <mutex>
@ -12,16 +11,16 @@ class ThreadPool {
long numOfThreads;
std::deque<std::function<void()>> _tasks;
std::mutex _queueLock;
std::atomic<bool> stopped{false};
bool startedAlready = false;
bool conclude_work = false;
bool started_already = false;
std::condition_variable _queueCond;
std::condition_variable _threadFinishedCond;
long numOfThreadsRunning = 0;
long num_of_threads_running = 0;
std::vector<std::thread> _threads;
protected:
inline void threadWorker();
inline void joinAll();
inline void thread_worker();
inline void join_all();
public:
inline ThreadPool();
@ -32,10 +31,10 @@ public:
inline void finish();
};
void ThreadPool::threadWorker() {
void ThreadPool::thread_worker() {
while (true) {
std::unique_lock<decltype(_queueLock)> lg(_queueLock);
while (_tasks.empty() && !stopped.load()) {
while (_tasks.empty() && !conclude_work) {
_queueCond.wait(lg);
}
if (!_tasks.empty()) {
@ -45,15 +44,15 @@ void ThreadPool::threadWorker() {
theTask();
lg.lock();
}
if (_tasks.empty() && stopped.load()) {
numOfThreadsRunning--;
if (_tasks.empty() && conclude_work) {
num_of_threads_running--;
_threadFinishedCond.notify_one();
break;
}
}
}
void ThreadPool::joinAll() {
void ThreadPool::join_all() {
for (long i = 0; i < numOfThreads; i++) {
if (_threads[i].joinable())
_threads[i].join();
@ -71,27 +70,28 @@ void ThreadPool::push(const std::function<void()> &task) {
}
void ThreadPool::start(const long NumOfThreads) {
if (!startedAlready)
startedAlready = true;
if (!started_already)
started_already = 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++;
_threads.push_back(
std::thread(std::bind(&ThreadPool::thread_worker, this)));
num_of_threads_running++;
}
}
void ThreadPool::finish() {
std::unique_lock<decltype(_queueLock)> lg(_queueLock);
stopped.store(true);
conclude_work = true;
_queueCond.notify_all();
while (numOfThreadsRunning > 0) {
while (num_of_threads_running > 0) {
_threadFinishedCond.wait(lg);
}
lg.unlock();
joinAll();
join_all();
}
#endif // THREADPOOL_H