| 1 | /*
 | 
|---|
| 2 |  * StockServer.cpp
 | 
|---|
| 3 |  *
 | 
|---|
| 4 |  *  Created on: 18.11.2011
 | 
|---|
| 5 |  *      Author: heber
 | 
|---|
| 6 |  */
 | 
|---|
| 7 | 
 | 
|---|
| 8 | #include <boost/asio.hpp>
 | 
|---|
| 9 | #include <boost/bind.hpp>
 | 
|---|
| 10 | #include <boost/lexical_cast.hpp>
 | 
|---|
| 11 | #include <iostream>
 | 
|---|
| 12 | #include <vector>
 | 
|---|
| 13 | #include "connection.hpp" // Must come before boost/serialization headers.
 | 
|---|
| 14 | #include <boost/serialization/vector.hpp>
 | 
|---|
| 15 | #include "stock.hpp"
 | 
|---|
| 16 | 
 | 
|---|
| 17 | #include "StockServer.hpp"
 | 
|---|
| 18 | 
 | 
|---|
| 19 | using namespace s11n_example;
 | 
|---|
| 20 | 
 | 
|---|
| 21 | StockServer::StockServer(boost::asio::io_service& io_service, unsigned short port) :
 | 
|---|
| 22 |   acceptor_(io_service,
 | 
|---|
| 23 |       boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port)
 | 
|---|
| 24 |   )
 | 
|---|
| 25 | {
 | 
|---|
| 26 |   // Create the data to be sent to each client.
 | 
|---|
| 27 |   stock s;
 | 
|---|
| 28 |   s.code = "ABC";
 | 
|---|
| 29 |   s.name = "A Big Company";
 | 
|---|
| 30 |   s.open_price = 4.56;
 | 
|---|
| 31 |   s.high_price = 5.12;
 | 
|---|
| 32 |   s.low_price = 4.33;
 | 
|---|
| 33 |   s.last_price = 4.98;
 | 
|---|
| 34 |   s.buy_price = 4.96;
 | 
|---|
| 35 |   s.buy_quantity = 1000;
 | 
|---|
| 36 |   s.sell_price = 4.99;
 | 
|---|
| 37 |   s.sell_quantity = 2000;
 | 
|---|
| 38 |   stocks_.push_back(s);
 | 
|---|
| 39 |   s.code = "DEF";
 | 
|---|
| 40 |   s.name = "Developer Entertainment Firm";
 | 
|---|
| 41 |   s.open_price = 20.24;
 | 
|---|
| 42 |   s.high_price = 22.88;
 | 
|---|
| 43 |   s.low_price = 19.50;
 | 
|---|
| 44 |   s.last_price = 19.76;
 | 
|---|
| 45 |   s.buy_price = 19.72;
 | 
|---|
| 46 |   s.buy_quantity = 34000;
 | 
|---|
| 47 |   s.sell_price = 19.85;
 | 
|---|
| 48 |   s.sell_quantity = 45000;
 | 
|---|
| 49 |   stocks_.push_back(s);
 | 
|---|
| 50 | 
 | 
|---|
| 51 |   // Start an accept operation for a new connection.
 | 
|---|
| 52 |   connection_ptr new_conn(new connection(acceptor_.get_io_service()));
 | 
|---|
| 53 |   acceptor_.async_accept(new_conn->socket(),
 | 
|---|
| 54 |     boost::bind(&StockServer::handle_accept, this,
 | 
|---|
| 55 |       boost::asio::placeholders::error, new_conn));
 | 
|---|
| 56 | }
 | 
|---|
| 57 | 
 | 
|---|
| 58 | void StockServer::handle_accept(const boost::system::error_code& e, connection_ptr conn)
 | 
|---|
| 59 | {
 | 
|---|
| 60 |   std::cout << "handle_accept called." << std::endl;
 | 
|---|
| 61 |   if (!e)
 | 
|---|
| 62 |   {
 | 
|---|
| 63 |     // Successfully accepted a new connection. Send the list of stocks to the
 | 
|---|
| 64 |     // client. The connection::async_write() function will automatically
 | 
|---|
| 65 |     // serialize the data structure for us.
 | 
|---|
| 66 |     conn->async_write(stocks_,
 | 
|---|
| 67 |       boost::bind(&StockServer::handle_write, this,
 | 
|---|
| 68 |       boost::asio::placeholders::error, conn));
 | 
|---|
| 69 | }
 | 
|---|
| 70 | 
 | 
|---|
| 71 |   connection_ptr new_conn(new connection(acceptor_.get_io_service()));
 | 
|---|
| 72 |   acceptor_.async_accept(new_conn->socket(),
 | 
|---|
| 73 |     boost::bind(&StockServer::handle_accept, this,
 | 
|---|
| 74 |       boost::asio::placeholders::error, new_conn));
 | 
|---|
| 75 | }
 | 
|---|
| 76 | 
 | 
|---|
| 77 | void StockServer::handle_write(const boost::system::error_code& e, connection_ptr conn)
 | 
|---|
| 78 | {
 | 
|---|
| 79 |   // Nothing to do. The socket will be closed automatically when the last
 | 
|---|
| 80 |   // reference to the connection object goes away.
 | 
|---|
| 81 | }
 | 
|---|