job_system.hpp

Go to the documentation of this file.

 
#ifndef __JOB_SYSTEM_H__
#define __JOB_SYSTEM_H__ 1
 
#pragma once
 
#include <optional>
#include <vector>
#include <thread>
#include <mutex>
#include <functional>
#include <queue>
#include <future>
#include <memory>
#include <raftel/global_macros.hpp>
 
namespace Raftel {
    class JobSystem
    {
    public:
      ~JobSystem();
      
      static std::unique_ptr<JobSystem> make();
      
 
      NO_COPYABLE_OR_MOVABLE(JobSystem)
      
       
      template<typename MyJobFunction, typename... Arguments>
      auto AddWork(MyJobFunction&& function, Arguments&&... params) -> std::future<decltype(function(params...))>;
      
      JobSystem();
 
    private:
      std::vector<std::thread> workers_;  
      std::mutex queue_mutex_; 
      std::condition_variable warner_job_avaliable_;
      std::queue<std::function<void()>> work_queue_;
      bool bStopThreads_; 
    };

    template<typename MyJobFunction, typename ...Arguments>
    inline auto JobSystem::AddWork(MyJobFunction&& function, Arguments && ...params) -> std::future<decltype(function(params...))>
    {
        auto packet = [function = std::move(function), ...params = std::move(params)](){ return function(params...); };
        std::lock_guard lock{queue_mutex_};
 
        using work_type = decltype(function(params...));
        auto return_type = std::make_shared<std::packaged_task<work_type()>>(packet);
 
        std::future<work_type> f = return_type->get_future();
        work_queue_.push([return_type](){(*return_type)();});
        warner_job_avaliable_.notify_one();
        return f;
    }
} // namespace Raftel
 
#endif // __JOB_SYSTEM_H__