#include <iostream>
#include <thread>
#include <future>void calculateResult(std::promise<int>& promiseObj) {// 模拟耗时计算std::this_thread::sleep_for(std::chrono::seconds(2));// 设置结果到 promise 中promiseObj.set_value(42);
}int main() {// 创建一个 promise 对象std::promise<int> promiseObj;// 获取与 promise 关联的 futurestd::future<int> futureObj = promiseObj.get_future();// 启动一个新的线程执行计算std::thread workerThread(calculateResult, std::ref(promiseObj));// 在主线程中等待任务完成,并获取结果int result = futureObj.get();std::cout << "Result: " << result << std::endl;// 等待工作线程结束workerThread.join();return 0;
}
使用future和promise可以获取到线程执行函数的结果,类似C#实现
#include <iostream>
#include <thread>
#include <future>
#include <vector>
#include <functional>
#include <queue>
class ThreadPool {
public:explicit ThreadPool(size_t numThreads) : stop(false) {for (size_t i = 0; i < numThreads; ++i) {threads.emplace_back([this] {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();}});}}~ThreadPool() {{std::unique_lock<std::mutex> lock(mutex);stop = true;}condition.notify_all();for (std::thread& thread : threads) {thread.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> result = task->get_future();{std::unique_lock<std::mutex> lock(mutex);if (stop) {throw std::runtime_error("enqueue on stopped ThreadPool");}tasks.emplace([task]() { (*task)(); });}condition.notify_one();return result;}private:std::vector<std::thread> threads;std::queue<std::function<void()>> tasks;std::mutex mutex;std::condition_variable condition;bool stop;
};int calculateResult() {// 模拟耗时计算std::this_thread::sleep_for(std::chrono::seconds(2));return 42;
}int main() {ThreadPool pool(4);std::future<int> futureObj = pool.enqueue(calculateResult);int result = futureObj.get();std::cout << "Result: " << result << std::endl;return 0;
}
在线程池中获取线程执行函数的返回值时,通常使用 std::future 而不是 std::promise 来传递返回值。这是因为线程池内部已经管理了任务的执行和结果的传递,你只需要将任务提交给线程池,并使用 std::future 来获取结果。
线程池内部一般会使用一个任务队列来存储待执行的任务,并使用一个线程池管理器来调度任务的执行。当你向线程池提交任务时,线程池会选择一个空闲的线程来执行任务,并将结果存储在与任务关联的 std::future 对象中。