ThreadPool.hpp

Go to the documentation of this file.

#ifndef THREADPOOL_HPP_
#define THREADPOOL_HPP_

#include <atomic>
#include <cstdlib>
#include <vector>
#include <thread>

#include "kaa/KaaThread.hpp"
#include "kaa/utils/KaaTimer.hpp"
#include "kaa/utils/IThreadPool.hpp"

namespace kaa {

class ThreadPool : public IThreadPool {
    friend class Worker;

public:
    ThreadPool(std::size_t workerCount = DEFAULT_WORKER_NUMBER);
    ~ThreadPool();

    virtual void add(const ThreadPoolTask& task);

    virtual void awaitTermination(std::size_t seconds);

    virtual void shutdown();
    virtual void shutdownNow();

public:
    static const std::size_t DEFAULT_WORKER_NUMBER = 1;

private:
    void start();
    void stop(bool force);

private:
    bool isRun_ = true;
    bool isPendingShutdown_ = false;

    std::list<std::thread>    workers_;
    std::size_t               workerCount_ = 0;

    std::list<ThreadPoolTask>    tasks_;

    KAA_MUTEX_DECLARE(threadPoolGuard_);
    KAA_CONDITION_VARIABLE    onNewTask_;

    std::unique_ptr<KaaTimer<void()>>    shutdownTimer_;
};

} /* namespace kaa */

#endif /* THREADPOOL_HPP_ */