diff options
Diffstat (limited to 'gn/util/worker_pool.h')
-rw-r--r-- | gn/util/worker_pool.h | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/gn/util/worker_pool.h b/gn/util/worker_pool.h new file mode 100644 index 00000000000..d0616449097 --- /dev/null +++ b/gn/util/worker_pool.h @@ -0,0 +1,37 @@ +// Copyright 2018 The Chromium Authors. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. + +#ifndef UTIL_WORKER_POOL_H_ +#define UTIL_WORKER_POOL_H_ + +#include <condition_variable> +#include <mutex> +#include <queue> +#include <thread> + +#include "base/logging.h" +#include "base/macros.h" +#include "util/task.h" + +class WorkerPool { + public: + WorkerPool(); + WorkerPool(size_t thread_count); + ~WorkerPool(); + + void PostTask(Task work); + + private: + void Worker(); + + std::vector<std::thread> threads_; + std::queue<base::OnceClosure> task_queue_; + std::mutex queue_mutex_; + std::condition_variable_any pool_notifier_; + bool should_stop_processing_; + + DISALLOW_COPY_AND_ASSIGN(WorkerPool); +}; + +#endif // UTIL_WORKER_POOL_H_ |