threadpool.hpp

Go to the documentation of this file.

#pragma once
 
#include <atomic>
#include <condition_variable>
#include <functional>
#include <mutex>
#include <queue>
#include <thread>
#include <vector>
 
#include "exkit/scheduler.hpp"
 
class ThreadPool : public IScheduler {
public:
    ThreadPool(size_t thread_count = std::thread::hardware_concurrency());
 
    ~ThreadPool();
 
    ThreadPool(const ThreadPool&) = delete;
 
    ThreadPool(ThreadPool&&) = delete;
 
    ThreadPool& operator=(const ThreadPool&) = delete;
 
    ThreadPool& operator=(ThreadPool&&) = delete;
 
    void shutdown() override;
 
    void schedule(std::function<void()> task) override;
 
    void wait() override;
 
private:
    void _worker_loop();
 
private:
    std::queue<std::function<void()>> _tasks;
    std::condition_variable _condition;
    std::vector<std::thread> _threads;
    std::atomic<size_t> _active_tasks;
    std::mutex _queue_mutex;
    std::atomic<bool> _stop;
};