pion  5.0.6
response_reader.hpp
1 // ---------------------------------------------------------------------
2 // pion: a Boost C++ framework for building lightweight HTTP interfaces
3 // ---------------------------------------------------------------------
4 // Copyright (C) 2007-2014 Splunk Inc. (https://github.com/splunk/pion)
5 //
6 // Distributed under the Boost Software License, Version 1.0.
7 // See http://www.boost.org/LICENSE_1_0.txt
8 //
9 
10 #ifndef __PION_HTTP_RESPONSE_READER_HEADER__
11 #define __PION_HTTP_RESPONSE_READER_HEADER__
12 
13 #include <boost/asio.hpp>
14 #include <boost/bind.hpp>
15 #include <boost/function.hpp>
16 #include <boost/function/function2.hpp>
17 #include <boost/shared_ptr.hpp>
18 #include <boost/enable_shared_from_this.hpp>
19 #include <pion/config.hpp>
20 #include <pion/http/response.hpp>
21 #include <pion/http/reader.hpp>
22 
23 
24 namespace pion { // begin namespace pion
25 namespace http { // begin namespace http
26 
27 
32  public http::reader,
33  public boost::enable_shared_from_this<response_reader>
34 {
35 
36 public:
37 
39  typedef boost::function3<void, http::response_ptr, tcp::connection_ptr,
40  const boost::system::error_code&> finished_handler_t;
41 
42 
43  // default destructor
44  virtual ~response_reader() {}
45 
53  static inline boost::shared_ptr<response_reader>
54  create(tcp::connection_ptr& tcp_conn, const http::request& http_request,
55  finished_handler_t handler)
56  {
57  return boost::shared_ptr<response_reader>
58  (new response_reader(tcp_conn, http_request, handler));
59  }
60 
62  inline void set_headers_parsed_callback(finished_handler_t& h) { m_parsed_headers = h; }
63 
64 
65 protected:
66 
74  response_reader(tcp::connection_ptr& tcp_conn, const http::request& http_request,
75  finished_handler_t handler)
76  : http::reader(false, tcp_conn), m_http_msg(new http::response(http_request)),
77  m_finished(handler)
78  {
79  m_http_msg->set_remote_ip(tcp_conn->get_remote_ip());
80  set_logger(PION_GET_LOGGER("pion.http.response_reader"));
81  }
82 
84  virtual void read_bytes(void) {
85  get_connection()->async_read_some(boost::bind(&response_reader::consume_bytes,
86  shared_from_this(),
87  boost::asio::placeholders::error,
88  boost::asio::placeholders::bytes_transferred));
89  }
90 
92  virtual void finished_parsing_headers(const boost::system::error_code& ec) {
93  // call the finished headers handler with the HTTP message
95  }
96 
98  virtual void finished_reading(const boost::system::error_code& ec) {
99  // call the finished handler with the finished HTTP message
101  }
102 
104  virtual http::message& get_message(void) { return *m_http_msg; }
105 
106 
108  http::response_ptr m_http_msg;
109 
111  finished_handler_t m_finished;
112 
114  finished_handler_t m_parsed_headers;
115 };
116 
117 
119 typedef boost::shared_ptr<response_reader> response_reader_ptr;
120 
121 
122 } // end namespace http
123 } // end namespace pion
124 
125 #endif
void consume_bytes(void)
Consumes bytes that have been read using an HTTP parser.
Definition: http_reader.cpp:66
http::response_ptr m_http_msg
The new HTTP message container being created.
tcp::connection_ptr & get_connection(void)
returns a shared pointer to the TCP connection
Definition: reader.hpp:40
boost::function3< void, http::response_ptr, tcp::connection_ptr, const boost::system::error_code & > finished_handler_t
function called after the HTTP message has been parsed
virtual void read_bytes(void)
Reads more bytes from the TCP connection.
virtual void finished_parsing_headers(const boost::system::error_code &ec)
Called after we have finished parsing the HTTP message headers.
response_reader(tcp::connection_ptr &tcp_conn, const http::request &http_request, finished_handler_t handler)
virtual http::message & get_message(void)
Returns a reference to the HTTP message being parsed.
virtual void finished_reading(const boost::system::error_code &ec)
Called after we have finished reading/parsing the HTTP message.
void set_logger(logger log_ptr)
sets the logger to be used
Definition: parser.hpp:294
finished_handler_t m_parsed_headers
function called after the HTTP message headers have been parsed
void set_headers_parsed_callback(finished_handler_t &h)
sets a function to be called after HTTP headers have been parsed
finished_handler_t m_finished
function called after the HTTP message has been parsed
static boost::shared_ptr< response_reader > create(tcp::connection_ptr &tcp_conn, const http::request &http_request, finished_handler_t handler)