c++ 协程

#include <coroutine>
#include <iostream>
#include <optional>

// 定义一个协程的承诺类型
struct MyCoroutinePromise {
    int current_value;

    MyCoroutinePromise() : current_value(0) {}

    // 返回挂起对象
    std::suspend_always initial_suspend() { return {}; }
    std::suspend_always final_suspend() noexcept { return {}; }

    // 返回值处理
    void return_value(int value) {
        current_value = value;
    }

    // 异常处理
    void unhandled_exception() {
        std::terminate();
    }

    // 获取协程句柄类型
    MyCoroutineHandle get_return_object() {
        return MyCoroutineHandle(this);
    }
};

// 定义协程句柄类型
using MyCoroutineHandle = std::coroutine_handle<MyCoroutinePromise>;

// 定义协程类型
struct MyCoroutine {
    MyCoroutine(MyCoroutineHandle handle) : coro(handle) {}
    ~MyCoroutine() {
        if (coro) {
            coro.destroy();
        }
    }

    bool next() {
        if (!coro.done()) {
            coro.resume();
            return !coro.done();
        }
        return false;
    }

    int value() const {
        return coro.promise().current_value;
    }

private:
    MyCoroutineHandle coro;
};

// 协程函数
MyCoroutine async_function() {
    co_await std::suspend_always{};
    co_return 42;
}

int main() {
    MyCoroutine coro = async_function();

    while (coro.next()) {
        std::cout << "Coroutine value: " << coro.value() << std::endl;
    }

    return 0;
}

解释说明

1. 定义承诺类型 (MyCoroutinePromise):

   • 承诺类型是协程的核心部分，它定义了协程的行为。包括初始挂起、最终挂起、返回值处理和异常处理等。

2. 定义协程句柄类型 (MyCoroutineHandle):

   • 使用 std::coroutine_handle 来管理协程的状态和控制流。

3. 定义协程类型 (MyCoroutine):

   • 包装协程句柄，提供简单的接口来操作协程（如 next() 和 value()）。

4. 协程函数 (async_function):

   • 这是一个简单的协程函数，使用 co_await 挂起自身，并在恢复时返回值 42。

5. 主函数 (main):

   • 创建并驱动协程，输出协程的返回值。

这个示例展示了如何使用 C++20 的协程特性来创建和操作简单的协程。