要讓C++線程池庫支持動態擴容,你需要實現以下功能:
num_threads。resize(size_t new_size)),用于改變線程池的大小。這個方法應該接受一個參數,表示新的線程池大小。resize方法中,根據新的線程池大小與當前線程池大小的關系,來決定是創建新的線程還是銷毀現有線程。std::thread創建線程并將其添加到線程池中。以下是一個簡單的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方法,可以用于動態調整線程池的大小。當線程池大小減小時,會向線程池中的線程發送空任務以通知它們退出。當線程池大小增加時,會創建新的線程并將其添加到線程池中。