58 lines
1.8 KiB
C++
58 lines
1.8 KiB
C++
#include <coroutine>
|
|
#include <iostream>
|
|
#include <stdexcept>
|
|
#include <string>
|
|
#include <type_traits>
|
|
|
|
namespace zt {
|
|
namespace demo{
|
|
template<class T>
|
|
struct Promise;
|
|
|
|
template<class T>
|
|
struct coroutine: std::coroutine_handle<Promise<T>>{
|
|
using promise_type = Promise<T>;
|
|
};
|
|
|
|
struct RAII{
|
|
std::string name;
|
|
RAII(const std::string &s):name(s){
|
|
std::cout<<__FUNCTION__<<" "<<s<<'\n';
|
|
}
|
|
~RAII(){
|
|
std::cout<<__FUNCTION__<<" "<<name<<'\n';
|
|
}
|
|
};
|
|
|
|
template<class T=void>
|
|
struct Promise{
|
|
Promise():raii("in promise"){};
|
|
RAII raii;
|
|
auto initial_suspend(){
|
|
std::cout<<__FUNCTION__<<'\n';
|
|
return std::suspend_always{};
|
|
// return std::suspend_never{};
|
|
}
|
|
auto final_suspend()noexcept{//最后要不要把句柄执行权限归还，如果不归还，那么协程被销毁，继续检查.done()则Segmentation fault
|
|
std::cout<<__FUNCTION__<<'\n';
|
|
return std::suspend_always{};
|
|
}
|
|
void unhandled_exception(){
|
|
std::cout<<__FUNCTION__<<'\n';
|
|
throw std::runtime_error("unhandled exeception\n");
|
|
}
|
|
void return_value(const std::decay_t<T> &t){
|
|
std::cout<<__FUNCTION__<<' '<<t<<'\n';
|
|
}
|
|
coroutine<T> get_return_object(){
|
|
std::cout<<__FUNCTION__<<'\n';
|
|
return {coroutine<T>::from_promise(*this)};
|
|
}
|
|
auto yield_value(const std::decay_t<T> &t){
|
|
std::cout<<__FUNCTION__<<' '<<t<<'\n';
|
|
return std::suspend_always{};
|
|
}
|
|
coroutine<T> h;
|
|
};
|
|
}
|
|
} |