data_thread_pool.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <condition_variable>
#include <functional>
#include <future>
#include <memory>
#include <mutex>
#include <queue>
#include <stdexcept>
#include <thread>
#include <type_traits>
#include <utility>
#include <vector>
 
namespace pulsatrix {
 
class DataThreadPool {
public:
    explicit DataThreadPool(unsigned num_threads);
    ~DataThreadPool();
 
    DataThreadPool(const DataThreadPool&) = delete;
    DataThreadPool& operator=(const DataThreadPool&) = delete;
 
    template <typename F>
    auto submit(F&& task) -> std::future<std::invoke_result_t<F>> {
        using ReturnType = std::invoke_result_t<F>;
        auto packaged = std::make_shared<std::packaged_task<ReturnType()>>(std::forward<F>(task));
        std::future<ReturnType> result = packaged->get_future();
        {
            std::lock_guard<std::mutex> lock(mutex_);
            if (stop_) {
                throw std::runtime_error("DataThreadPool::submit: pool is stopped");
            }
            tasks_.emplace([packaged]() { (*packaged)(); });
        }
        cv_.notify_one();
        return result;
    }
 
private:
    std::vector<std::thread> workers_;
    std::queue<std::function<void()>> tasks_;
    std::mutex mutex_;
    std::condition_variable cv_;
    bool stop_ = false;
};
 
}  // namespace pulsatrix