thread_pool_context.h

Go to the documentation of this file.

 
#pragma once
 
#include <condition_variable>
#include <memory>
#include <mutex>
#include <queue>
#include <thread>
#include <utility>
 
#include <boost/asio/thread_pool.hpp>
#include <boost/asio/ts/executor.hpp>
 
 
namespace knp::backends::cpu_executors
{
 
class ThreadPoolContext
{
public:
    explicit ThreadPoolContext(size_t num_threads = std::thread::hardware_concurrency());
 
    ~ThreadPoolContext();
 
    // Move and assignment are implicitly deleted because of mutex.
 
 
private:
    enum class Usage
    {
        FINISHED,
        READY,
        WORKING,
        STOPPING
    };
 
    struct Function
    {
        std::shared_ptr<size_t> work_count_;
        // cppcheck-suppress unusedStructMember
        void (*execute_)(std::shared_ptr<Function> &p);
    };
 
    void do_work_started(const std::shared_ptr<size_t> &task_count) noexcept;
 
    void do_work_finished(const std::shared_ptr<size_t> &task_count);
 
    void execute(std::unique_lock<std::mutex> &lock, std::shared_ptr<Function> &work);
 
    bool execute_next(std::unique_lock<std::mutex> &lock);
 
    void stop();
 
    void post(const std::shared_ptr<Function> &task, const std::shared_ptr<size_t> &task_count);
 
private:
    friend class ThreadPoolExecutor;
    std::mutex mutex_;
    std::condition_variable condition_;
    Usage usage_state_ = Usage::READY;
    // cppcheck-suppress unusedStructMember
    std::queue<std::shared_ptr<Function>> work_queue_;
    boost::asio::thread_pool pool_;
};
 
}  // namespace knp::backends::cpu_executors