在C++中使用線程池通常需要以下步驟:
<thread>
頭文件,用于創建線程。以下是一個簡單的C++線程池示例代碼:
#include <iostream>
#include <vector>
#include <thread>
#include <queue>
#include <functional>
class ThreadPool {
public:
ThreadPool(int numThreads) : stop(false) {
for (int i = 0; i < numThreads; ++i) {
threads.emplace_back(std::bind(&ThreadPool::worker, this));
}
}
template <class F, class... Args>
void addTask(F&& f, Args&&... args) {
tasks.emplace(std::bind(std::forward<F>(f), std::forward<Args>(args)...));
}
~ThreadPool() {
{
std::unique_lock<std::mutex> lock(mutex);
stop = true;
}
condition.notify_all();
for (auto& thread : threads) {
thread.join();
}
}
private:
std::vector<std::thread> threads;
std::queue<std::function<void()>> tasks;
std::mutex mutex;
std::condition_variable condition;
bool stop;
void worker() {
while (true) {
std::function<void()> task;
{
std::unique_lock<std::mutex> lock(mutex);
condition.wait(lock, [this]() { return stop || !tasks.empty(); });
if (stop && tasks.empty()) {
return;
}
task = std::move(tasks.front());
tasks.pop();
}
task();
}
}
};
void printHello() {
std::cout << "Hello" << std::endl;
}
int main() {
ThreadPool pool(4);
for (int i = 0; i < 8; ++i) {
pool.addTask(printHello);
}
return 0;
}
在上述示例中,線程池類 ThreadPool
包含了一個線程池容器 threads
和一個任務隊列 tasks
。通過調用 addTask
方法,可以將任務添加到任務隊列中。在 worker
方法中,線程池線程會不斷從任務隊列中取出任務并執行。在主函數中,創建了一個具有4個線程的線程池對象,并添加了8個打印Hello的任務。