]>
Commit | Line | Data |
---|---|---|
1 | // | |
2 | // server.cpp | |
3 | // ~~~~~~~~~~ | |
4 | // | |
5 | // Copyright (c) 2003-2017 Christopher M. Kohlhoff (chris at kohlhoff dot com) | |
6 | // | |
7 | // Distributed under the Boost Software License, Version 1.0. (See accompanying | |
8 | // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) | |
9 | // | |
10 | ||
11 | #include <boost/asio.hpp> | |
12 | #include <boost/bind.hpp> | |
13 | #include <boost/lexical_cast.hpp> | |
14 | #include <iostream> | |
15 | #include <vector> | |
16 | #include "connection.hpp" // Must come before boost/serialization headers. | |
17 | #include <boost/serialization/vector.hpp> | |
18 | #include "stock.hpp" | |
19 | ||
20 | namespace s11n_example { | |
21 | ||
22 | /// Serves stock quote information to any client that connects to it. | |
23 | class server | |
24 | { | |
25 | public: | |
26 | /// Constructor opens the acceptor and starts waiting for the first incoming | |
27 | /// connection. | |
28 | server(boost::asio::io_context& io_context, unsigned short port) | |
29 | : acceptor_(io_context, | |
30 | boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port)) | |
31 | { | |
32 | // Create the data to be sent to each client. | |
33 | stock s; | |
34 | s.code = "ABC"; | |
35 | s.name = "A Big Company"; | |
36 | s.open_price = 4.56; | |
37 | s.high_price = 5.12; | |
38 | s.low_price = 4.33; | |
39 | s.last_price = 4.98; | |
40 | s.buy_price = 4.96; | |
41 | s.buy_quantity = 1000; | |
42 | s.sell_price = 4.99; | |
43 | s.sell_quantity = 2000; | |
44 | stocks_.push_back(s); | |
45 | s.code = "DEF"; | |
46 | s.name = "Developer Entertainment Firm"; | |
47 | s.open_price = 20.24; | |
48 | s.high_price = 22.88; | |
49 | s.low_price = 19.50; | |
50 | s.last_price = 19.76; | |
51 | s.buy_price = 19.72; | |
52 | s.buy_quantity = 34000; | |
53 | s.sell_price = 19.85; | |
54 | s.sell_quantity = 45000; | |
55 | stocks_.push_back(s); | |
56 | ||
57 | // Start an accept operation for a new connection. | |
58 | connection_ptr new_conn(new connection(acceptor_.get_io_context())); | |
59 | acceptor_.async_accept(new_conn->socket(), | |
60 | boost::bind(&server::handle_accept, this, | |
61 | boost::asio::placeholders::error, new_conn)); | |
62 | } | |
63 | ||
64 | /// Handle completion of a accept operation. | |
65 | void handle_accept(const boost::system::error_code& e, connection_ptr conn) | |
66 | { | |
67 | if (!e) | |
68 | { | |
69 | // Successfully accepted a new connection. Send the list of stocks to the | |
70 | // client. The connection::async_write() function will automatically | |
71 | // serialize the data structure for us. | |
72 | conn->async_write(stocks_, | |
73 | boost::bind(&server::handle_write, this, | |
74 | boost::asio::placeholders::error, conn)); | |
75 | } | |
76 | ||
77 | // Start an accept operation for a new connection. | |
78 | connection_ptr new_conn(new connection(acceptor_.get_io_context())); | |
79 | acceptor_.async_accept(new_conn->socket(), | |
80 | boost::bind(&server::handle_accept, this, | |
81 | boost::asio::placeholders::error, new_conn)); | |
82 | } | |
83 | ||
84 | /// Handle completion of a write operation. | |
85 | void handle_write(const boost::system::error_code& e, connection_ptr conn) | |
86 | { | |
87 | // Nothing to do. The socket will be closed automatically when the last | |
88 | // reference to the connection object goes away. | |
89 | } | |
90 | ||
91 | private: | |
92 | /// The acceptor object used to accept incoming socket connections. | |
93 | boost::asio::ip::tcp::acceptor acceptor_; | |
94 | ||
95 | /// The data to be sent to each client. | |
96 | std::vector<stock> stocks_; | |
97 | }; | |
98 | ||
99 | } // namespace s11n_example | |
100 | ||
101 | int main(int argc, char* argv[]) | |
102 | { | |
103 | try | |
104 | { | |
105 | // Check command line arguments. | |
106 | if (argc != 2) | |
107 | { | |
108 | std::cerr << "Usage: server <port>" << std::endl; | |
109 | return 1; | |
110 | } | |
111 | unsigned short port = boost::lexical_cast<unsigned short>(argv[1]); | |
112 | ||
113 | boost::asio::io_context io_context; | |
114 | s11n_example::server server(io_context, port); | |
115 | io_context.run(); | |
116 | } | |
117 | catch (std::exception& e) | |
118 | { | |
119 | std::cerr << e.what() << std::endl; | |
120 | } | |
121 | ||
122 | return 0; | |
123 | } |