基于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_when_all.cpp 3.1KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126
  1. 
  2. #include <chrono>
  3. #include <iostream>
  4. #include <string>
  5. #include <conio.h>
  6. #include <thread>
  7. #include <inttypes.h>
  8. #include "librf.h"
  9. #if _HAS_CXX17 || RESUMEF_USE_BOOST_ANY
  10. using namespace resumef;
  11. void test_when_any()
  12. {
  13. using namespace std::chrono;
  14. GO
  15. {
  16. auto vals = co_await when_any();
  17. vals = co_await when_any(
  18. []() ->future_t<int>
  19. {
  20. auto dt = rand() % 1000;
  21. co_await sleep_for(1ms * dt);
  22. std::cout << dt << "@a" << std::endl;
  23. co_return dt;
  24. }(),
  25. []() ->future_t<>
  26. {
  27. auto dt = rand() % 1000;
  28. co_await sleep_for(1ms * dt);
  29. std::cout << dt << "@b" << std::endl;
  30. }(),
  31. []() ->future_t<>
  32. {
  33. auto dt = rand() % 1000;
  34. co_await sleep_for(1ms * dt);
  35. std::cout << dt << "@c" << std::endl;
  36. }());
  37. if (vals.first == 0)
  38. std::cout << "first done! value is " << resumef::any_cast<int>(vals.second) << std::endl;
  39. else
  40. std::cout << "any done! index is " << vals.first << std::endl;
  41. co_await sleep_for(1010ms);
  42. std::cout << std::endl;
  43. auto my_sleep = [](const char * name) -> future_t<int>
  44. {
  45. auto dt = rand() % 1000;
  46. co_await sleep_for(1ms * dt);
  47. std::cout << dt << "@" << name << std::endl;
  48. co_return dt;
  49. };
  50. std::vector<future_t<int> > v{ my_sleep("g"), my_sleep("h"), my_sleep("i") };
  51. vals = co_await when_any(std::begin(v), std::end(v));
  52. std::cout << "any range done! index is " << vals.first << ", valus is " << resumef::any_cast<int>(vals.second) << std::endl;
  53. };
  54. this_scheduler()->run_until_notask();
  55. }
  56. void test_when_all()
  57. {
  58. using namespace std::chrono;
  59. auto my_sleep = [](const char * name) -> future_t<int>
  60. {
  61. auto dt = rand() % 1000;
  62. co_await sleep_for(1ms * dt);
  63. std::cout << dt << "@" << name << std::endl;
  64. co_return dt;
  65. };
  66. auto my_sleep_v = [](const char * name) -> future_t<>
  67. {
  68. auto dt = rand() % 1000;
  69. co_await sleep_for(1ms * dt);
  70. std::cout << dt << "@" << name << std::endl;
  71. };
  72. GO
  73. {
  74. co_await when_all();
  75. std::cout << "when all: zero!" << std::endl << std::endl;
  76. auto ab = co_await when_all(my_sleep("a"), my_sleep_v("b"));
  77. //ab.1 is std::ignore
  78. std::cout << "when all:" << std::get<0>(ab) << std::endl << std::endl;
  79. auto c = co_await my_sleep("c");
  80. std::cout << "when all:" << c << std::endl << std::endl;
  81. auto def = co_await when_all(my_sleep("d"), my_sleep_v("e"), my_sleep("f"));
  82. //def.1 is std::ignore
  83. std::cout << "when all:" << std::get<0>(def) << "," << std::get<2>(def) << std::endl << std::endl;
  84. std::vector<future_t<int> > v{ my_sleep("g"), my_sleep("h"), my_sleep("i") };
  85. auto vals = co_await when_all(std::begin(v), std::end(v));
  86. std::cout << "when all:" << vals[0] << "," << vals[1] << "," << vals[2] << "," << std::endl << std::endl;
  87. std::cout << "all range done!" << std::endl;
  88. };
  89. this_scheduler()->run_until_notask();
  90. }
  91. #endif
  92. void resumable_main_when_all()
  93. {
  94. #if _HAS_CXX17 || RESUMEF_USE_BOOST_ANY
  95. srand((uint32_t)time(nullptr));
  96. test_when_any();
  97. std::cout << std::endl;
  98. test_when_all();
  99. #endif
  100. }