基于C++ Coroutines提案 ‘Stackless Resumable Functions’编写的协程库
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

test_async_modern_cb.cpp 15KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423
  1. #include "librf.h"
  2. #include <chrono>
  3. #include <iostream>
  4. #include <string>
  5. #include <thread>
  6. #include <future>
  7. //原旨主义的异步函数,其回调写法大致如下
  8. template<typename _Input_t, typename _Callable_t>
  9. void tostring_async_originalism(_Input_t&& value, _Callable_t&& token)
  10. {
  11. std::thread([callback = std::move(token), value = std::forward<_Input_t>(value)]
  12. {
  13. callback(std::to_string(value));
  14. }).detach();
  15. }
  16. //使用原旨主义的方式扩展异步方法来支持future
  17. template<typename _Input_t>
  18. auto tostring_async_originalism_future(_Input_t&& value)
  19. {
  20. std::promise<std::string> _promise;
  21. std::future<std::string> _future = _promise.get_future();
  22. std::thread([_promise = std::move(_promise), value = std::forward<_Input_t>(value)]() mutable
  23. {
  24. _promise.set_value(std::to_string(value));
  25. }).detach();
  26. return std::move(_future);
  27. }
  28. //----------------------------------------------------------------------------------------------------------------------
  29. //以下演示如何通过现代回调(Modern Callback), 使用回调适配器模型,
  30. //将异步回调函数扩展到支持future模式,调用链模式,以及协程。
  31. //首先,准备modern_call_return_void_t和modern_callback_adapter_t给异步函数使用
  32. //通过一个间接的类来解决返回void的语法问题,以便于优化返回值
  33. struct modern_call_return_void_t
  34. {
  35. void get(){}
  36. };
  37. //回调适配器的模板类
  38. //_Callable_t 要符合 _Signature_t 签名
  39. //这个类除了转移token外,不做任何有效的工作
  40. //有效工作等待特列化的类去做
  41. template<typename _Callable_t, typename _Signature_t>
  42. struct modern_callback_adapter_t
  43. {
  44. using callback_type = _Callable_t;
  45. using return_type = modern_call_return_void_t;
  46. static std::tuple<callback_type, return_type> traits(_Callable_t&& token)
  47. {
  48. return { std::forward<_Callable_t>(token), {} };
  49. }
  50. };
  51. //一个使用回调处理结果的异步函数,会涉及以下概念:
  52. //_Input_t:异步函数的输入参数;
  53. //_Signature_t: 此异步回调的函数签名;应当满足‘void(_Exception_t, _Result_t...)’或者‘void(_Result_t...)’类型;
  54. //_Callable_t:回调函数或标记,如果是回调函数,则需要符合_Signature_t的签名类型。这个回调,必须调用一次,且只能调用一次;
  55. //_Return_t:异步函数的返回值;
  56. //_Result_t...:异步函数完成后的结果值,作为回调函数的入参部分;这个参数可以有零至多个;
  57. //_Exception_t:回调函数的异常, 如果不喜欢异常的则忽略这个部分,但就得异步代码将异常处置妥当;
  58. //
  59. //在回调适配器模型里,_Input_t/_Result_t/_Exception_t(可选)是异步函数提供的功能所固有的部分;_Callable_t/_Return_t
  60. //部分并不直接使用,而是通过适配器去另外处理。这样给予适配器一次扩展到future模式,调用链模式的机会,以及支持协程的机会。
  61. //
  62. //tostring_async 演示了在其他线程里,将_Input_t的输入值,转化为std::string类型的_Result_t。
  63. //然后调用_Signature_t为 ‘void(std::string &&)’ 类型的 _Callable_t。
  64. //忽视异常处理,故没有_Exception_t。
  65. //
  66. template<typename _Input_t, typename _Callable_t>
  67. auto tostring_async(_Input_t&& value, _Callable_t&& token)
  68. {
  69. //适配器类型
  70. using _Adapter_t = modern_callback_adapter_t<typename resumef::remove_cvref_t<_Callable_t>, void(std::string)>;
  71. //通过适配器获得兼容_Signature_t类型的真正的回调,以及返回值_Return_t
  72. auto adapter = _Adapter_t::traits(std::forward<_Callable_t>(token));
  73. //callback与token未必是同一个变量,甚至未必是同一个类型
  74. std::thread([callback = std::move(std::get<0>(adapter)), value = std::forward<_Input_t>(value)]
  75. {
  76. using namespace std::literals;
  77. std::this_thread::sleep_for(0.1s);
  78. callback(std::to_string(value));
  79. }).detach();
  80. //返回适配器的_Return_t变量
  81. return std::move(std::get<1>(adapter)).get();
  82. }
  83. //或者宏版本写法
  84. #define MODERN_CALLBACK_TRAITS(_Token_value, _Signature_t) \
  85. using _Adapter_t = modern_callback_adapter_t<typename resumef::remove_cvref_t<_Callable_t>, _Signature_t>; \
  86. auto _Adapter_value = _Adapter_t::traits(std::forward<_Callable_t>(_Token_value))
  87. #define MODERN_CALLBACK_CALL() std::move(std::get<0>(_Adapter_value))
  88. #define MODERN_CALLBACK_RETURN() return std::move(std::get<1>(_Adapter_value)).get()
  89. template<typename _Input_t, typename _Callable_t>
  90. auto tostring_async_macro(_Input_t&& value, _Callable_t&& token)
  91. {
  92. MODERN_CALLBACK_TRAITS(token, void(std::string));
  93. std::thread([callback = MODERN_CALLBACK_CALL(), value = std::forward<_Input_t>(value)]
  94. {
  95. callback(std::to_string(value));
  96. }).detach();
  97. MODERN_CALLBACK_RETURN();
  98. }
  99. //----------------------------------------------------------------------------------------------------------------------
  100. //下面演示如何扩展tostring_async函数,以支持future模式
  101. //future库有多种,但应当都提供遵循promise/future对,兼容std::promise/std::future用法
  102. //这样的话,可以做一个更加通用的支持future的callback类
  103. //实现use_future_callback_t的基类,避免写一些重复代码
  104. template<typename _Promise_traits, typename _Result_t>
  105. struct use_future_callback_base_t
  106. {
  107. //回调函数的结果类型,已经排除掉了异常参数
  108. using result_type = _Result_t;
  109. //通过_Promise_traits获取真正的promise类型
  110. using promise_type = typename _Promise_traits::template promise_type<result_type>;
  111. //此类持有一个std::promise<_Result_t>,便于设置值和异常
  112. //而将与promise关联的future作为返回值_Return_t,让tostring_async返回。
  113. mutable promise_type _promise;
  114. auto get_future() const
  115. {
  116. return this->_promise.get_future();
  117. }
  118. };
  119. //此类的实例作为真正的callback,交给异步回调函数,替换token。
  120. //在实际应用中,需要针对是否有异常参数,结果值为0,1,多个等情况做特殊处理,故还需要通过更多的偏特化版本来支持。
  121. //具体的异常参数,需要根据实际应用去特里化。这里仅演示通过std::exception_ptr作为异常传递的情况。
  122. template<typename...>
  123. struct use_future_callback_t;
  124. //无异常,无结果的callback类型:void()
  125. template<typename _Promise_traits>
  126. struct use_future_callback_t<_Promise_traits> : public use_future_callback_base_t<_Promise_traits, void>
  127. {
  128. using use_future_callback_base_t<_Promise_traits, void>::use_future_callback_base_t;
  129. void operator()() const
  130. {
  131. this->_promise.set_value();
  132. }
  133. };
  134. //有异常,无结果的callback类型:void(exception_ptr)
  135. template<typename _Promise_traits>
  136. struct use_future_callback_t<_Promise_traits, std::exception_ptr> : public use_future_callback_base_t<_Promise_traits, void>
  137. {
  138. using use_future_callback_base_t<_Promise_traits, void>::use_future_callback_base_t;
  139. void operator()(std::exception_ptr eptr) const
  140. {
  141. if (!eptr)
  142. this->_promise.set_value();
  143. else
  144. this->_promise.set_exception(std::move(eptr));
  145. }
  146. };
  147. //无异常,单结果的callback类型:void(_Result_t)
  148. template<typename _Promise_traits, typename _Result_t>
  149. struct use_future_callback_t<_Promise_traits, _Result_t> : public use_future_callback_base_t<_Promise_traits, _Result_t>
  150. {
  151. using use_future_callback_base_t<_Promise_traits, _Result_t>::use_future_callback_base_t;
  152. template<typename Arg>
  153. void operator()(Arg && arg) const
  154. {
  155. this->_promise.set_value(std::forward<Arg>(arg));
  156. }
  157. };
  158. //有异常,单结果的callback类型:void(std::exception_ptr, _Result_t)
  159. template<typename _Promise_traits, typename _Result_t>
  160. struct use_future_callback_t<_Promise_traits, std::exception_ptr, _Result_t> : public use_future_callback_base_t<_Promise_traits, _Result_t>
  161. {
  162. using use_future_callback_base_t<_Promise_traits, _Result_t>::use_future_callback_base_t;
  163. template<typename Arg>
  164. void operator()(std::exception_ptr eptr, Arg && arg) const
  165. {
  166. if (!eptr)
  167. this->_promise.set_value(std::forward<Arg>(arg));
  168. else
  169. this->_promise.set_exception(std::move(eptr));
  170. }
  171. };
  172. //无异常,多结果的callback类型:void(_Result_t...)
  173. template<typename _Promise_traits, typename... _Result_t>
  174. struct use_future_callback_t<_Promise_traits, _Result_t...> : public use_future_callback_base_t<_Promise_traits, std::tuple<_Result_t...> >
  175. {
  176. using use_future_callback_base_t<_Promise_traits, std::tuple<_Result_t...> >::use_future_callback_base_t;
  177. template<typename... Args>
  178. void operator()(Args&&... args) const
  179. {
  180. static_assert(sizeof...(Args) == sizeof...(_Result_t), "");
  181. this->_promise.set_value(std::make_tuple(std::forward<Args>(args)...));
  182. }
  183. };
  184. //有异常,多结果的callback类型:void(std::exception_ptr, _Result_t...)
  185. template <typename _Promise_traits, typename... _Result_t>
  186. struct use_future_callback_t<_Promise_traits, std::exception_ptr, _Result_t...> : public use_future_callback_base_t<_Promise_traits, std::tuple<_Result_t...> >
  187. {
  188. using use_future_callback_base_t<_Promise_traits, std::tuple<_Result_t...> >::use_future_callback_base_t;
  189. template<typename... Args>
  190. void operator()(std::exception_ptr eptr, Args&&... args) const
  191. {
  192. static_assert(sizeof...(Args) == sizeof...(_Result_t), "");
  193. if (!eptr)
  194. this->_promise.set_value(std::make_tuple(std::forward<Args>(args)...));
  195. else
  196. this->_promise.set_exception(std::move(eptr));
  197. }
  198. };
  199. //与use_future_callback_t配套的获得_Return_t的类
  200. template<typename _Future_traits, typename _Result_t>
  201. struct use_future_return_t
  202. {
  203. using result_type = _Result_t;
  204. using future_type = typename _Future_traits::template future_type<result_type>;
  205. future_type _future;
  206. use_future_return_t(future_type && ft)
  207. : _future(std::move(ft)) {}
  208. future_type get()
  209. {
  210. return std::move(_future);
  211. }
  212. };
  213. //利用use_future_callback_t + use_future_return_t 实现的callback适配器
  214. template<typename _Token_as_callable_t, typename... _Result_t>
  215. struct modern_callback_adapter_impl_t
  216. {
  217. using traits_type = _Token_as_callable_t;
  218. using callback_type = use_future_callback_t<traits_type, _Result_t...>;
  219. using result_type = typename callback_type::result_type;
  220. using return_type = use_future_return_t<traits_type, result_type>;
  221. static std::tuple<callback_type, return_type> traits(const _Token_as_callable_t& /*没人关心这个变量*/)
  222. {
  223. callback_type callback{};
  224. auto future = callback.get_future();
  225. return { std::move(callback), std::move(future) };
  226. }
  227. };
  228. //----------------------------------------------------------------------------------------------------------------------
  229. //一、做一个使用std::promise/std::future的辅助类。
  230. //这个类还负责萃取promise/future对的类型。
  231. struct std_future_t
  232. {
  233. template<typename _Result_t>
  234. using promise_type = std::promise<_Result_t>;
  235. template<typename _Result_t>
  236. using future_type = std::future<_Result_t>;
  237. };
  238. //二、申明这个辅助类的全局变量。不申明这个变量也行,就是每次要写use_future_t{},麻烦些。
  239. //以后就使用std_future,替代tostring_async的token参数了。
  240. //这个参数其实不需要实质传参,最后会被编译器优化没了。
  241. //仅仅是要指定_Callable_t的类型为std_future_t,
  242. //从而在tostring_async函数内,使用偏特化的modern_callback_adapter_t<std_future_t, ...>而已。
  243. constexpr std_future_t std_future{};
  244. //三、偏特化_Callable_t为std_future_t类型的modern_callback_adapter_t
  245. //真正的回调类型是use_future_callback_t,返回类型_Return_t是use_future_return_t。
  246. //配合use_future_callback_t的promise<result_type>,和use_future_return_t的future<result_type>,正好组成一对promise/future对。
  247. //promise在真正的回调里设置结果值;
  248. //future返回给调用者获取结果值。
  249. template<typename R, typename... _Result_t>
  250. struct modern_callback_adapter_t<std_future_t, R(_Result_t...)> : public modern_callback_adapter_impl_t<std_future_t, _Result_t...>
  251. {
  252. };
  253. //----------------------------------------------------------------------------------------------------------------------
  254. //同理,可以制作支持C++20的协程的下列一系列类(其实,这才是我的最终目的)
  255. struct use_librf_t
  256. {
  257. template<typename _Result_t>
  258. using promise_type = resumef::awaitable_t<_Result_t>;
  259. template<typename _Result_t>
  260. using future_type = resumef::future_t<_Result_t>;
  261. };
  262. constexpr use_librf_t use_librf{};
  263. template<typename R, typename... _Result_t>
  264. struct modern_callback_adapter_t<use_librf_t, R(_Result_t...)> : public modern_callback_adapter_impl_t<use_librf_t, _Result_t...>
  265. {
  266. };
  267. //所以,我现在的看法是,支持异步操作的库,尽可能如此设计回调。这样便于支持C++20的协程。以及future::then这样的任务链。
  268. //这才是“摩登C++”!
  269. //----------------------------------------------------------------------------------------------------------------------
  270. //使用范例
  271. //演示异步库有多个异步回调函数,只要按照Modern Callback范式去做回调,就不再需要写额外的代码,就可以适配到future+librf,以及更多的其他库
  272. template<typename _Ty1, typename _Ty2, typename _Callable_t>
  273. auto add_async(_Ty1&& val1, _Ty2&& val2, _Callable_t&& token)
  274. {
  275. MODERN_CALLBACK_TRAITS(token, void(decltype(val1 + val2)));
  276. std::thread([=, callback = MODERN_CALLBACK_CALL()]
  277. {
  278. using namespace std::literals;
  279. std::this_thread::sleep_for(0.1s);
  280. callback(val1 + val2);
  281. }).detach();
  282. MODERN_CALLBACK_RETURN();
  283. }
  284. //演示异步库有多个异步回调函数,只要按照Modern Callback范式去做回调,就不再需要写额外的代码,就可以适配到future+librf,以及更多的其他库
  285. template<typename _Ty1, typename _Ty2, typename _Callable_t>
  286. auto muldiv_async(_Ty1&& val1, _Ty2&& val2, _Callable_t&& token)
  287. {
  288. MODERN_CALLBACK_TRAITS(token, void(std::exception_ptr, decltype(val1 * val2), decltype(val1 / val2)));
  289. std::thread([=, callback = MODERN_CALLBACK_CALL()]
  290. {
  291. using namespace std::literals;
  292. std::this_thread::sleep_for(0.1s);
  293. auto v1 = val1 * val2;
  294. if (val2 == 0)
  295. callback(std::make_exception_ptr(std::logic_error("divided by zero")), v1, 0);
  296. else
  297. callback(nullptr, v1, val1 / val2);
  298. }).detach();
  299. MODERN_CALLBACK_RETURN();
  300. }
  301. void resumable_main_modern_cb()
  302. {
  303. using namespace std::literals;
  304. //使用lambda作为异步回调函数,传统用法
  305. tostring_async_originalism(-1.0, [](std::string && value)
  306. {
  307. std::cout << value << std::endl;
  308. });
  309. std::this_thread::sleep_for(0.5s);
  310. tostring_async(1.0, [](std::string && value)
  311. {
  312. std::cout << value << std::endl;
  313. });
  314. std::this_thread::sleep_for(0.5s);
  315. std::cout << "......" << std::endl;
  316. //支持future的用法
  317. std::future<std::string> f1 = tostring_async_originalism_future(5);
  318. std::cout << f1.get() << std::endl;
  319. std::future<std::string> f2 = tostring_async(6.0f, std_future);
  320. std::cout << f2.get() << std::endl;
  321. //支持librf的用法
  322. GO
  323. {
  324. #ifndef __clang__
  325. try
  326. #endif
  327. {
  328. int val = co_await add_async(1, 2, use_librf);
  329. std::cout << val << std::endl;
  330. //muldiv_async函数可能会抛异常,取决于val是否是0
  331. //异常将会带回到本协程里的代码,所以需要try-catch
  332. auto ab = co_await muldiv_async(9, val, use_librf);
  333. //C++17:
  334. //auto [a, b] = co_await muldiv_async(9, val, use_librf);
  335. std::string result = co_await tostring_async(std::get<0>(ab) + std::get<1>(ab), use_librf);
  336. std::cout << result << std::endl;
  337. }
  338. #ifndef __clang__
  339. catch (const std::exception& e)
  340. {
  341. std::cout << "exception signal : " << e.what() << std::endl;
  342. }
  343. catch (...)
  344. {
  345. std::cout << "exception signal : who knows?" << std::endl;
  346. }
  347. #endif
  348. };
  349. resumef::this_scheduler()->run_until_notask();
  350. }