thread_pool.cpp

Go to the documentation of this file.

// SPDX-License-Identifier: GPL-2.0-only
//
// Copyright (c) 2024-present FERS Contributors (see AUTHORS.md).
//
// See the GNU GPLv2 LICENSE file in the FERS project root for more information.
 
/**
 * @file thread_pool.cpp
 * @brief A simple thread pool implementation.
 */
 
#include "thread_pool.h"
 
#include "logging.h"
 
namespace pool
{
    ThreadPool::ThreadPool(const unsigned numThreads)
    {
        _workers.reserve(numThreads);
        for (unsigned i = 0; i < numThreads; ++i)
        {
            _workers.emplace_back(
                [this]
                {
                    while (true)
                    {
                        Task task;
                        {
                            std::unique_lock lock(_queue_mutex);
                            _condition.wait(lock, [this] { return _stop || !_tasks.empty(); });
                            if (_stop && _tasks.empty())
                            {
                                return;
                            }
                            task = std::move(_tasks.front());
                            _tasks.pop();
                        }
                        try
                        {
                            task();
                        }
                        catch (const std::exception& e)
                        {
                            LOG(logging::Level::ERROR, "Exception in thread pool: {}", e.what());
                        }
 
                        {
                            std::unique_lock lock(_queue_mutex);
                            --_pending_tasks;
                            if (_pending_tasks == 0)
                            {
                                _done_condition.notify_all();
                            }
                        }
                    }
                });
        }
    }
 
    ThreadPool::~ThreadPool()
    {
        {
            std::unique_lock lock(_queue_mutex);
            _stop = true;
            _condition.notify_all();
        }
        for (std::thread& worker : _workers)
        {
            worker.join();
        }
    }
 
    unsigned ThreadPool::getAvailableThreads()
    {
        std::unique_lock lock(_queue_mutex);
        const unsigned active_threads = _pending_tasks;
        const unsigned total_threads = _workers.size();
        return total_threads > active_threads ? total_threads - active_threads : 0;
    }
}