boost::asio 异步主动连接多个服务器

ATC.hpp

1. #include <boost/asio/deadline_timer.hpp>
   
2. #include <boost/asio/io_service.hpp>
   
3. #include <boost/asio/ip/tcp.hpp>
   
4. #include <boost/asio/read_until.hpp>
   
5. #include <boost/asio/streambuf.hpp>
   
6. #include <boost/asio/write.hpp>
   
7. #include <boost/asio.hpp>
   
8. #include <boost/bind.hpp>
   
9. #include <boost/shared_ptr.hpp>
   
10. #include <boost/enable_shared_from_this.hpp>
    
11. #include <iostream>
    
12. #include <set>
    
13. #include <map>
    
14. #include "Message.hpp"
    
15. #include "SQLite.hpp"
    
16. 
    
17. using boost::asio::deadline_timer;
    
18. using boost::asio::ip::tcp;
    
19. using boost::asio::ip::udp;
    
20. 
    
21. using namespace std;
    
22. using namespace PWRD;
    
23. using namespace PWRD::SQL;
    
24. using namespace PWRD::Protocol;
    
25. 
    
26. typedef map<string, int> Peers;
    
27. typedef map<string, int> Stats;
    
28. typedef map<string, string> Files;
    
29. typedef map<string, tcp::socket* > Sockets;
    
30. #define DEADLINE 10
    
31. #define MAXLEN 1024
    
32. #define SECONDS 5
    
33. 
    
34. namespace PWRD{
    
35.     namespace Network{
    
36.         class Session{
    
37.             public:
    
38.                 Session(boost::asio::ip::tcp::socket &_socket, boost::asio::io_service &io_service, const char *path);
    
39.                 ~Session();
    
40.                 void start();
    
41. 
    
42.                 void stop();
    
43.                 void delivery(string &ip);
    
44. 
    
45.             private:
    
46.                 void handle_connect(const boost::system::error_code& ec,
    
47.                         tcp::resolver::iterator endpoint_iter);
    
48. 
    
49.                 void start_read();
    
50. 
    
51.                 void handle_read_header(const boost::system::error_code& ec);
    
52. 
    
53.                 void start_delivery();
    
54. 
    
55.                 void handle_read_body(const boost::system::error_code& ec);
    
56. 
    
57.                 void start_write_heartbeat();
    
58. 
    
59.                 void handle_write_heartbeat(const boost::system::error_code& ec);
    
60.                 void handle_wait(const boost::system::error_code& ec);
    
61.                 void handle_write_config(const boost::system::error_code& ec, Configure &conf);
    
62. 
    
63.             private:
    
64.                 bool stopped_;
    
65.                 tcp::socket &socket_;
    
66.                 Header header_;
    
67.                 HeartbeatMessage heartbeat_message_;
    
68.                 DataMessage packet_message_;
    
69.                 ConfigMessage conf_message_;
    
70.                 deadline_timer heartbeat_timer_;
    
71.                 Callback *callback_;
    
72.                 char *input_stream_;
    
73. 
    
74.         };
    
75. 
    
76.         class SessionPool{
    
77.             public:
    
78.                 SessionPool();
    
79.                 ~SessionPool();
    
80.                 void join(string&, Session *);
    
81.                 void delivery(string&);
    
82.                 void leave(string &ip);
    
83. 
    
84.             private: