diff options
Diffstat (limited to 'src/rgw/rgw_asio_client.h')
-rw-r--r-- | src/rgw/rgw_asio_client.h | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/src/rgw/rgw_asio_client.h b/src/rgw/rgw_asio_client.h new file mode 100644 index 000000000..e2ab943dd --- /dev/null +++ b/src/rgw/rgw_asio_client.h @@ -0,0 +1,62 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab ft=cpp + +#pragma once + +#include <boost/asio/ip/tcp.hpp> +#include <boost/beast/core.hpp> +#include <boost/beast/http.hpp> +#include "include/ceph_assert.h" + +#include "rgw_client_io.h" + +namespace rgw { +namespace asio { + +namespace beast = boost::beast; +using parser_type = beast::http::request_parser<beast::http::buffer_body>; + +class ClientIO : public io::RestfulClient, + public io::BuffererSink { + protected: + parser_type& parser; + private: + const bool is_ssl; + using endpoint_type = boost::asio::ip::tcp::endpoint; + endpoint_type local_endpoint; + endpoint_type remote_endpoint; + + RGWEnv env; + + rgw::io::StaticOutputBufferer<> txbuf; + bool sent100continue = false; + + public: + ClientIO(parser_type& parser, bool is_ssl, + const endpoint_type& local_endpoint, + const endpoint_type& remote_endpoint); + ~ClientIO() override; + + int init_env(CephContext *cct) override; + size_t complete_request() override; + void flush() override; + size_t send_status(int status, const char *status_name) override; + size_t send_100_continue() override; + size_t send_header(const std::string_view& name, + const std::string_view& value) override; + size_t send_content_length(uint64_t len) override; + size_t complete_header() override; + + size_t send_body(const char* buf, size_t len) override { + return write_data(buf, len); + } + + RGWEnv& get_env() noexcept override { + return env; + } + + bool sent_100_continue() const { return sent100continue; } +}; + +} // namespace asio +} // namespace rgw |