74 lines
1.6 KiB
C++
74 lines
1.6 KiB
C++
#include "http_server.h"
|
|
#include "session_base.h"
|
|
|
|
#include <boost/asio/dispatch.hpp>
|
|
#include <iostream>
|
|
|
|
namespace http_server
|
|
{
|
|
void ReportError(beast::error_code ec, std::string_view what)
|
|
{
|
|
std::cerr << what << ": "sv << ec.message() << std::endl;
|
|
}
|
|
SessionBase::SessionBase(tcp::socket&& socket)
|
|
: stream_(std::move(socket))
|
|
{
|
|
|
|
}
|
|
void SessionBase::Run()
|
|
{
|
|
net::dispatch(stream_.get_executor(),
|
|
beast::bind_front_handler(&SessionBase::Read,
|
|
GetSharedThis()));
|
|
}
|
|
void SessionBase::Read()
|
|
{
|
|
using namespace std::literals;
|
|
request_ = {};
|
|
stream_.expires_after(30s);
|
|
http::async_read(stream_, buffer_, request_,
|
|
beast::bind_front_handler(&SessionBase::OnRead,
|
|
GetSharedThis()));
|
|
}
|
|
void SessionBase::OnRead(beast::error_code ec, size_t bytes_read)
|
|
{
|
|
if (ec == http::error::end_of_stream)
|
|
{
|
|
return Close();
|
|
}
|
|
|
|
if (ec)
|
|
{
|
|
return ReportError(ec, "read"sv);
|
|
}
|
|
|
|
HandleRequest(std::move(request_));
|
|
}
|
|
void SessionBase::Close()
|
|
{
|
|
beast::error_code ec;
|
|
|
|
stream_.socket().shutdown(tcp::socket::shutdown_send, ec);
|
|
}
|
|
|
|
void SessionBase::OnWrite(bool close,
|
|
boost::beast::error_code ec,
|
|
std::size_t bytes_written)
|
|
{
|
|
if (ec)
|
|
{
|
|
return ReportError(ec, "write"sv);
|
|
}
|
|
|
|
if (close)
|
|
{
|
|
// Семантика ответа требует закрыть соединение
|
|
return Close();
|
|
}
|
|
|
|
// Считываем следующий запрос
|
|
Read();
|
|
}
|
|
|
|
} // namespace http_server
|