溫馨提示×

C++線程池庫如何支持動態擴容

c++
小樊
146
2024-09-06 10:36:58
欄目: 編程語言

要讓C++線程池庫支持動態擴容,你需要實現以下功能:

  1. 在線程池類中維護一個線程數量的變量,例如num_threads。
  2. 提供一個方法(例如resize(size_t new_size)),用于改變線程池的大小。這個方法應該接受一個參數,表示新的線程池大小。
  3. resize方法中,根據新的線程池大小與當前線程池大小的關系,來決定是創建新的線程還是銷毀現有線程。
  4. 如果需要創建新的線程,可以使用std::thread創建線程并將其添加到線程池中。
  5. 如果需要銷毀現有線程,可以通過向線程發送一個特殊的任務(例如一個空任務)來通知線程退出,然后等待線程結束并將其從線程池中移除。

以下是一個簡單的C++線程池庫示例,支持動態擴容:

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

class ThreadPool {
public:
    ThreadPool(size_t num_threads) : num_threads_(num_threads), running_(true) {
        resize(num_threads);
    }

    ~ThreadPool() {
        stop();
    }

    void resize(size_t new_size) {
        std::unique_lock<std::mutex> lock(mutex_);
        if (new_size > num_threads_) {
            // 創建新的線程
            for (size_t i = num_threads_; i < new_size; ++i) {
                threads_.emplace_back(&ThreadPool::worker, this);
            }
        } else if (new_size < num_threads_) {
            // 銷毀現有線程
            for (size_t i = new_size; i < num_threads_; ++i) {
                tasks_.push([]() {}); // 添加空任務以通知線程退出
            }
        }
        num_threads_ = new_size;
    }

    void add_task(const std::function<void()>& task) {
        std::unique_lock<std::mutex> lock(mutex_);
        tasks_.push(task);
        cv_.notify_one();
    }

private:
    void worker() {
        while (running_) {
            std::function<void()> task;
            {
                std::unique_lock<std::mutex> lock(mutex_);
                cv_.wait(lock, [this]() { return !tasks_.empty() || !running_; });
                if (!running_) {
                    break;
                }
                task = std::move(tasks_.front());
                tasks_.pop();
            }
            task(); // 執行任務
        }
    }

    void stop() {
        {
            std::unique_lock<std::mutex> lock(mutex_);
            running_ = false;
        }
        cv_.notify_all(); // 喚醒所有線程以便它們可以退出
        for (auto& thread : threads_) {
            thread.join();
        }
    }

    size_t num_threads_;
    bool running_;
    std::vector<std::thread> threads_;
    std::queue<std::function<void()>> tasks_;
    std::mutex mutex_;
    std::condition_variable cv_;
};

這個示例中的ThreadPool類包含了一個resize方法,可以用于動態調整線程池的大小。當線程池大小減小時,會向線程池中的線程發送空任務以通知它們退出。當線程池大小增加時,會創建新的線程并將其添加到線程池中。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女