带返回值的任务提交 on Dorkytiger 的小屋

问题

Mon, 01 Jan 0001 00:00:00 +0000

目前 enqueue(f) 无法获取返回值。如果任务返回一个 int，调用者想知道结果：

int compute(int x) { return x * x; }

pool.enqueue([&] {
    int result = compute(42);
    // 怎么把 result 传给调用者？
});

解决方案：std::packaged_task + std::future。

std::packaged_task<R()> 包装一个可调用对象，它关联一个 std::future<R>。当你执行 packaged_task 时，结果被存入关联的 future，调用者通过 future.get() 获取。

带返回值的 enqueue

Mon, 01 Jan 0001 00:00:00 +0000

class ThreadPool {
    // ... 同第一阶段 ...

    template <typename F, typename... Args>
    auto enqueue(F&& f, Args&&... args)
        -> std::future<std::invoke_result_t<F, Args...>>
    {
        using ReturnType = std::invoke_result_t<F, Args...>;

        // 把函数和参数打包成一个 packaged_task
        auto task = std::make_shared<std::packaged_task<ReturnType()>>(
            std::bind(std::forward<F>(f), std::forward<Args>(args)...)
        );

        std::future<ReturnType> result = task->get_future();

        {
            std::lock_guard<std::mutex> lock(mtx_);
            if (stop_) throw std::runtime_error("enqueue on stopped ThreadPool");
            tasks_.push([task]() { (*task)(); });
        }
        cv_.notify_one();

        return result;
    }
};

用法：

auto future = pool.enqueue(compute, 42);
// 在这里可以干其他事情...
int result = future.get();  // 阻塞等待结果

// 如果 compute 抛异常，future.get() 会重新抛出异常

执行过程

Mon, 01 Jan 0001 00:00:00 +0000

调用者线程：                 工作线程：
enqueue(compute, 42)
  ↓
创建 packaged_task<int()>
  ↓
把 task 包装成 void() lambda
放入任务队列
  ↓
cv.notify_one() ──────────→ 唤醒等待的工作线程
  ↓                           ↓
返回 future<int>             取任务，执行 (*task)()
  ↓                           ↓
调用 future.get() ←────────── 结果存入 future
  ↓                           阻塞直到结果就绪
拿到结果 42

`std::bind` vs Lambda——选哪个？

Mon, 01 Jan 0001 00:00:00 +0000

上面用了 std::bind，其实用 Lambda 也可以：

// bind 写法
auto task = std::make_shared<std::packaged_task<ReturnType()>>(
    std::bind(std::forward<F>(f), std::forward<Args>(args)...)
);

// lambda 写法（C++17 起更推荐）
auto task = std::make_shared<std::packaged_task<ReturnType()>>(
    [f = std::forward<F>(f), ...args = std::forward<Args>(args)]() mutable {
        return std::invoke(std::move(f), std::move(args)...);
    }
);

Lambda 写法更直观，但注意 C++17 才支持 lambda 捕获参数包展开。面试用 std::bind 更不会出错。