TaskSystem: Implement shutdown

1 files changed, 4 insertions, 0 deletions

diff --git a/src/buildtool/multithreading/task_system.hpp b/src/buildtool/multithreading/task_system.hpp
index 888ba090..5dbd92a6 100644
--- a/src/buildtool/multithreading/task_system.hpp
+++ b/src/buildtool/multithreading/task_system.hpp
@@ -48,6 +48,9 @@ class TaskSystem {
         return thread_count_;
     }
 
+    // Initiate shutdown, skip execution of pending tasks
+    void Shutdown() noexcept;
+
     // Wait for all queues to become empty _and_ all tasks to finish.
     void Finish() noexcept;
 
@@ -57,6 +60,7 @@ class TaskSystem {
     std::vector<std::thread> threads_{};
     std::vector<NotificationQueue> queues_{};
     std::atomic<std::size_t> index_{0};
+    std::atomic<bool> shutdown_{};
     WaitableZeroCounter total_workload_{};
 
     static constexpr std::size_t kNumberOfAttempts = 5;