#ifndef THREADPOOL_H #define THREADPOOL_H #include #include #include #include #include #include #include "pthread.h" #include "LockGuard_pthread.h" /** * @brief The ThreadPool class * A basic thread pool implementation using pthread * * Call with the constructor ThreadPool::ThreadPool(N) to get N threads. * The default constructor will detect the number of available cores/core threads and use it. * Constructor throws std::runtime_error() on failure to spawn processes. * * * Add tasks using ThreadPool::push_task(function, args). More info at the function description. * * Use ThreadPool::finish() to wait for all tasks to process and join all threads. * * * * */ class ThreadPool_pthread { pthread_cond_t task_queue_cond; //signaled when a new task is pushed pthread_cond_t thread_finished_cond; //signaled when a thread finishes and no more is there to do pthread_mutex_t mutex; bool no_more_to_push; typedef void*(*FunctionType)(void*); //queue of tasks as an std::queue of std::pair of functions and args std::queue > tasks_queue; void launch_threads(); int num_of_threads; //stores the number of threads requested std::vector threads_objects; friend void* _thread_worker(void*); bool joined; //prevents rejoining twice, true when all threads are joined long threads_running; //keeps track of threads that are executing tasks void join_all(); public: ThreadPool_pthread(int threads_count); virtual ~ThreadPool_pthread(); /** * @brief push_task * Adds a task to the processing queue * * Results are passed through parameters too. No return value is considered. * * @param function * Function pointer to the function to be executed. * The function is of type void*(void*) * All parameters should be passed through a single void pointer * * @param params * Parameters of the function as void*. */ void push_task(FunctionType function, void* params); /** * @brief finish * Waits for all threads to finish and joins them. */ void finish(); }; template std::string _to_string(T const& value) { std::stringstream sstr; sstr << value; return sstr.str(); } #endif // THREADPOOL_H