C++20协程入门：理解协程基础

传统的函数调用采用同步方式，调用方必须等待被调用函数执行完毕才能继续。C++20引入的协程（Coroutines）则提供了一种异步编程的新范式，允许函数在执行过程中挂起（suspend）并在之后恢复（resume），而不会阻塞调用线程。协程的精髓在于其执行状态能够在挂起时被保存，这使得编写异步代码更加直观和高效。

理解C++20协程需要掌握三个核心概念：协程帧（coroutine frame）、承诺类型（promise type）和协程句柄（coroutine handle）。协程帧是一个在堆上分配的内存块，用于存储协程的局部变量、参数和暂停点的状态。承诺类型则是一个由用户定义的类，它控制着协程的行为，例如初始挂起、最终挂起、返回值以及未处理异常等。协程句柄则提供了对协程帧的访问，允许外部代码恢复挂起的协程或销毁协程帧。

协程的关键组件

当一个函数被声明为协程时（通常通过在其函数体中包含`co_await`、`co_yield`或`co_return`关键字），编译器会对其进行大幅度的转换。转换后的代码会依赖承诺类型来构建协程的整个生命周期。承诺类型必须包含特定的方法，例如`get_return_object`用于创建返回给调用者的对象，`initial_suspend`和`final_suspend`用于控制协程在开始和结束时的挂起行为。

构建自定义异步任务框架

虽然C++20标准提供了底层的协程机制，但它并未提供现成的高级异步任务类型（如`std::async`的替代品）。要实现一个功能完备的异步任务框架，我们需要利用这些底层工具进行封装。我们的目标是创建一个`Task`模板类，它能够代表一个异步计算，最终产生一个类型为`T`的值，并能够被安全地等待。

该框架的核心是定义`Task`类的承诺类型。这个承诺类型负责管理异步任务的状态和结果。它需要包含一个`std::promise`或类似的机制来存储计算结果，以及一个用于通知等待者任务已完成的策略（例如，通过`std::coroutine_handle`来恢复等待它的协程）。

定义Task类型及其承诺类型

首先，我们定义`Task`类本身。它应该持有一个协程句柄，该句柄指向代表其异步计算的协程帧。

```cpptemplateclass Task {public: // 承诺类型的声明，在Task内部定义 struct promise_type { // 用于存储最终计算结果 std::optional result_; // 用于在任务完成时恢复等待者的协程句柄 std::coroutine_handle<> continuation_; Task get_return_object() { // 创建Task对象，并使其内部的句柄指向当前承诺对象创建的协程 return Task{std::coroutine_handle::from_promise(this)}; } std::suspend_always initial_suspend() noexcept { return {}; } // 最终挂起时需要处理，以便通知等待者 auto final_suspend() noexcept { struct Awaiter { bool await_ready() noexcept { return false; } // 当Task协程执行完毕，在最终挂起时，恢复正在等待它的那个协程（如果有的话） void await_suspend(std::coroutine_handle h) noexcept { auto& promise = h.promise(); if (promise.continuation_) { promise.continuation_.resume(); } } void await_resume() noexcept {} }; return Awaiter{}; } void unhandled_exception() { // 简单起见，将异常捕获并存储在result_中（实际框架需更健壮的错误处理） try { std::rethrow_exception(std::current_exception()); } catch (const std::exception& e) { // 处理异常... 例如存储在result_中或另设异常成员 } } // 处理co_return value; template U> void return_value(U&& value) { result_ = std::forward(value); } // 处理co_return; (对于void特化) void return_void() {} };private: std::coroutine_handle coroutine_handle_;public: explicit Task(std::coroutine_handle handle) : coroutine_handle_(handle) {} ~Task() { if (coroutine_handle_) { coroutine_handle_.destroy(); } } // 禁止拷贝，允许移动 Task(const Task&) = delete; Task& operator=(const Task&) = delete; Task(Task&& other) noexcept : coroutine_handle_(std::exchange(other.coroutine_handle_, {})) {} Task& operator=(Task&& other) noexcept { if (std::addressof(other) != this) { if (coroutine_handle_) { coroutine_handle_.destroy(); } coroutine_handle_ = std::exchange(other.coroutine_handle_, {}); } return this; }};```