8f54b709ce
Also allow creating the object using an endpoint expression. Note: this moves code to the .cc file, because it introduces a few more lines, and I feel we have to much stuff in headers as is.
63 lines
1.9 KiB
C++
63 lines
1.9 KiB
C++
/*
|
|
* Copyright (C) 2023-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <seastar/core/seastar.hh>
|
|
#include <seastar/core/shared_future.hh>
|
|
#include <seastar/http/client.hh>
|
|
#include <seastar/net/dns.hh>
|
|
#include <seastar/net/tls.hh>
|
|
|
|
#include "seastarx.hh"
|
|
#include "utils/log.hh"
|
|
|
|
namespace utils::http {
|
|
|
|
future<shared_ptr<tls::certificate_credentials>> system_trust_credentials();
|
|
|
|
class dns_connection_factory : public seastar::http::experimental::connection_factory {
|
|
protected:
|
|
std::string _host;
|
|
int _port;
|
|
logging::logger& _logger;
|
|
struct state {
|
|
bool initialized = false;
|
|
socket_address addr;
|
|
shared_ptr<tls::certificate_credentials> creds;
|
|
state(shared_ptr<tls::certificate_credentials>);
|
|
};
|
|
lw_shared_ptr<state> _state;
|
|
shared_future<> _done;
|
|
|
|
// This method can out-live the factory instance, in case `make()` is never called before the instance is destroyed.
|
|
static future<> initialize(lw_shared_ptr<state> state, std::string host, int port, bool use_https, logging::logger& logger);
|
|
public:
|
|
dns_connection_factory(dns_connection_factory&&);
|
|
dns_connection_factory(std::string host, int port, bool use_https, logging::logger& logger, shared_ptr<tls::certificate_credentials> = {});
|
|
dns_connection_factory(std::string endpoint_url, logging::logger& logger, shared_ptr<tls::certificate_credentials> = {});
|
|
|
|
virtual future<connected_socket> make(abort_source*) override;
|
|
};
|
|
|
|
// simple URL parser, just enough to handle required aspects for normal endpoint usage
|
|
// could use boost::url, but this requires additional libraries being added to
|
|
// install-dependencies etc.
|
|
struct url_info {
|
|
std::string scheme;
|
|
std::string host;
|
|
std::string path;
|
|
uint16_t port;
|
|
|
|
bool is_https() const;
|
|
};
|
|
|
|
url_info parse_simple_url(std::string_view uri);
|
|
|
|
} // namespace utils::http
|