溫馨提示×

C++ Task和線程池的配合使用

c++
小樊
94
2024-07-21 12:12:04
欄目: 編程語言

在C++中，可以使用Task和線程池來實現并發(fā)執(zhí)行任務的功能。Task代表一個可執(zhí)行的任務，可以是一個函數或者一個類的成員函數。線程池則是一組線程，用來執(zhí)行任務隊列中的任務。

以下是一個簡單的示例代碼，演示如何在C++中使用Task和線程池：

#include <iostream>
#include <vector>
#include <thread>
#include <queue>
#include <functional>
#include <mutex>
#include <condition_variable>

class Task {
public:
    Task(std::function<void()> func) : m_func(func) {}

    void execute() {
        m_func();
    }

private:
    std::function<void()> m_func;
};

class ThreadPool {
public:
    ThreadPool(int numThreads) {
        for (int i = 0; i < numThreads; ++i) {
            m_threads.emplace_back([this] {
                while (true) {
                    Task* task = nullptr;
                    {
                        std::unique_lock<std::mutex> lock(m_mutex);
                        m_condition.wait(lock, [this] { return !m_tasks.empty(); });
                        task = m_tasks.front();
                        m_tasks.pop();
                    }
                    task->execute();
                    delete task;
                }
            });
        }
    }

    ~ThreadPool() {
        for (std::thread& thread : m_threads) {
            thread.join();
        }
    }

    template<class F, class... Args>
    void addTask(F&& func, Args&&... args) {
        auto task = new Task(std::bind(std::forward<F>(func), std::forward<Args>(args)...));
        {
            std::lock_guard<std::mutex> lock(m_mutex);
            m_tasks.push(task);
        }
        m_condition.notify_one();
    }

private:
    std::vector<std::thread> m_threads;
    std::queue<Task*> m_tasks;
    std::mutex m_mutex;
    std::condition_variable m_condition;
};

void printNumber(int number) {
    std::cout << "Number: " << number << std::endl;
}

int main() {
    ThreadPool threadPool(4);

    for (int i = 0; i < 10; ++i) {
        threadPool.addTask(printNumber, i);
    }

    std::this_thread::sleep_for(std::chrono::seconds(1));

    return 0;
}

在這個示例中，我們定義了一個Task類來表示一個任務，然后定義了一個ThreadPool類來管理線程池。在主函數中，我們創(chuàng)建一個ThreadPool對象，并向線程池中提交了10個任務，每個任務打印一個數字。最后，我們讓主線程休眠一秒鐘，等待所有任務執(zhí)行完畢。

這樣，我們就實現了在C++中使用Task和線程池配合使用的功能。通過線程池可以有效地管理線程的生命周期，并實現多線程并發(fā)執(zhí)行任務的功能。

0 踩