理解C++20的革命特性------协程支持2:编写简单的协程调度器
仓库:https://github.com/Charliechen114514/co_practices
前言
在上一篇博客中,我们已经理解了C++20中最为简单的协程调度接口(尽管这一点也不简单)。显然,在这篇博客之前,我们的协程之间还是在使用单协程的调度器进行调度。看起来协程好鸡肋。啥也干不了。但是别着急,为了我们可以进一步的发挥协程的威力。笔者需要你动手完成这个简单的小任务,这个小任务并不困难:
- 实现一个
Task<T>
,可以co_await
一个返回值。(理解coroutine_handle
的 resume/suspend 生命周期。),并利用Task<int>
写一个协程函数co_add(a,b)
,返回 a+b,调用方co_await
获取结果。
如果你看上面的题目一头雾水,觉得不知道我在说什么的话------您可以先阅读下面的调用代码,然后返回我的上一篇博客琢磨一下该怎么写。(你怎么知道我找到这个练习的时候自己也是一头雾水的)
cpp
Task<int> co_add(int a, int b) {
simple_log_with_func_name(
std::format("Get a: {} and b: {}, "
"expected a + b = {}",
a, b, a + b));
co_return a + b;
}
Task<void> examples(int a, int b) {
simple_log("About to call co_add");
int result = co_await co_add(a, b);
simple_log(std::format("Get the result: {}", result));
co_return;
}
int main() {
simple_log_with_func_name();
examples(1, 2);
simple_log("Done!");
}
你所需要做的,就是上面的代码跑起来,跑起来的办法就是实现Task<T>
。如果您做好了,请您参考下面的代码对比一下实现。我们后面会复用Task<T>
来完成我们本篇博客的主题------带有返回值支持的调度器。
这里是笔者的代码。"helpers.h"
已经在上一篇博客被给出了,没有任何变化,放心使用。
cpp
#include "helpers.h"
#include <coroutine>
#include <format>
template <typename T>
class Task {
public:
struct promise_type;
using coro_handle = std::coroutine_handle<promise_type>;
Task(coro_handle h)
: coroutine_handle(h) {
simple_log_with_func_name();
}
~Task() {
simple_log_with_func_name();
if (coroutine_handle) {
coroutine_handle.destroy();
}
}
Task(Task&& o)
: coroutine_handle(o.coroutine_handle) {
o.coroutine_handle = nullptr;
}
Task& operator=(Task&& o) {
coroutine_handle = std::move(o.coroutine_handle);
o.coroutine_handle = nullptr;
return *this;
}
// concept requires
struct promise_type {
T cached_value;
Task get_return_object() {
simple_log_with_func_name();
return { coro_handle::from_promise(*this) };
}
// we dont need suspend when first suspend
std::suspend_never initial_suspend() {
simple_log_with_func_name();
return {};
}
// suspend always for the Task clean ups
std::suspend_always final_suspend() noexcept {
simple_log_with_func_name();
return {};
}
void return_value(T value) {
simple_log_with_func_name(std::format("value T {} is received!", value));
cached_value = std::move(value);
}
void unhandled_exception() {
// process notings
}
};
bool await_ready() {
simple_log_with_func_name();
return false; // always need suspend
}
void await_suspend(std::coroutine_handle<> h) {
simple_log_with_func_name(); // Should never be here
h.resume(); // resume these always
}
T await_resume() {
simple_log_with_func_name();
return coroutine_handle.promise().cached_value;
}
private:
coro_handle coroutine_handle;
private:
Task(const Task&) = delete;
Task& operator=(const Task&) = delete;
};
template <>
class Task<void> {
public:
struct promise_type;
using coro_handle = std::coroutine_handle<promise_type>;
Task(coro_handle h)
: coroutine_handle(h) {
simple_log_with_func_name();
}
~Task() {
simple_log_with_func_name();
if (coroutine_handle) {
coroutine_handle.destroy();
}
}
Task(Task&& o)
: coroutine_handle(o.coroutine_handle) {
o.coroutine_handle = nullptr;
}
Task& operator=(Task&& o) {
coroutine_handle = std::move(o.coroutine_handle);
o.coroutine_handle = nullptr;
return *this;
}
// concept requires
struct promise_type {
Task get_return_object() {
simple_log_with_func_name();
return { coro_handle::from_promise(*this) };
}
// we dont need suspend when first suspend
std::suspend_never initial_suspend() {
simple_log_with_func_name();
return {};
}
// suspend always for the Task clean ups
std::suspend_always final_suspend() noexcept {
simple_log_with_func_name();
return {};
}
void return_void() { simple_log_with_func_name(); }
void unhandled_exception() {
// process notings
}
};
private:
coro_handle coroutine_handle;
private:
Task(const Task&) = delete;
Task& operator=(const Task&) = delete;
};
Task<int> co_add(int a, int b) {
simple_log_with_func_name(
std::format("Get a: {} and b: {}, "
"expected a + b = {}",
a, b, a + b));
co_return a + b;
}
Task<void> examples(int a, int b) {
simple_log("About to call co_add");
int result = co_await co_add(a, b);
simple_log(std::format("Get the result: {}", result));
co_return;
}
int main() {
simple_log_with_func_name();
examples(1, 2);
simple_log("Done!");
}
如果你没有看懂发生了什么,请继续看下面的内容,如果实现跟你的差不多,您可以再翻上去继续编写调度器了。
实现一个最简单的调度器
我们下面马上就来实现一个最简单的调度器了。下面是我们的要求
- 写一个单例的单线程调度器 (event loop),能调度多个
Task
。(建议编写一个单例模板玩,另外,Task的基本代码由上一个任务已经做完了)- 实现
sleep(ms)
awaiter- 检测能不能用------写 3 个协程并发运行:打印 "A"、"B"、"C",交替输出。
第一步------实现一个单例的模板
笔者决定实现一个简单的单例模板,方便我们其他项目的复用。关于单例模式的探讨,尽管依赖注入(DI)是更加合适的,但是我们还是编写基于static的单例模板(协程是C++20才有的,C++11以上已经保证了static的初始化是线程安全的)
single_instance.hpp
cpp
#pragma once
template <typename SingleInstanceType>
class SingleInstance {
public:
static SingleInstanceType& instance() {
static SingleInstanceType instance;
return instance;
}
protected:
SingleInstance() = default;
virtual ~SingleInstance() = default;
private:
SingleInstance(const SingleInstance&) = delete;
SingleInstance& operator=(const SingleInstance&) = delete;
SingleInstance(SingleInstance&&) = delete;
SingleInstance& operator=(SingleInstance&&) = delete;
};
很显然,我们禁用了任何形式的拷贝和构造,而且为了后续的使用方便,咱们要采用安全的虚析构函数。SingleInstance()
要放到保护域下,咱们的单例子类要访问这个保证我们在语法上回避第二个实例的创建。使用上,咱们只需要这样书写:
cpp
class Schedular : public SingleInstance<Schedular>
{
Schedular() = default; // 还是藏起来我们的构造函数
public:
friend class SingleInstance<Schedular>;
}
好巧不巧,笔者写过单例模式的探讨,实现也是C++20的,参考博客:
第二步:初步修改一下我们的Task
,让调度器有机会接管我们的协程
很显然------我们现在决定利用调度器来调度我们的协程了------那么,任何的挂起操作都需要我们来控制而不是返回结构体自行裁决,为此,我们的初始化也需要立马被挂起:
cpp
// we need suspend when first suspend
std::suspend_always initial_suspend() {
// simple_log_with_func_name();
return {};
}
不管是哪一个泛型实现还是偏特化实现亦是如此。
第三步:思考调度器支持的接口
我们下面准备思考调度器的接口了,好在我们的协程不是抢占式调度,代码写起来非常的容易(但是容易不太可能),只需要遵循无让出时的FIFO调度就好了。
首先,调度器需要支持Sleep调用,也就是让当前的协程睡大觉(有其他协程任务就做其他的,没有咱们说明当前的线程是需要空闲的,调用std::this_thread::sleep_*
接口就好了)
所以,我们需要让调度器知道哪一些协程是需要睡大觉的------调度器需要有一个容器管理谁需要睡觉,和一个推送需要睡觉的指定协程。
需要知道一个事情------标准库为了方便,是存在一个叫做sleep_until
的接口的,所以,为了方便管理和复用标准库接口,我们设计一个调度器的sleep_until接口------它表明我们要休眠到指定的时间点就可以准备被调度(再次强调,我们需要注意协程的调度是协同调度,我们只能保证休眠的下限事件)。
cpp
void Schedular::sleep_until(std::coroutine_handle<> which, // 谁需要休眠?
std::chrono::steady_clock::time_point until_when);
另外,咱们还要有一个推送接口:spawn接口,用来接受协程函数返回结构体。这个结构体的所有调度都要被调度器接管。所以,别忘记在Task处声明调度器类为友元。
cpp
template <typename T>
void Schedular::spawn(Task<T>&& task); // Task只可以被移动,所以放这个接口进来
最后还有一个调度接口------run接口
cpp
void Schedular::run();
他将会开始我们的协程调度。就三个!
第四步:实现上面的接口
实现spawn接口,托管协程函数返回的协程返回结构体
我们先从调度本身开始,首先,我们需要缓存预备列的协程接口(注意不是Task本身,我们在调度协程而不是协程的返回结构体),上面提到了咱们的调度策略是FIFO,因此,先来先到就要求我们采用队列来处理我们的存储。
cpp
std::queue<std::coroutine_handle<>> ready_coroutines; // 一个简单的队列即可
所以,咱们的spawn接口变得非常好实现------
cpp
void Schedular::internal_spawn(std::coroutine_handle<> h) {
// private实现,用户不应该直接随意的触碰调度队列
ready_coroutines.push(h); // 加入调度队列
}
// spawn是一个桥接的接口,我们会取出来Task内托管的coroutine_handle协程句柄,交给我们的
// 调度器来管理
template <typename T>
inline void Schedular::spawn(Task<T>&& task) {
internal_spawn(task.coroutine_handle);
task.coroutine_handle = nullptr; // 让Task不再托管coroutine_handle本身
}
实现睡眠机制
睡眠需要登记我们睡多久,谁睡觉,而且还要按照一定的优先级排序(你想,如果有三个睡眠请求100ms,200ms,300ms的睡眠,肯定是优先睡眠100ms的,再睡200ms,再睡300ms,反过来的话,前两个黄瓜菜都凉了),显然我们立马想到了优先级队列。但是优先级队列需要提供比较方法从而产生小/大根堆。所以我们需要抽象SleepItem
结构体------它登记我们的根是睡眠事件最小的。或者说离当前事件点最近的。
cpp
struct SleepItem {
SleepItem(std::coroutine_handle<> h,
std::chrono::steady_clock::time_point tp)
: coro_handle(h)
, sleep(tp) {
}
std::chrono::steady_clock::time_point sleep;
std::coroutine_handle<> coro_handle;
bool operator<(const SleepItem& other) const {
return sleep > other.sleep;
}
};
std::priority_queue<SleepItem> sleepys;
但是我们还没有实现用户侧代码,用户期盼我们可以这样睡眠:
cpp
co_await sleep(300ms);
欸,怎么说的?看到co_await就要条件反射实现awaitable接口。所以------
cpp
struct AwaitableSleep {
AwaitableSleep(std::chrono::milliseconds how_long)
: duration(how_long)
, wake_time(std::chrono::steady_clock::now() + how_long) { }
/**
* @brief await_ready always lets the sessions sleep!
*
*/
bool await_ready() { return false; } // 总是我们接管剩下的流程
void await_suspend(std::coroutine_handle<> h) {
// 执行推送,然后后面我们自己的调度器会取出来这个句柄扔到就绪队列中
Schedular::instance().sleep_until(h, wake_time);
}
// 什么都不做
void await_resume() { }
private:
std::chrono::milliseconds duration; // 方便获取接口或者调试,性能优先下可以踢掉这个
std::chrono::steady_clock::time_point wake_time;
};
inline AwaitableSleep sleep(std::chrono::milliseconds s) {
return { s };
}
实现调度逻辑
首先,睡眠是没活干才做,实现上的优先级很明显了------优先处理活跃的协程!
cpp
void run() {
// if there is any corotines ready or sleepy unfinished
while (!ready_coroutines.empty() || !sleepys.empty()) {
// 进来这个逻辑,就表明我们现在是有事情做的------不管是睡大觉还是拉起一个协程。
while (!ready_coroutines.empty()) {
auto front_one = ready_coroutines.front();
ready_coroutines.pop();
front_one.resume(); // OK, hang this on!
}
...
}
}
如果我们任何活跃代码都执行完毕了,我们才会去检查睡眠队列中有没有待唤醒的家伙------
cpp
auto now = current(); // current返回std::chrono::steady_clock::now()
while (!sleepys.empty() && sleepys.top().sleep <= now) {
ready_coroutines.push(sleepys.top().coro_handle);
sleepys.pop();
}
非常好,如果我们现在的事件越过了指定睡眠唤醒的时间点(也就是sleepys.top().sleep),咱们就要放所有越过了时间点的协程送到咱们的就绪队列中。
下一步,如果我们还有协程需要睡觉,且没有新的就绪队列到来,我们立马就对本线程进行睡眠
cpp
void run() {
// if there is any corotines ready or sleepy unfinished
while (!ready_coroutines.empty() || !sleepys.empty()) {
while (!ready_coroutines.empty()) {
auto front_one = ready_coroutines.front();
ready_coroutines.pop();
front_one.resume(); // OK, hang this on!
}
auto now = current();
while (!sleepys.empty() && sleepys.top().sleep <= now) {
ready_coroutines.push(sleepys.top().coro_handle);
sleepys.pop();
}
if (ready_coroutines.empty() && !sleepys.empty()) {
// OK, we can sleep
std::this_thread::sleep_until(sleepys.top().sleep);
}
}
}
继续修改Task的接口
现在任务需要直接向队列里推送了,我们需要思考这些问题。我们使用调度器会这样使用:
cpp
Task<int> co_add(int a, int b) {
co_await sleep(300ms);
co_return a + b;
}
Task<void> worker(const char* name, int a, int b) {
int result = co_await co_add(a, b);
std::println("{}: {} + {} = {}", name, a, b, result);
}
Task<void> main_task() {
co_await worker("TaskA", 1, 2);
co_await worker("TaskB", 3, 4);
co_await worker("TaskC", 5, 6);
}
所有的父协程会放下自身的运行,按照C++20无栈协程的逻辑------我们要自己保存协程的句柄。所以我们很容易想到------Task本身要存储父协程的句柄,方便我们子协程恢复的时候恢复父协程的运行,才能继续代码。
可能太跳跃了,我们一个一个慢慢来------我们的父协程里写下代码------co_await worker("TaskA", 1, 2);
的时候,父协程就要放弃自己的运行,等待worker的结果。这个时候,我们回忆第一篇博客我们协程框架的运行逻辑:走await_ready
查看是否挂起------我们显然返回了否,要自己接管逻辑。所以下一步的执行流被转发到了await_suspend
中,这一步就是我们要的------父协程要被挂起,所以子协程要被推送!
cpp
// 在创建的子协程的协程返回体中
void await_suspend(std::coroutine_handle<> h) {
// simple_log_with_func_name(); // Should never be here
simple_log("Current Routine will be suspend!");
coroutine_handle.promise().parent_coroutine = h;
simple_log("Child Routine will be called resume!");
Schedular::instance().internal_spawn(coroutine_handle);
}
coroutine_handle.promise().parent_coroutine = h;
设置了子协程的父协程为当前线程,然后将子协程放到就绪队列里去。没啥毛病!(注意这个代码是子协程返回结构体)
现在,我们的子协程就被送到就绪队列中,而且令人兴奋的是------它会被送到就绪的处理逻辑里,当我们的调度器执行就绪的协程队列代码的时候,我们就会执行这个逻辑------
cpp
while (!ready_coroutines.empty()) {
auto front_one = ready_coroutines.front();
ready_coroutines.pop();
front_one.resume(); // OK, hang this on!
}
子协程在这里被resume了,执行的就是worker的代码------子协程这下就被挂起。当worker执行结束之后,我们仍然按照流程------调用的是final_suspend
,还记得我们存储的parent_coroutine嘛?这里发力了------子协程的结束要求父协程放下执行代码.所以事情变得非常的容易:
cpp
std::suspend_always final_suspend() noexcept {
// simple_log_with_func_name();
if (parent_coroutine) {
simple_log("parent_coroutine will be wake up");
// 父协程拉起来执行代码
Schedular::instance().internal_spawn(parent_coroutine);
}
return {}; // 子协程由Task结构体托管,这个逻辑不会发生改变
}
走到这里,我们所有的代码都完成了.我们编译运行一下:
cpp
[charliechen@Charliechen coroutines]$ build/schedular/schedular
10:36:12 :Current Routine will be suspend!
10:36:12 :Child Routine will be called resume!
10:36:12 :Current Routine will be suspend!
10:36:12 :Child Routine will be called resume!
10:36:13 :parent_coroutine will be wake up
TaskA: 1 + 2 = 3
10:36:13 :Current Routine will be suspend!
10:36:13 :Child Routine will be called resume!
10:36:13 :Current Routine will be suspend!
10:36:13 :Child Routine will be called resume!
10:36:13 :parent_coroutine will be wake up
TaskB: 3 + 4 = 7
10:36:13 :Current Routine will be suspend!
10:36:13 :Child Routine will be called resume!
10:36:13 :Current Routine will be suspend!
10:36:13 :Child Routine will be called resume!
10:36:13 :parent_coroutine will be wake up
TaskC: 5 + 6 = 11
代码工作的非常完美.
上面的日志如何产生的?答案如下:
[charliechen@Charliechen coroutines]$ build/schedular/schedular 10:36:12 :Current Routine will be suspend! // main_task准备被挂起 10:36:12 :Child Routine will be called resume! // worker("TaskA", 1, 2);准备干活 10:36:12 :Current Routine will be suspend! // worker("TaskA", 1, 2)准备被挂起 10:36:12 :Child Routine will be called resume! // co_add准备干活 10:36:13 :parent_coroutine will be wake up // co_add作为叶子协程,准备结束自己,拉起父协程worker干活 TaskA: 1 + 2 = 3 // worker被拉起,执行打印逻辑 // 如下的逻辑是类似的 10:36:13 :Current Routine will be suspend! 10:36:13 :Child Routine will be called resume! 10:36:13 :Current Routine will be suspend! 10:36:13 :Child Routine will be called resume! 10:36:13 :parent_coroutine will be wake up TaskB: 3 + 4 = 7 10:36:13 :Current Routine will be suspend! 10:36:13 :Child Routine will be called resume! 10:36:13 :Current Routine will be suspend! 10:36:13 :Child Routine will be called resume! 10:36:13 :parent_coroutine will be wake up TaskC: 5 + 6 = 11
附录:实现协程的加法函数co_add
为了防止你来回翻阅,笔者仍然直接把代码CV一份放在这里。
cpp
#include "helpers.h"
#include <coroutine>
#include <format>
template <typename T>
class Task {
public:
struct promise_type;
using coro_handle = std::coroutine_handle<promise_type>;
Task(coro_handle h)
: coroutine_handle(h) {
simple_log_with_func_name();
}
~Task() {
simple_log_with_func_name();
if (coroutine_handle) {
coroutine_handle.destroy();
}
}
Task(Task&& o)
: coroutine_handle(o.coroutine_handle) {
o.coroutine_handle = nullptr;
}
Task& operator=(Task&& o) {
coroutine_handle = std::move(o.coroutine_handle);
o.coroutine_handle = nullptr;
return *this;
}
// concept requires
struct promise_type {
T cached_value;
Task get_return_object() {
simple_log_with_func_name();
return { coro_handle::from_promise(*this) };
}
// we dont need suspend when first suspend
std::suspend_never initial_suspend() {
simple_log_with_func_name();
return {};
}
// suspend always for the Task clean ups
std::suspend_always final_suspend() noexcept {
simple_log_with_func_name();
return {};
}
void return_value(T value) {
simple_log_with_func_name(std::format("value T {} is received!", value));
cached_value = std::move(value);
}
void unhandled_exception() {
// process notings
}
};
bool await_ready() {
simple_log_with_func_name();
return false; // always need suspend
}
void await_suspend(std::coroutine_handle<> h) {
simple_log_with_func_name(); // Should never be here
h.resume(); // resume these always
}
T await_resume() {
simple_log_with_func_name();
return coroutine_handle.promise().cached_value;
}
private:
coro_handle coroutine_handle;
private:
Task(const Task&) = delete;
Task& operator=(const Task&) = delete;
};
template <>
class Task<void> {
public:
struct promise_type;
using coro_handle = std::coroutine_handle<promise_type>;
Task(coro_handle h)
: coroutine_handle(h) {
simple_log_with_func_name();
}
~Task() {
simple_log_with_func_name();
if (coroutine_handle) {
coroutine_handle.destroy();
}
}
Task(Task&& o)
: coroutine_handle(o.coroutine_handle) {
o.coroutine_handle = nullptr;
}
Task& operator=(Task&& o) {
coroutine_handle = std::move(o.coroutine_handle);
o.coroutine_handle = nullptr;
return *this;
}
// concept requires
struct promise_type {
Task get_return_object() {
simple_log_with_func_name();
return { coro_handle::from_promise(*this) };
}
// we dont need suspend when first suspend
std::suspend_never initial_suspend() {
simple_log_with_func_name();
return {};
}
// suspend always for the Task clean ups
std::suspend_always final_suspend() noexcept {
simple_log_with_func_name();
return {};
}
void return_void() { simple_log_with_func_name(); }
void unhandled_exception() {
// process notings
}
};
private:
coro_handle coroutine_handle;
private:
Task(const Task&) = delete;
Task& operator=(const Task&) = delete;
};
首先,上一篇博客咱们已经提到了------任何跑在协程的函数必须返回协程返回类型 ,这个事情要求你不容商量的内嵌一个结构体struct promise_type
,而且要求你必须实现接口------
cpp
struct promise_type {
T cached_value;
Task get_return_object() {
simple_log_with_func_name();
return { coro_handle::from_promise(*this) };
}
// we dont need suspend when first suspend
std::suspend_never initial_suspend() {
simple_log_with_func_name();
return {};
}
// suspend always for the Task clean ups
std::suspend_always final_suspend() noexcept {
simple_log_with_func_name();
return {};
}
void return_value(T value) {
simple_log_with_func_name(std::format("value T {} is received!", value));
cached_value = std::move(value);
}
void unhandled_exception() {
// process notings
}
};
本篇样例中,我们不难理解的是------co_add
并不需要创建协程即挂起,所以咱们只需要返回std::suspend_never
即可,让我们立马执行返回的结果co_return a + b
上,a + b
被计算好后,会被送到return_value
当中去,需要注意的是------上一篇博客我们已经讨论了生命周期上返回类型和协程句柄本身谁要更长了,这也是为什么我们选择挂起,这样的话让更上一层的Task
来负责析构协程对象,而不是它自己解决它自己。这个结构你不会感到陌生的,上一篇博客已经说明了这个结构到底在干啥。
co_await要求等待Task<int>
,所以任何非空的Task
还要实现Awaitable接口(注意,不是说带有PromiseType接口的返回结构体都要实现Awaitable接口,而是我们需要co_await这个接口的时候才需要实现Awaitable接口,请各位搞清楚逻辑关系。)
cpp
bool await_ready() {
simple_log_with_func_name();
return false; // always need suspend
}
void await_suspend(std::coroutine_handle<> h) {
simple_log_with_func_name(); // Should never be here
h.resume(); // resume these always, call await_resume then
}
T await_resume() {
simple_log_with_func_name();
return coroutine_handle.promise().cached_value;
}
尽管逻辑上,咱们实际上不需要挂起接口,但是我们的结果存储在coroutine_handle的promise_type里了,这个时候------我们需要接管等待的逻辑,所以还是要挂起来。
await_ready实际上也可以表达为------我们需要接管等待的逻辑做我们自己的处理
第一篇博客在:
- CSDN链接:CSDN
- 我自己博客的链接:charliechen114514.tech
附录2:调度器的代码
schedular.cpp: example的主代码
cpp
#include "schedular.hpp"
#include <print>
using namespace std::chrono_literals;
Task<int> co_add(int a, int b) {
co_await sleep(300ms);
co_return a + b;
}
Task<void> worker(const char* name, int a, int b) {
int result = co_await co_add(a, b);
std::println("{}: {} + {} = {}", name, a, b, result);
}
Task<void> main_task() {
co_await worker("TaskA", 1, 2);
co_await worker("TaskB", 3, 4);
co_await worker("TaskC", 5, 6);
}
int main() {
Schedular::instance().spawn(main_task());
Schedular::instance().run();
}
schedular.hpp:调度器代码
cpp
#pragma once
#include "single_instance.hpp"
#include <chrono>
#include <coroutine>
#include <queue>
#include <thread>
template <typename T>
class Task;
struct AwaitableSleep;
class Schedular : public SingleInstance<Schedular> {
struct SleepItem {
SleepItem(std::coroutine_handle<> h,
std::chrono::steady_clock::time_point tp)
: coro_handle(h)
, sleep(tp) {
}
std::chrono::steady_clock::time_point sleep;
std::coroutine_handle<> coro_handle;
bool operator<(const SleepItem& other) const {
return sleep > other.sleep;
}
};
std::queue<std::coroutine_handle<>> ready_coroutines;
std::priority_queue<SleepItem> sleepys;
private:
Schedular() = default;
~Schedular() override {
run();
}
friend class AwaitableSleep;
template <typename T>
friend class Task;
static std::chrono::steady_clock::time_point
current() {
return std::chrono::steady_clock::now();
}
void sleep_until(std::coroutine_handle<> which,
std::chrono::steady_clock::time_point until_when) {
sleepys.emplace(which, until_when);
}
void internal_spawn(std::coroutine_handle<> h) {
ready_coroutines.push(h);
}
public:
friend class SingleInstance<Schedular>;
template <typename T>
void spawn(Task<T>&& task);
void run() {
// if there is any corotines ready or sleepy unfinished
while (!ready_coroutines.empty() || !sleepys.empty()) {
while (!ready_coroutines.empty()) {
auto front_one = ready_coroutines.front();
ready_coroutines.pop();
front_one.resume(); // OK, hang this on!
}
auto now = current();
while (!sleepys.empty() && sleepys.top().sleep <= now) {
ready_coroutines.push(sleepys.top().coro_handle);
sleepys.pop();
}
if (ready_coroutines.empty() && !sleepys.empty()) {
// OK, we can sleep
std::this_thread::sleep_until(sleepys.top().sleep);
}
}
}
};
struct AwaitableSleep {
AwaitableSleep(std::chrono::milliseconds how_long)
: duration(how_long)
, wake_time(std::chrono::steady_clock::now() + how_long) { }
/**
* @brief await_ready always lets the sessions sleep!
*
*/
bool await_ready() { return false; }
void await_suspend(std::coroutine_handle<> h) {
Schedular::instance().sleep_until(h, wake_time);
}
void await_resume() { }
private:
std::chrono::milliseconds duration;
std::chrono::steady_clock::time_point wake_time;
};
inline AwaitableSleep sleep(std::chrono::milliseconds s) {
return { s };
}
#include "task.hpp"
template <typename T>
inline void Schedular::spawn(Task<T>&& task) {
internal_spawn(task.coroutine_handle);
task.coroutine_handle = nullptr;
}
task.hpp: Task的最终抽象
cpp
#pragma once
#include "helpers.h"
#include "schedular.hpp"
#include <coroutine>
#include <utility>
template <typename T>
class Task {
public:
friend class Schedular;
struct promise_type;
using coro_handle = std::coroutine_handle<promise_type>;
Task(coro_handle h)
: coroutine_handle(h) {
// simple_log_with_func_name();
}
~Task() {
// simple_log_with_func_name();
if (coroutine_handle) {
coroutine_handle.destroy();
}
}
Task(Task&& o)
: coroutine_handle(o.coroutine_handle) {
o.coroutine_handle = nullptr;
}
Task& operator=(Task&& o) {
coroutine_handle = std::move(o.coroutine_handle);
o.coroutine_handle = nullptr;
return *this;
}
// concept requires
struct promise_type {
T cached_value;
std::coroutine_handle<> parent_coroutine;
Task get_return_object() {
// simple_log_with_func_name();
return { coro_handle::from_promise(*this) };
}
// we dont need suspend when first suspend
std::suspend_always initial_suspend() {
// simple_log_with_func_name();
return {};
}
// suspend always for the Task clean ups
std::suspend_always final_suspend() noexcept {
// simple_log_with_func_name();
if (parent_coroutine) {
simple_log("parent_coroutine will be wake up");
Schedular::instance().internal_spawn(parent_coroutine);
}
return {};
}
void return_value(T value) {
// simple_log_with_func_name(std::format("value T {} is received!", value));
cached_value = std::move(value);
}
void unhandled_exception() {
// process notings
}
};
bool await_ready() {
// simple_log_with_func_name();
return false; // always need suspend
}
void await_suspend(std::coroutine_handle<> h) {
// simple_log_with_func_name(); // Should never be here
simple_log("Current Routine will be suspend!");
coroutine_handle.promise().parent_coroutine = h;
simple_log("Child Routine will be called resume!");
Schedular::instance().internal_spawn(coroutine_handle);
}
T await_resume() {
// simple_log_with_func_name();
return coroutine_handle.promise().cached_value;
}
private:
coro_handle coroutine_handle;
private:
Task(const Task&) = delete;
Task& operator=(const Task&) = delete;
};
template <>
class Task<void> {
public:
friend class Schedular;
struct promise_type;
using coro_handle = std::coroutine_handle<promise_type>;
Task(coro_handle h)
: coroutine_handle(h) {
// simple_log_with_func_name();
}
~Task() {
// simple_log_with_func_name();
if (coroutine_handle) {
coroutine_handle.destroy();
}
}
Task(Task&& o)
: coroutine_handle(o.coroutine_handle) {
o.coroutine_handle = nullptr;
}
Task& operator=(Task&& o) {
coroutine_handle = std::move(o.coroutine_handle);
o.coroutine_handle = nullptr;
return *this;
}
bool await_ready() {
// simple_log_with_func_name();
return false; // always need suspend
}
void await_suspend(std::coroutine_handle<> h) {
// simple_log_with_func_name(); // Should never be here
simple_log("Current Routine will be suspend!");
coroutine_handle.promise().parent_coroutine = h;
simple_log("Child Routine will be called resume!");
Schedular::instance().internal_spawn(coroutine_handle);
}
void await_resume() {
// simple_log_with_func_name();
}
// concept requires
struct promise_type {
std::coroutine_handle<> parent_coroutine;
Task get_return_object() {
// simple_log_with_func_name();
return { coro_handle::from_promise(*this) };
}
// we need suspend when first suspend
std::suspend_always initial_suspend() {
// simple_log_with_func_name();
return {};
}
// suspend always for the Task clean ups
std::suspend_always final_suspend() noexcept {
// simple_log_with_func_name();
if (parent_coroutine) {
Schedular::instance().internal_spawn(parent_coroutine);
}
return {};
}
void return_void() {
// simple_log_with_func_name();
}
void unhandled_exception() {
// process notings
}
};
private:
coro_handle coroutine_handle;
private:
Task(const Task&) = delete;
Task& operator=(const Task&) = delete;
};
剩下的helpers.h/helpers.cpp和single_instance.hpp,笔者已经在正文给出了。不再重复。