#include <vector> #include <queue> #include <thread> #include <mutex> #include <condition_variable> #include <functional> #include <future> class ThreadPool { public: ThreadPool(size_t threads) : stop(false) { for (size_t i = 0; i < threads; ++i) { workers.emplace_back([this] { for (;;) { std::function<void()> task; { std::unique_lockstd::mutex lock(this->queue_mutex); this->condition.wait(lock, [this] { return this->stop || !this->tasks.empty(); }); if (this->stop && this->tasks.empty()) return; task = std::move(this->tasks.front()); this->tasks.pop(); } task(); } }); } } ~ThreadPool() { { std::unique_lockstd::mutex lock(queue_mutex); stop = true; } condition.notify_all(); for (std::thread &worker : workers) worker.join(); } template<class F, class... 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_lockstd::mutex lock(queue_mutex); if (stop) throw std::runtime_error("enqueue on stopped ThreadPool"); tasks.emplace([task] { (*task)(); }); } condition.notify_one(); return res; } private: std::vectorstd::thread workers; std::queue<std::function<void()>> tasks; std::mutex queue_mutex; std::condition_variable condition; bool stop; }; 怎么获取结果

可以使用`std::future`来获取线程池中任务的返回值。在`enqueue`函数中，任务被包装在`std::packaged_task`中，这个包装器可以将一个函数包装成一个可调用对象，并且可以使用`std::future`来获取函数的返回值。

在`enqueue`函数中，我们使用`std::make_shared`创建了一个`std::packaged_task`，并将要执行的任务`f`和其参数`args`绑定在一起。然后，我们将这个`std::packaged_task`封装在一个`std::shared_ptr`中，以便可以在其他线程中访问它。

接下来，我们使用`std::future`获取`std::packaged_task`的返回值。`std::future`是一个异步结果的占位符，可以用来检查任务是否已经完成，并且可以获取任务的返回值。

具体地，我们可以在调用`enqueue`函数后，使用返回的`std::future`对象的`get()`函数来获取任务的返回值。`get()`函数会阻塞当前线程，直到任务执行完毕并返回结果。

例如，假设我们要执行一个函数`int add(int x, int y)`，我们可以使用以下方式来获取其结果：

ThreadPool pool(4); // 创建线程池，有4个线程

// 将任务加入线程池，并获取返回值的future对象
auto result = pool.enqueue(add, 3, 4);

// 等待任务执行完成，并获取返回值
int res = result.get();

std::cout << "3 + 4 = " << res << std::endl;

这里的`result.get()`会阻塞当前线程，直到任务执行完毕并返回结果。最后，我们将`res`输出到控制台。