diff --git a/src/EC/ThreadPool.hpp b/src/EC/ThreadPool.hpp index 003a617..b85c698 100644 --- a/src/EC/ThreadPool.hpp +++ b/src/EC/ThreadPool.hpp @@ -28,8 +28,9 @@ using ThreadStackType = std::vector>; using ThreadStacksType = std::deque; using ThreadStacksMutexesT = std::deque; using ThreadCountersT = std::deque; +using PtrsHoldT = std::deque; using PointersT = - std::tuple; + std::tuple; } // namespace Internal /*! @@ -147,7 +148,7 @@ class ThreadPool { } else { sequentiallyRunTasks(); } - return {nullptr, nullptr, nullptr}; + return {nullptr, nullptr, nullptr, nullptr}; } /*! @@ -184,6 +185,7 @@ class ThreadPool { { std::lock_guard lock(*std::get<1>(pointers)); if (std::get<0>(pointers)->empty()) { + std::get<3>(pointers)->store(false); break; } } @@ -219,6 +221,7 @@ class ThreadPool { Internal::TPQueueType fnQueue; std::mutex queueMutex; Internal::ThreadCountersT threadCounters; + Internal::PtrsHoldT ptrsHoldBools; std::mutex dequesMutex; void sequentiallyRunTasks() { @@ -253,9 +256,12 @@ class ThreadPool { erased = false; { std::lock_guard lock(threadStackMutexes.front()); - if (threadStacks.front().empty()) { + if (ptrsHoldBools.front().load()) { + break; + } else if (threadStacks.front().empty()) { threadStacks.pop_front(); threadCounters.pop_front(); + ptrsHoldBools.pop_front(); erased = true; } } @@ -265,7 +271,7 @@ class ThreadPool { break; } } while (!threadStacks.empty() && !threadStackMutexes.empty() && - !threadCounters.empty()); + !threadCounters.empty() && !ptrsHoldBools.empty()); } Internal::PointersT newStackEntry() { @@ -274,9 +280,11 @@ class ThreadPool { threadStackMutexes.emplace_back(); threadCounters.emplace_back(); threadCounters.back().store(0); + ptrsHoldBools.emplace_back(); + ptrsHoldBools.back().store(true); return {&threadStacks.back(), &threadStackMutexes.back(), - &threadCounters.back()}; + &threadCounters.back(), &ptrsHoldBools.back()}; } };