ThreadPool.h

Go to the documentation of this file.

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 * All rights reserved.
 *
 * This source code is licensed under both the BSD-style license (found in the
 * LICENSE file in the root directory of this source tree) and the GPLv2 (found
 * in the COPYING file in the root directory of this source tree).
 */
#pragma once
 
#include "utils/WorkQueue.h"
 
#include <cstddef>
#include <functional>
#include <thread>
#include <vector>
 
namespace pzstd {
class ThreadPool {
  std::vector<std::thread> threads_;
 
  WorkQueue<std::function<void()>> tasks_;
 
 public:
  explicit ThreadPool(std::size_t numThreads) {
    threads_.reserve(numThreads);
    for (std::size_t i = 0; i < numThreads; ++i) {
      threads_.emplace_back([this] {
        std::function<void()> task;
        while (tasks_.pop(task)) {
          task();
        }
      });
    }
  }
 
  ~ThreadPool() {
    tasks_.finish();
    for (auto& thread : threads_) {
      thread.join();
    }
  }
 
  void add(std::function<void()> task) {
    tasks_.push(std::move(task));
  }
};
}