ThreadPool.h

Go to the documentation of this file.

/*
 * Copyright (C) 2025 Andrew S. Rightenburg
 * Bash++: Bash with classes
 * SPDX-License-Identifier: GPL-3.0-or-later
 */
 
#pragma once
#include <atomic>
#include <condition_variable>
#include <functional>
#include <mutex>
#include <queue>
#include <vector>
#include <thread>
 
class ThreadPool {
    private:
        std::vector<std::thread> workers;
        std::queue<std::function<void()>> tasks;
        std::mutex queue_mutex;
        std::condition_variable condition;
        bool stop = false;
        std::atomic<bool> active{false}; // Whether the thread pool has accepted any tasks yet
    public:
        explicit ThreadPool(size_t threads = std::thread::hardware_concurrency());
        ~ThreadPool();
 
        ThreadPool(const ThreadPool& other) = delete; // Non-copyable
        ThreadPool& operator=(const ThreadPool& other) = delete;
        ThreadPool(ThreadPool&& other) noexcept = delete; // Non-movable
        ThreadPool& operator=(ThreadPool&& other) noexcept = delete;
 
        void enqueue(std::function<void()> task);
        void cleanup();
        size_t getThreadCount() const;
        bool isActive() const;
};