worker_pool.h

Go to the documentation of this file.

#ifndef GRPPI_NATIVE_WORKER_POOL_H
#define GRPPI_NATIVE_WORKER_POOL_H

#include <thread>

namespace grppi {

class worker_pool {
  public:

    worker_pool(int num_threads) noexcept : num_threads_{num_threads} {}

    ~worker_pool() noexcept { this->wait(); }

    worker_pool(worker_pool &&) noexcept = default;
    worker_pool & operator=(worker_pool &&) noexcept = default;
    
    template <typename E, typename F, typename ... Args>
    void launch(const E & ex, F f, Args && ... args) {
      workers_.emplace_back([=,&ex]() {
        auto manager = ex.thread_manager();
        f(args...);
      });
    }

    template <typename E, typename F, typename ... Args>
    void launch_tasks(const E & ex, F && f, Args && ... args) {
      for (int i=0; i<num_threads_; ++i) {
        workers_.emplace_back([=,&ex]() {
          auto manager = ex.thread_manager();
          f(args...);
        });
      }
    }

    void wait() noexcept {
      for (auto && w : workers_) { w.join(); }
      workers_.clear();
    }

  private:
    const int num_threads_;
    std::vector<std::thread> workers_;
};

}

#endif