#include "ThreadPool.hpp"

#include <iostream>
#include "ThreadPool.hpp"

int main() {
    // Create a thread pool with 6 threads
    ThreadPool threadPool(6);

    // Define a task that calculates the sum of integers in the range [start, end) modulo (1e9 + 7)
    auto sumTask = [](std::uint64_t start, std::uint64_t end, std::atomic<int>& result) {
        int sum = 0;
        for (std::uint64_t i = start; i < end; ++i) {
            sum = (sum + i) % 1000000007;
        }
        result = (result + sum) % 1000000007;
    };

    // Queue 100 tasks to calculate the sum of integers in the range [0, 1e9) in parallel
    std::atomic<int> result(0);
    for (std::uint64_t i = 0; i < 100; ++i) {
        threadPool.queue([i, &sumTask, &result] { sumTask(i * 10000000, (i + 1) * 10000000, result); });
    }

    // Wait for all tasks to complete
    threadPool.waitForTasks();
    std::cout << "Sum of integers in the range [0, 1e9) modulo (1e9 + 7) = " << result << std::endl;

    // Stop processing tasks: no more tasks will be queued, but the threads will continue processing the remaining tasks
    // threadPool.stopProcessing();

    return 0;
}