fixup bug as description in mtrebi#39

muyuuuu · Aug 14, 2021 · da53a3c · da53a3c
1 parent dac6bdd
commit da53a3c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/include/ThreadPool.h b/include/ThreadPool.h
@@ -36,6 +36,18 @@ class ThreadPool {
           func();
         }
       }
+
+      // If the task queue is not empty, continue obtain task from task queue, 
+      // the multithread continues execution until the queue is empty
+      while (!m_pool->m_queue.empty()) {
+        {
+          std::unique_lock<std::mutex> lock(m_pool->m_conditional_mutex);
+          dequeued = m_pool->m_queue.dequeue(func);
+          if (dequeued) {
+            func();
+          }
+        }
+      }
     }
   };