You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

78 lines
1.9 KiB
C++

#ifndef WEBCC_SERVER_H_
#define WEBCC_SERVER_H_
#include <regex>
#include <string>
#include "boost/asio/io_context.hpp"
#include "boost/asio/ip/tcp.hpp"
#include "boost/asio/signal_set.hpp"
#include "webcc/connection.h"
#include "webcc/connection_pool.h"
#include "webcc/request_handler.h"
#include "webcc/view.h"
namespace webcc {
class Server {
public:
explicit Server(std::uint16_t port, std::size_t workers = 1,
const Path& doc_root = {});
virtual ~Server() = default;
Server(const Server&) = delete;
Server& operator=(const Server&) = delete;
// Route a URL to a view.
// The URL should start with "/". E.g., "/instances".
bool Route(const std::string& url, ViewPtr view,
const Strings& methods = { "GET" }) {
return request_handler_.Route(url, view, methods);
}
// Route a regular expression URL to a view.
// The URL should start with "/" and be a regular expression.
// E.g., "/instances/(\\d+)".
bool Route(const RegexUrl& regex_url, ViewPtr view,
const Strings& methods = { "GET" }) {
return request_handler_.Route(regex_url, view, methods);
}
// Run the loop.
void Run();
private:
// Register to handle the signals that indicate when the server should exit.
void RegisterSignals();
// Initiate an asynchronous accept operation.
void DoAccept();
// Wait for a request to stop the server.
void DoAwaitStop();
// The io_context used to perform asynchronous operations.
boost::asio::io_context io_context_;
// Acceptor used to listen for incoming connections.
boost::asio::ip::tcp::acceptor acceptor_;
// The connection pool which owns all live connections.
ConnectionPool pool_;
// The signals for processing termination notifications.
boost::asio::signal_set signals_;
// The number of worker threads.
std::size_t workers_;
// The handler for incoming requests.
RequestHandler request_handler_;
};
} // namespace webcc
#endif // WEBCC_SERVER_H_