3#include "../type_traits.hpp"
4#include "../../core/type_traits.hpp"
5#include "../../core/http/request.hpp"
6#include "../../core/http/response.hpp"
7#include "../../core/http/type_traits.hpp"
9#include <boost/asio/strand.hpp>
10#include <boost/beast/core.hpp>
11#include <spdlog/logger.h>
16namespace malloy::client::http
24 template<
class Derived, malloy::http::concepts::body ReqBody, concepts::response_filter Filter,
typename Callback>
28 using resp_t =
typename Filter::response_type;
29 using callback_t = Callback;
31 connection(std::shared_ptr<spdlog::logger> logger, boost::asio::io_context& io_ctx,
const std::uint64_t body_limit) :
32 m_logger(std::move(logger)),
33 m_resolver(boost::asio::make_strand(io_ctx))
37 throw std::invalid_argument(
"no valid logger provided.");
40 m_parser.body_limit(body_limit);
47 std::promise<malloy::error_code> err_channel,
52 m_req_filter = std::move(filter);
53 m_req = std::move(req);
54 m_err_channel = std::move(err_channel);
55 m_cb.emplace(std::move(cb));
58 m_resolver.async_resolve(
59 m_req.base()[malloy::http::field::host],
60 std::to_string(m_req.
port()),
61 boost::beast::bind_front_handler(
62 &connection::on_resolve,
63 derived().shared_from_this()
69 std::shared_ptr<spdlog::logger> m_logger;
75 boost::beast::http::async_write(
78 boost::beast::bind_front_handler(
79 &connection::on_write,
80 derived().shared_from_this()
86 boost::asio::ip::tcp::resolver m_resolver;
87 boost::beast::flat_buffer m_buffer;
88 boost::beast::http::response_parser<boost::beast::http::empty_body> m_parser;
90 std::promise<malloy::error_code> m_err_channel;
91 std::optional<callback_t> m_cb;
98 return static_cast<Derived&
>(*this);
102 on_resolve(
const boost::beast::error_code& ec, boost::asio::ip::tcp::resolver::results_type results)
104 m_logger->trace(
"on_resolve()");
107 m_logger->error(
"on_resolve: {}", ec.message());
108 m_err_channel.set_value(ec);
113 boost::beast::get_lowest_layer(derived().stream()).expires_after(std::chrono::seconds(30));
116 boost::beast::get_lowest_layer(derived().stream()).async_connect(
118 boost::beast::bind_front_handler(
119 &connection::on_connect,
120 derived().shared_from_this()
126 on_connect(
const boost::beast::error_code& ec, boost::asio::ip::tcp::resolver::results_type::endpoint_type)
128 m_logger->trace(
"on_connect()");
131 m_logger->error(
"on_connect: {}", ec.message());
132 m_err_channel.set_value(ec);
137 boost::beast::get_lowest_layer(derived().stream()).expires_after(std::chrono::seconds(30));
140 derived().hook_connected();
144 on_write(
const boost::beast::error_code& ec, [[maybe_unused]]
const std::size_t bytes_transferred)
147 m_logger->error(
"on_write: {}", ec.message());
148 m_err_channel.set_value(ec);
153 boost::beast::http::async_read_header(
157 malloy::bind_front_handler(
158 &connection::on_read_header,
159 derived().shared_from_this()
168 m_logger->error(
"on_read_header: '{}'", ec.message());
169 m_err_channel.set_value(ec);
174 auto bodies = m_req_filter.body_for(m_parser.get().base());
175 std::visit([
this](
auto&& body) {
176 using body_t = std::decay_t<
decltype(body)>;
178 auto parser = std::make_shared<boost::beast::http::response_parser<body_t>>(std::move(m_parser));
179 m_req_filter.setup_body(parser->get().base(), parser->get().body());
181 boost::beast::http::async_read(
185 [
this, parser, me = derived().shared_from_this()](
auto ec,
auto) {
187 m_logger->error(
"on_read(): {}", ec.message());
188 m_err_channel.set_value(ec);
206 boost::beast::get_lowest_layer(derived().stream()).socket().shutdown(boost::asio::ip::tcp::socket::shutdown_both, ec);
209 if (ec && ec != boost::beast::errc::not_connected)
210 m_logger->error(
"shutdown: {}", ec.message());
212 m_err_channel.set_value(ec);
Definition: connection.hpp:26
Definition: request.hpp:19
constexpr std::uint16_t port() const noexcept
Definition: request.hpp:143
Definition: response.hpp:22
boost::beast::error_code error_code
Error code used to signify errors without throwing. Truthy means it holds an error.
Definition: error.hpp:9