基于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.

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. #include <chrono>
  2. #include <iostream>
  3. #include <string>
  4. #include <thread>
  5. #include "librf/librf.h"
  6. using namespace librf;
  7. template<class _Ctype>
  8. static void callback_get_long(int64_t val, _Ctype&& cb)
  9. {
  10. using namespace std::chrono;
  11. std::thread([val, cb = std::forward<_Ctype>(cb)]
  12. {
  13. std::this_thread::sleep_for(500ms);
  14. cb(val * val);
  15. }).detach();
  16. }
  17. //这种情况下,没有生成 frame-context,因此,并没有promise_type被内嵌在frame-context里
  18. static future_t<int64_t> async_get_long(int64_t val)
  19. {
  20. awaitable_t<int64_t> awaitable;
  21. callback_get_long(val, [awaitable](int64_t val)
  22. {
  23. awaitable.set_value(val);
  24. });
  25. return awaitable.get_future();
  26. }
  27. static future_t<int64_t> wait_get_long(int64_t val)
  28. {
  29. val = co_await async_get_long(val);
  30. co_return val;
  31. }
  32. //这种情况下,会生成对应的 frame-context,一个promise_type被内嵌在frame-context里
  33. static future_t<int64_t> resumable_get_long(int64_t val)
  34. {
  35. std::cout << val << std::endl;
  36. val = co_await wait_get_long(val);
  37. std::cout << val << std::endl;
  38. val = co_await wait_get_long(val);
  39. std::cout << val << std::endl;
  40. val = co_await wait_get_long(val);
  41. std::cout << val << std::endl;
  42. co_return val;
  43. }
  44. static future_t<int64_t> loop_get_long(int64_t val)
  45. {
  46. std::cout << val << std::endl;
  47. for (int i = 0; i < 5; ++i)
  48. {
  49. val = co_await async_get_long(val);
  50. std::cout << val << std::endl;
  51. }
  52. co_return val;
  53. }
  54. static future_t<std::string&> async_get_string(std::string & ref_string)
  55. {
  56. awaitable_t<std::string&> awaitable;
  57. callback_get_long(std::stoi(ref_string), [awaitable, &ref_string](int64_t val)
  58. {
  59. ref_string = std::to_string(val);
  60. awaitable.set_value(ref_string);
  61. });
  62. return awaitable.get_future();
  63. }
  64. static future_t<std::string&> resumable_get_string(std::string& val)
  65. {
  66. std::cout << val << std::endl;
  67. val = co_await async_get_string(val);
  68. std::cout << val << std::endl;
  69. val = co_await async_get_string(val);
  70. std::cout << val << std::endl;
  71. val = co_await async_get_string(val);
  72. std::cout << val << std::endl;
  73. co_return static_cast<std::string&>(val);
  74. }
  75. void resumable_main_cb()
  76. {
  77. std::cout << __FUNCTION__ << std::endl;
  78. //由于使用者可能不能明确的区分是resume function返回的awaitor还是awaitable function返回的awaitor
  79. //导致均有可能加入到协程里去调度。
  80. //所以,协程调度器应该需要能处理这种情况。
  81. go async_get_long(3);
  82. this_scheduler()->run_until_notask();
  83. std::string ref_string{"2"};
  84. go resumable_get_string(ref_string);
  85. this_scheduler()->run_until_notask();
  86. GO
  87. {
  88. auto val = co_await resumable_get_long(2);
  89. std::cout << "GO:" << val << std::endl;
  90. };
  91. go loop_get_long(3);
  92. this_scheduler()->run_until_notask();
  93. }
  94. #if LIBRF_TUTORIAL_STAND_ALONE
  95. int main()
  96. {
  97. resumable_main_cb();
  98. return 0;
  99. }
  100. #endif