mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-27 03:45:11 +00:00
ae047e6419
Add a TTL-based timer to connection_factory to automatically refresh
resolved host name addresses when they expire.
(cherry picked from commit 6eb7dba352)
51 lines
1.5 KiB
C++
51 lines
1.5 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;
|
|
semaphore _init_semaphore{1};
|
|
bool _creds_init = false;
|
|
std::optional<std::vector<net::inet_address>> _addr_list;
|
|
shared_ptr<tls::certificate_credentials> _creds;
|
|
uint16_t _addr_pos{0};
|
|
bool _use_https;
|
|
std::chrono::seconds _address_ttl{0};
|
|
timer<lowres_clock> _addr_update_timer;
|
|
|
|
future<> init_credentials();
|
|
future<net::inet_address> get_address();
|
|
future<shared_ptr<tls::certificate_credentials>> get_creds();
|
|
future<connected_socket> connect(net::inet_address address);
|
|
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> = {});
|
|
|
|
virtual future<connected_socket> make(abort_source*) override;
|
|
future<> close() override;
|
|
};
|
|
|
|
} // namespace utils::http
|