mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-04 15:40:49 +00:00
74 lines
2.2 KiB
C++
74 lines
2.2 KiB
C++
#pragma once
|
|
#include <Poco/Net/IPAddress.h>
|
|
#include <Poco/Net/SocketAddress.h>
|
|
#include <memory>
|
|
#include <common/types.h>
|
|
#include <Core/Names.h>
|
|
#include <boost/noncopyable.hpp>
|
|
#include <common/logger_useful.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/// A singleton implementing DNS names resolving with optional DNS cache
|
|
/// The cache is being updated asynchronous in separate thread (see DNSCacheUpdater)
|
|
/// or it could be updated manually via drop() method.
|
|
class DNSResolver : private boost::noncopyable
|
|
{
|
|
public:
|
|
typedef std::vector<Poco::Net::IPAddress> IPAddresses;
|
|
|
|
static DNSResolver & instance();
|
|
|
|
DNSResolver(const DNSResolver &) = delete;
|
|
|
|
/// Accepts host names like 'example.com' or '127.0.0.1' or '::1' and resolves its IP
|
|
Poco::Net::IPAddress resolveHost(const std::string & host);
|
|
|
|
/// Accepts host names like 'example.com' or '127.0.0.1' or '::1' and resolves all its IPs
|
|
IPAddresses resolveHostAll(const std::string & host);
|
|
|
|
/// Accepts host names like 'example.com:port' or '127.0.0.1:port' or '[::1]:port' and resolves its IP and port
|
|
Poco::Net::SocketAddress resolveAddress(const std::string & host_and_port);
|
|
|
|
Poco::Net::SocketAddress resolveAddress(const std::string & host, UInt16 port);
|
|
|
|
/// Accepts host IP and resolves its host name
|
|
String reverseResolve(const Poco::Net::IPAddress & address);
|
|
|
|
/// Get this server host name
|
|
String getHostName();
|
|
|
|
/// Disables caching
|
|
void setDisableCacheFlag(bool is_disabled = true);
|
|
|
|
/// Drops all caches
|
|
void dropCache();
|
|
|
|
/// Updates all known hosts in cache.
|
|
/// Returns true if IP of any host has been changed.
|
|
bool updateCache();
|
|
|
|
~DNSResolver();
|
|
|
|
private:
|
|
template<typename UpdateF, typename ElemsT>
|
|
bool updateCacheImpl(UpdateF && update_func, ElemsT && elems, const String & log_msg);
|
|
|
|
DNSResolver();
|
|
|
|
struct Impl;
|
|
std::unique_ptr<Impl> impl;
|
|
Poco::Logger * log;
|
|
|
|
/// Updates cached value and returns true it has been changed.
|
|
bool updateHost(const String & host);
|
|
bool updateAddress(const Poco::Net::IPAddress & address);
|
|
|
|
void addToNewHosts(const String & host);
|
|
void addToNewAddresses(const Poco::Net::IPAddress & address);
|
|
};
|
|
|
|
}
|