threadpool.cpp

#include "threadpool.h"

ThreadPool::ThreadPool(size_t threads) : stop(false) {
    for(size_t i = 0; i < threads; ++i) {
        networks.emplace_back(std::make_unique<NeuralNetwork>());
        network_busy.push_back(false);
        workers.emplace_back(
            [this, i] {
                for(;;) {
                    std::function<void()> task;
                    {
                        std::unique_lock<std::mutex> lock(this->queue_mutex);
                        this->condition.wait(lock,
                            [this]{ return this->stop || !this->tasks.empty(); });
                        if(this->stop && this->tasks.empty())
                            return;
                        task = std::move(this->tasks.front());
                        this->tasks.pop();
                    }
                    task();
                }
            }
        );
    }
}

ThreadPool::~ThreadPool() {
    {
        std::unique_lock<std::mutex> lock(queue_mutex);
        stop = true;
    }
    condition.notify_all();
    for(std::thread &worker: workers)
        worker.join();
}

NeuralNetwork* ThreadPool::getAvailableNetwork() {
    std::lock_guard<std::mutex> lock(network_mutex);
    for (size_t i = 0; i < networks.size(); ++i) {
        if (!network_busy[i]) {
            network_busy[i] = true;
            return networks[i].get();
        }
    }
    return nullptr; // Все сети заняты
}