ceph/src/boost/libs/asio/example/cpp11/http/server/server.cpp

   1 //
   2 // server.cpp
   3 // ~~~~~~~~~~
   4 //
   5 // Copyright (c) 2003-2019 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 "server.hpp"
  12 #include <signal.h>
  13 #include <utility>
  14
  15 namespace http {
  16 namespace server {
  17
  18 server::server(const std::string& address, const std::string& port,
  19     const std::string& doc_root)
  20   : io_context_(1),
  21     signals_(io_context_),
  22     acceptor_(io_context_),
  23     connection_manager_(),
  24     request_handler_(doc_root)
  25 {
  26   // Register to handle the signals that indicate when the server should exit.
  27   // It is safe to register for the same signal multiple times in a program,
  28   // provided all registration for the specified signal is made through Asio.
  29   signals_.add(SIGINT);
  30   signals_.add(SIGTERM);
  31 #if defined(SIGQUIT)
  32   signals_.add(SIGQUIT);
  33 #endif // defined(SIGQUIT)
  34
  35   do_await_stop();
  36
  37   // Open the acceptor with the option to reuse the address (i.e. SO_REUSEADDR).
  38   boost::asio::ip::tcp::resolver resolver(io_context_);
  39   boost::asio::ip::tcp::endpoint endpoint =
  40     *resolver.resolve(address, port).begin();
  41   acceptor_.open(endpoint.protocol());
  42   acceptor_.set_option(boost::asio::ip::tcp::acceptor::reuse_address(true));
  43   acceptor_.bind(endpoint);
  44   acceptor_.listen();
  45
  46   do_accept();
  47 }
  48
  49 void server::run()
  50 {
  51   // The io_context::run() call will block until all asynchronous operations
  52   // have finished. While the server is running, there is always at least one
  53   // asynchronous operation outstanding: the asynchronous accept call waiting
  54   // for new incoming connections.
  55   io_context_.run();
  56 }
  57
  58 void server::do_accept()
  59 {
  60   acceptor_.async_accept(
  61       [this](boost::system::error_code ec, boost::asio::ip::tcp::socket socket)
  62       {
  63         // Check whether the server was stopped by a signal before this
  64         // completion handler had a chance to run.
  65         if (!acceptor_.is_open())
  66         {
  67           return;
  68         }
  69
  70         if (!ec)
  71         {
  72           connection_manager_.start(std::make_shared<connection>(
  73               std::move(socket), connection_manager_, request_handler_));
  74         }
  75
  76         do_accept();
  77       });
  78 }
  79
  80 void server::do_await_stop()
  81 {
  82   signals_.async_wait(
  83       [this](boost::system::error_code /*ec*/, int /*signo*/)
  84       {
  85         // The server is stopped by cancelling all outstanding asynchronous
  86         // operations. Once all operations have finished the io_context::run()
  87         // call will exit.
  88         acceptor_.close();
  89         connection_manager_.stop_all();
  90       });
  91 }
  92
  93 } // namespace server
  94 } // namespace http