C++ 中的协程：异步编程的现代实现

作者

协程（coroutine）是 C++20 标准中引入的一项重要特性，它提供了一种轻量级的协作式多任务机制，使得异步编程变得更加直观和高效。与传统的线程或基于回调的异步模型相比，协程在语义上更接近同步代码，同时保持了非阻塞的执行特点。本文将从协程的基本概念、实现原理、使用场景以及性能考虑等方面进行系统阐述，并给出一段完整的示例代码，帮助读者快速上手。

1. 协程的基本概念

协程是一种可挂起的函数，能够在执行过程中“暂停”并“恢复”，从而实现多任务并发。C++20 对协程的支持主要体现在以下几方面：

  • co_await：用于挂起协程，等待某个异步操作完成。
  • co_yield：用于从协程返回一个值，并暂停执行。
  • co_return：用于结束协程并返回最终结果。
  • std::suspend_always / std::suspend_never：控制协程的挂起行为。
  • Promise 类型：协程的返回值由 promise 生成器实现，决定了协程的结果类型。

协程的生命周期由编译器生成的状态机管理，状态机会保存局部变量、分支信息等，以便在挂起后能够恢复执行。

2. 协程实现原理

C++ 协程实际上是把一个函数拆分成若干个“暂停点”，并生成一个状态机对象。编译器在编译阶段会把 co_await、co_yield、co_return 替换成对 promise 对象的调用，生成的代码大致如下：

struct coro_state {
    std::coroutine_handle<> handle;
    // 保存局部变量
    int counter;
    // ...
};

auto my_coroutine() -> std::future <int> {
    // 创建 promise 对象
    struct promise_type {
        int value;
        std::future <int> get_return_object() { /* ... */ }
        std::suspend_always initial_suspend() { return {}; }
        std::suspend_always final_suspend() noexcept { return {}; }
        void return_value(int v) { value = v; }
        void unhandled_exception() { /* ... */ }
    };

    // 状态机
    // ...
}

在 co_await 处，编译器会生成一段代码，用于挂起协程并返回控制权；在恢复时，状态机会根据内部的状态继续执行。

3. 常见协程模型

模型 说明 典型库
任务/Future 基于 std::future，支持异步结果 std::future, std::promise
流 使用 co_yield 返回序列 generator、std::ranges
事件循环 协程作为任务挂载到事件循环 asio::awaitable (Boost.Asio), cppcoro
Coroutine Scheduler 手动调度协程 folly::coro, cppcoro

4. 使用场景

  1. 高性能网络 I/O
    通过协程，可以把异步 socket 操作写成同步样式，避免回调地狱。许多网络库（如 Boost.Asio）已将协程整合进核心。
  2. 并发流处理
    co_yield 生成器可用于处理流式数据，如解析大文件、处理事件流等。
  3. 微任务调度
    在游戏、渲染引擎中，协程可实现细粒度任务切换，提高帧率和响应性。
  4. 延迟计算
    co_await 可以实现懒加载或缓存回调，提升资源利用率。

5. 性能与注意事项

  • 栈占用：协程默认使用堆分配的状态机，避免了栈分配导致的栈溢出。若需要更小的占用，可使用 std::coroutine_handle::promise() 的自定义 allocator。
  • 上下文切换成本：协程的挂起/恢复成本比线程轻量，但如果频繁切换仍会产生一定开销。应避免过细粒度的协程划分。
  • 异常传播：协程内部异常会被包装到 promise 对象，可通过 co_await 捕获或 future.get() 抛出。
  • 标准库兼容：在不同编译器（MSVC、Clang、GCC）中，协程实现仍在完善，编写时需留意特定编译器的扩展。

6. 示例：基于协程的异步文件读取

下面给出一个完整示例，演示如何使用协程读取文件并处理数据。示例中使用了 cppcoro 库（若不想依赖外部库，可以改为标准 std::future）：

#include <cppcoro/generator.hpp>
#include <cppcoro/io_service.hpp>
#include <cppcoro/stream.hpp>
#include <fstream>
#include <iostream>

cppcoro::generator<std::string> readLines(const std::string& filename)
{
    std::ifstream file(filename);
    std::string line;
    while (std::getline(file, line)) {
        co_yield line;                 // 每行返回一次
    }
}

cppcoro::task <void> processFile(const std::string& filename)
{
    auto gen = readLines(filename);
    for (auto&& line : gen) {
        // 在协程上下文中处理每一行
        std::cout << "Line: " << line << '\n';
        co_await cppcoro::resume_on_thread_pool(); // 模拟异步操作
    }
}

int main()
{
    cppcoro::io_service ioService;
    ioService.run([]{
        return processFile("example.txt");
    });
    return 0;
}

该示例展示了如何把同步的文件读取过程包装为协程，使用 co_yield 逐行返回，同时在处理时利用 co_await 实现非阻塞等待。通过 cppcoro::io_service 可以把协程与线程池或事件循环绑定，从而真正发挥协程的异步优势。

7. 结语

C++ 协程为语言提供了一种全新的异步编程范式，既保持了同步代码的可读性，又不牺牲性能。随着编译器优化和标准库扩展，协程将成为构建高性能、可维护网络应用、数据处理管道以及游戏引擎的重要工具。希望本文能帮助你快速理解协程的核心概念，并在实际项目中灵活运用。

发表评论