適配異步代碼到 asio 協程

zuiwuchang

要理解asio的內部實現 非一日之功 孤亦不能完全掌握 且孤以爲 僅需知道其大概原理 之後 瞭解如何靈活使用即可
(這樣節省下的 許多時間 就可以做更多有意義的事 比如 聽一整天 阿梅 mui 的音樂)

要適配 異步到 asio 異常簡單 僅需要按部就班的 執行如下操作
1  創建一個 接受 boost::asio::yield_context& 參數的函數
2 在 函數中 使用 yield_context 創建一個 boost::asio::detail::async_result_init 型別 init
3 創建 init.handler 副本 coro
4 調用 異步代碼 在異步完成時 通過 調用 coro() 通知asio 喚醒協程
5 執行 return init.result.get(); 讓出cpu 並且等待 被喚醒時 返回 返回值

example 自定義了一個類似 asio::deadline_timer 的 async_wait 示例
(代碼儘量 保存簡介 以避免被 asio更多的 宏 模板 干擾 理解
如果是項目代碼 可能需要 修改 比如 像 asio一樣 返回值 型別 由調用傳入的yield_context 確認 而非此處固定爲 int )

1. template<typename Handler>
   
2. int async_wait(Handler& handler)
   
3. {
   
4.     //創建 異步流程
   
5.     boost::asio::detail::async_result_init<decltype(handler), void(int)> init(handler);
   
6. 
   
7.     //異步操作 需要傳入 init.handler 的副本
   
8.     boost::thread t([&,coro = init.handler]()
   
9.     {
   
10.         //模擬異步耗時
    
11.         boost::this_thread::sleep(ACTION_DURATION);
    
12. 
    
13.         //想 service 請求 響應
    
14.         handler.handler_.dispatcher_.dispatch(
    
15.             [&, coro = std::move(coro)]() mutable
    
16.         {
    
17.             //喚醒等待 攜程 並傳回 返回值
    
18.             coro(1);
    
19.         });
    
20.     });
    
21. 
    
22.     //讓出cpu 並等待 異步返回值
    
23.     return init.result.get();
    
24. }

复制代码

async_result_init 的第二個模板 參數 確定了 init.result.get() 返回值型別 比如 無返回值 可以 寫作 void(void)
handler.handler_.dispatcher_ 保存了 協程 執行的 io_service
需要使用 io_service.dispatch 將 協程 喚醒 發送到 io_service 中 而不能是異步代碼的 線程中


下面是完整代碼

1. #include <cstdio>
   
2. 
   
3. #include <boost/asio.hpp>
   
4. #include <boost/thread.hpp>
   
5. #include <boost/bind.hpp>
   
6. #include <boost/asio/deadline_timer.hpp>
   
7. #include <boost/asio/spawn.hpp>
   
8. #include <boost/smart_ptr.hpp>
   
9. 
   
10. //使用boost的 定時器 模擬異步操作
    
11. typedef boost::asio::deadline_timer deadline_timer_t;
    
12. typedef boost::shared_ptr<deadline_timer_t> deadline_timer_spt;
    
13. #define ACTION_DURATION boost::posix_time::seconds(1)
    
14. 
    
15. //協程函數
    
16. void coroutine(std::string name,deadline_timer_spt timer,boost::asio::yield_context ctx)
    
17. {
    
18.     boost::system::error_code e;
    
19.     int x = 0;
    
20. 
    
21.     //step 0
    
22.     timer->expires_from_now(ACTION_DURATION);
    
23.     std::cout<<name<<"step 0 value = "<<x++<<"\n";
    
24.     timer->async_wait(ctx[e]); //yield_context 重載了 [] 用於返回 異步返回值
    
25.     if(e) //獲取返回值
    
26.     {
    
27.         std::cout<<e<<"\n";
    
28.     }
    
29. 
    
30.     //step 1
    
31.     std::cout<<name<<"step 1 value = "<<x++<<"\n";
    
32.     timer->expires_from_now(ACTION_DURATION);
    
33.     timer->async_wait(ctx[e]);
    
34.     if(e)
    
35.     {
    
36.         std::cout<<e<<"\n";
    
37.     }
    
38. 
    
39.     //step 2
    
40.     std::cout<<name<<"step 2 value = "<<x++<<"\n";
    
41.     timer->expires_from_now(ACTION_DURATION);
    
42.     timer->async_wait(ctx[e]);
    
43.     if(e)
    
44.     {
    
45.         std::cout<<e<<"\n";
    
46.     }
    
47. }
    
48. 
    
49. 
    
50. template<typename Handler>
    
51. int async_wait(Handler& handler)
    
52. {
    
53.     //創建 異步流程
    
54.     boost::asio::detail::async_result_init<decltype(handler), void(int)> init(handler);
    
55. 
    
56.     //異步操作 需要傳入 init.handler 的副本
    
57.     boost::thread t([&,coro = init.handler]()
    
58.     {
    
59.         //模擬異步耗時
    
60.         boost::this_thread::sleep(ACTION_DURATION);
    
61. 
    
62.         //想 service 請求 響應
    
63.         handler.handler_.dispatcher_.dispatch(
    
64.             [&, coro = std::move(coro)]() mutable
    
65.         {
    
66.             //喚醒等待 攜程 並傳回 返回值
    
67.             coro(1);
    
68.         });
    
69.     });
    
70. 
    
71.     //讓出cpu 並等待 異步返回值
    
72.     return init.result.get();
    
73. }
    
74. void coroutine(std::string name,boost::asio::yield_context ctx)
    
75. {
    
76.     int x = 0;
    
77.     std::cout<<name<<"step 0 value = "<<x++<<"\n";
    
78.     async_wait(ctx);
    
79. 
    
80.     std::cout<<name<<"step 1 value = "<<x++<<"\n";
    
81.     async_wait(ctx);
    
82. 
    
83.     std::cout<<name<<"step 2 value = "<<x++<<"\n";
    
84.     async_wait(ctx);
    
85. 
    
86. }
    
87. int main(int argc, char* argv[])
    
88. {
    
89.     boost::asio::io_service service;
    
90. 
    
91.     //啓動兩個協程 在其中模擬 異步操作
    
92.     deadline_timer_spt timer0 = boost::make_shared<deadline_timer_t>(service,ACTION_DURATION);
    
93.     deadline_timer_spt timer1 = boost::make_shared<deadline_timer_t>(service,ACTION_DURATION);
    
94. 
    
95.     //deadline_timer_spt timer = boost::make_shared<deadline_timer_t>(service,boost::posix_time::hours(1));
    
96.     // timer->async_wait([](const boost::system::error_code& e) {});
    
97. 
    
98. 
    
99.     //  boost::asio::spawn(service,boost::bind(coroutine,"[one] ",timer0,_1));
    
100.     boost::asio::spawn(service,boost::bind(coroutine,"[my] ",_1));
     
101.     boost::asio::spawn(service,boost::bind(coroutine,"[two] ",timer1,_1));
     
102. 
     
103.     //運行asio服務
     
104.     service.run();
     
105. 
     
106.     return 0;
     
107. }

复制代码