写一个使用线程池使用实例
时间: 2023-09-05 17:13:23 浏览: 107
好的,下面是使用线程池的实例:
首先需要引入线程池的头文件:
```
#include <thread>
#include <vector>
#include <functional>
#include <queue>
```
接下来定义一个类 ThreadPool,它包含一个任务队列和一组工作线程:
```
class ThreadPool {
public:
ThreadPool(size_t numThreads) {
for (size_t i = 0; i < numThreads; ++i) {
workers_.emplace_back([this] {
for (;;) {
std::function<void()> task;
{
std::unique_lock<std::mutex> lock{ tasks_mutex_ };
tasks_cv_.wait(lock, [this]() {
return stop_ || !tasks_.empty();
});
if (stop_ && tasks_.empty())
return;
task = std::move(tasks_.front());
tasks_.pop();
}
task();
}
});
}
}
template<typename F, typename... Args>
auto enqueue(F&& f, Args&&... args)
-> std::future<typename std::result_of<F(Args...)>::type> {
using return_type = typename std::result_of<F(Args...)>::type;
auto task = std::make_shared<std::packaged_task<return_type()>>(
std::bind(std::forward<F>(f), std::forward<Args>(args)...)
);
std::future<return_type> res = task->get_future();
{
std::unique_lock<std::mutex> lock{ tasks_mutex_ };
if (stop_)
throw std::runtime_error("enqueue on stopped ThreadPool");
tasks_.emplace([task]() { (*task)(); });
}
tasks_cv_.notify_one();
return res;
}
~ThreadPool() {
{
std::unique_lock<std::mutex> lock{ tasks_mutex_ };
stop_ = true;
}
tasks_cv_.notify_all();
for (std::thread &worker : workers_)
worker.join();
}
private:
std::vector<std::thread> workers_;
std::queue<std::function<void()>> tasks_;
std::mutex tasks_mutex_;
std::condition_variable tasks_cv_;
bool stop_ = false;
};
```
这个线程池实现了一个 enqueue 函数(模板函数),用于将任务添加到任务队列中。下面是一个使用线程池的示例程序:
```
#include <iostream>
#include <chrono>
#include <string>
void print(const std::string& str, int num) {
for (int i = 0; i < num; ++i) {
std::cout << str << std::endl;
std::this_thread::sleep_for(std::chrono::milliseconds(100));
}
}
int main() {
ThreadPool pool{ 2 };
auto res1 = pool.enqueue(print, "hello", 10);
auto res2 = pool.enqueue(print, "world", 10);
res1.get();
res2.get();
return 0;
}
```
本示例中创建了一个 ThreadPool 对象,然后将两个打印任务添加到任务队列中。这两个任务将交替打印 "hello" 和 "world" 字符串,每行打印一次,每次打印间隔 100 毫秒。最后,程序等待这两个任务执行完成,释放线程池资源。
阅读全文