thread_pool.h

Go to the documentation of this file.

 
#pragma once
#include <memory>
 
#include "thread_pool_context.h"
#include "thread_pool_executor.h"
 
 
namespace knp::backends::cpu_executors
{
class ThreadPool
{
public:
    explicit ThreadPool(size_t num_threads)
        : context_(std::make_unique<ThreadPoolContext>(num_threads)), executor_(*context_)
    {
    }
 
    template <class Func, typename... Args>
    void post(Func func, Args... args)
    {
        boost::asio::post(executor_, std::bind(func, args...));
    }
 
    void join() { executor_.join(); }
 
private:
    // Do not change the order of declarations.
    std::unique_ptr<ThreadPoolContext> context_;
    ThreadPoolExecutor executor_;
};
 
}  // namespace knp::backends::cpu_executors