如何在C++线程类中实现任务分配

在 C++ 中，可以使用` 库来创建线程并分配任务

#include<iostream>
#include<vector>
#include<thread>
#include <mutex>

class Task {
public:
    virtual void execute() = 0;
};

class PrintTask : public Task {
public:
    PrintTask(const std::string& message) : message_(message) {}

    void execute() override {
        std::cout << "Thread "<< std::this_thread::get_id() << ": "<< message_<< std::endl;
    }

private:
    std::string message_;
};

class ThreadPool {
public:
    ThreadPool(size_t num_threads) {
        for (size_t i = 0; i < num_threads; ++i) {
            threads_.emplace_back(&ThreadPool::worker, this);
        }
    }

    ~ThreadPool() {
        {
            std::unique_lock<std::mutex> lock(queue_mutex_);
            stop_ = true;
        }
        condition_.notify_all();
        for (auto& thread : threads_) {
            thread.join();
        }
    }

    void add_task(Task* task) {
        std::unique_lock<std::mutex> lock(queue_mutex_);
        tasks_.push(task);
        condition_.notify_one();
    }

private:
    void worker() {
        while (true) {
            Task* task = nullptr;
            {
                std::unique_lock<std::mutex> lock(queue_mutex_);
                condition_.wait(lock, [this] { return !tasks_.empty() || stop_; });
                if (stop_) {
                    break;
                }
                task = tasks_.front();
                tasks_.pop();
            }
            task->execute();
        }
    }

    std::vector<std::thread> threads_;
    std::queue<Task*> tasks_;
    std::mutex queue_mutex_;
    std::condition_variable condition_;
    bool stop_ = false;
};

int main() {
    ThreadPool pool(4); // 创建一个包含 4 个线程的线程池

    // 添加任务到线程池
    for (int i = 0; i < 10; ++i) {
        pool.add_task(new PrintTask("Hello from task " + std::to_string(i)));
    }

    // 等待所有任务完成
    // 注意：这里没有删除任务对象，实际应用中需要考虑内存管理
    std::this_thread::sleep_for(std::chrono::seconds(2));

    return 0;
}

这个示例中，我们定义了一个 Task 基类和一个派生自 Task 的 PrintTask 类。ThreadPool 类负责管理线程和任务队列。add_task 方法用于向线程池添加新任务。线程池中的每个线程都会从任务队列中获取任务并执行它。

请注意，这个示例没有处理任务对象的内存管理。在实际应用中，你需要考虑如何安全地删除任务对象以避免内存泄漏。可以考虑使用智能指针（如 std::shared_ptr 或 std::unique_ptr）来管理任务对象的生命周期。