server.hpp

//
// server.hpp
// ~~~~~~~~~~
//
// Copyright (c) 2003-2008 Christopher M. Kohlhoff (chris at kohlhoff dot com)
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//

#ifndef HTTP_SERVER3_SERVER_HPP
#define HTTP_SERVER3_SERVER_HPP

#include <boost/asio.hpp>
#include <string>
#include <vector>
#include <boost/noncopyable.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/thread.hpp>
#include "connection.hpp"
#include "request_handler.hpp"

namespace http
{
  namespace server
  {
    /// The top-level class of the HTTP server.
    class server: public boost::enable_shared_from_this<server>, private boost::noncopyable
    {
    public:
      /// Construct the server to listen on the specified TCP address and port, and
      /// serve up files from the given directory.
      explicit
      server(const std::string& address, const std::string& port, const std::string& doc_root,
             std::size_t thread_pool_size);

      /// Run the server's io_service loop.
      void
      run();

      /// Stop the server.
      void
      stop();

      void
      start();

      /// Register a request handler for a specific path, and request_type
      /// request_type GET, POST, ....
      void
      register_request_handler(const std::string& path, const std::string& request_type,
                               const RequestHandlerFnc& handler);
      ///get a list of the request handlers registered for the particular request type.
      std::vector<std::string>
      get_request_handlers(const std::string& request_type, const std::string& match) const;
    private:
      /// Handle completion of an asynchronous accept operation.
      void
      handle_accept(const boost::system::error_code& e);

      /// The number of threads that will call io_service::run().
      std::size_t thread_pool_size_;

      /// The io_service used to perform asynchronous operations.
      boost::asio::io_service io_service_;

      /// Acceptor used to listen for incoming connections.
      boost::asio::ip::tcp::acceptor acceptor_;

      /// The next connection to be accepted.
      connection_ptr new_connection_;

      /// The handler for all incoming requests.
      request_handler request_handler_;

      boost::shared_ptr<boost::thread> run_thread_;
    };

  } // namespace server3
} // namespace http

#endif // HTTP_SERVER3_SERVER_HPP