ThreadPool.cpp

#include "ThreadPool.h"

ThreadPool::ThreadPool(size_t numThreads) : stop(false), tasksCompleted(0) {
    for (size_t i = 0; i < numThreads; ++i) {
        workers.emplace_back([this]() { this->workerThread(); });
    }
}

ThreadPool::~ThreadPool() {
    {
        unique_lock<mutex> lock(queueMutex);
        stop = true;
    }
    condition.notify_all();

    for (auto& worker : workers) {
        worker.join();
    }
}

void ThreadPool::addTask(const function<void()>& task) {
    unique_lock<mutex> lock(queueMutex);
    tasks.push(task);
    condition.notify_one();
}

void ThreadPool::waitForAllTasks() {
    unique_lock<mutex> lock(queueMutex);
    condition.wait(lock, [this]() { return tasks.empty() && tasksCompleted == workers.size(); });
}

void ThreadPool::workerThread() {
    while (true) {
        function<void()> task;
        {
            unique_lock<mutex> lock(queueMutex);
            condition.wait(lock, [this]() { return stop || !tasks.empty(); });

            if (stop && tasks.empty()) {
                condition.notify_all();
                return;
            }

            task = move(tasks.front());
            tasks.pop();
        }
        task();
        ++tasksCompleted; // 已经完成任务的数量
    }
}