通八洲科技

C++怎么实现一个简单的协程调度器_C++异步编程底层原理剖析

日期:2025-12-08 00:00 / 作者:冰火之心
答案:C++20协程通过co_await、co_yield、co_return实现异步,结合promise_type、handle和awaiter构建状态机;示例中Task封装协程句柄,Scheduler以FIFO队列管理并轮流恢复执行,体现协作式多任务调度机制。

协程是现代C++异步编程的重要组成部分,尤其从C++20开始引入了原生的协程支持。实现一个简单的协程调度器,有助于理解其底层运行机制。下面我们一步步剖析如何用C++20实现一个基础但完整的协程调度器,并解释背后的异步原理。

协程的基本概念与C++20语法

在C++20中,协程通过三个关键字支持:co_awaitco_yieldco_return。函数只要包含其中之一,就会被编译器视为协程。

协程的核心结构包括:

协程不会像普通函数那样“调用即执行”,而是创建后处于暂停状态,需要手动恢复。

实现一个简单的调度器

我们设计一个单线程的FIFO调度器,能注册并运行多个协程。

#include iostream>
#include
#include
#include
struct Task {
struct promise_type {
Task get_return_object() {
return Task{std::coroutine_handle::from_promise(*this)};
}
std::suspend_always initial_suspend() { return {}; }
std::suspend_always final_suspend() noexcept { return {}; }
void return_void() {}
void unhandled_exception() {}
};
std::coroutine_handle handle;
~Task() {
if (handle) handle.destroy();
}
Task(const Task&) = delete;
Task& operator=(const Task&) = delete;
Task(Task&& other) : handle(other.handle) {
other.handle = nullptr;
}
};
class Scheduler {
private:
std::queue<:coroutine_handle>> tasks;
public:
void schedule(Task task) {
tasks.push(task.handle);
}
void run() {
while (!tasks.empty()) {
auto handle = tasks.front();
tasks.pop();
if (!handle.done()) {
handle.resume(); // 恢复协程
}
if (!handle.done()) {
tasks.push(handle); // 若未结束,重新入队
} else {
handle.destroy(); // 清理已完成的协程
}
}
}
};

上面的代码定义了一个最简任务类型 Task,它默认挂起初始和结尾状态。调度器使用队列保存待执行的协程句柄,逐个恢复运行。

编写可调度的协程示例

下面是一个模拟异步操作的协程,每执行一次输出后让出控制权。

Task async_task(Scheduler& sched, int id) {
for (int i = 0; i std::cout co_await std::suspend_always{}; // 主动挂起
}
}
int main() {
Scheduler sched;
sched.schedule(async_task(sched, 1));
sched.schedule(async_task(sched, 2));
sched.schedule(async_task(sched, 3));
sched.run();
return 0;
}

输出结果会交替执行各个任务:

Task 1 step 0
Task 2 step 0
Task 3 step 0
Task 1 step 1
Task 2 step 1
Task 3 step 1
Task 1 step 2
Task 2 step 2
Task 3 step 2

这体现了协作式多任务的特点:每个任务主动让出CPU,调度器轮流唤醒。

底层原理剖析

C++协程的底层依赖于编译器生成的状态机。当协程遇到 co_await,编译器会:

协程的开销主要在于堆内存分配(除非优化掉)和上下文切换逻辑。但相比线程,它轻量得多,没有内核态切换成本。

调度器的本质就是管理这些 coroutine_handle 的生命周期和执行顺序,可以扩展为支持优先级、定时唤醒、事件驱动等高级特性。

基本上就这些。C++20协程虽然语法复杂,但一旦理解了 handle、promise 和 awaiter 三者的关系,就能构建出灵活高效的异步系统。这个简单调度器是迈向高性能异步框架的第一步。