可以帮我写一个线程池吗,用C++
时间: 2023-07-20 19:26:53 浏览: 45
当然可以,以下是一个简单的 C++ 线程池实现:
```cpp
#include <iostream>
#include <vector>
#include <queue>
#include <thread>
#include <mutex>
#include <condition_variable>
class ThreadPool {
public:
ThreadPool(size_t size) : stop(false) {
for (size_t i = 0; i < size; i++) {
threads.emplace_back(std::thread(&ThreadPool::worker, this));
}
}
template <typename F>
void enqueue(F&& task) {
{
std::unique_lock<std::mutex> lock(queue_mutex);
tasks.push(std::forward<F>(task));
}
condition.notify_one();
}
~ThreadPool() {
{
std::unique_lock<std::mutex> lock(queue_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 queue_mutex;
std::condition_variable condition;
bool stop;
void worker() {
while (true) {
std::function<void()> task;
{
std::unique_lock<std::mutex> lock(queue_mutex);
condition.wait(lock, [this]() { return stop || !tasks.empty(); });
if (stop && tasks.empty()) {
return;
}
task = std::move(tasks.front());
tasks.pop();
}
task();
}
}
};
```
使用方法:
```cpp
ThreadPool pool(4);
pool.enqueue([]() {
std::cout << "Task 1" << std::endl;
});
pool.enqueue([]() {
std::cout << "Task 2" << std::endl;
});
```
这个线程池的实现比较简单,主要包含一个任务队列和一组工作线程。每个工作线程会从任务队列中取出一个任务执行,当队列为空时,则等待新的任务到来。同时,线程池也实现了安全的停止操作,即等待队列中的所有任务都执行完毕后,再停止所有工作线程。