mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 11:22:12 +00:00
351 lines
11 KiB
C++
351 lines
11 KiB
C++
#pragma once
|
|
|
|
#include <IO/ReadHelpers.h>
|
|
#include <IO/ReadWriteBufferFromHTTP.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Access/AccessType.h>
|
|
#include <Parsers/IdentifierQuotingStyle.h>
|
|
#include <Poco/File.h>
|
|
#include <Poco/Logger.h>
|
|
#include <Poco/Net/HTTPRequest.h>
|
|
#include <Poco/Path.h>
|
|
#include <Poco/URI.h>
|
|
#include <Poco/Util/AbstractConfiguration.h>
|
|
#include <Common/ShellCommand.h>
|
|
#include <common/logger_useful.h>
|
|
#include <ext/range.h>
|
|
|
|
#if !defined(ARCADIA_BUILD)
|
|
# include <Common/config.h>
|
|
#endif
|
|
|
|
namespace DB
|
|
{
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int EXTERNAL_SERVER_IS_NOT_RESPONDING;
|
|
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
|
|
}
|
|
|
|
/**
|
|
* Class for Helpers for Xdbc-bridges, provide utility methods, not main request
|
|
*/
|
|
class IXDBCBridgeHelper
|
|
{
|
|
public:
|
|
static constexpr inline auto DEFAULT_FORMAT = "RowBinary";
|
|
|
|
virtual std::vector<std::pair<std::string, std::string>> getURLParams(const std::string & cols, UInt64 max_block_size) const = 0;
|
|
virtual void startBridgeSync() const = 0;
|
|
virtual Poco::URI getMainURI() const = 0;
|
|
virtual Poco::URI getColumnsInfoURI() const = 0;
|
|
virtual IdentifierQuotingStyle getIdentifierQuotingStyle() = 0;
|
|
virtual bool isSchemaAllowed() = 0;
|
|
virtual String getName() const = 0;
|
|
|
|
virtual ~IXDBCBridgeHelper() = default;
|
|
};
|
|
|
|
using BridgeHelperPtr = std::shared_ptr<IXDBCBridgeHelper>;
|
|
|
|
template <typename BridgeHelperMixin>
|
|
class XDBCBridgeHelper : public IXDBCBridgeHelper
|
|
{
|
|
private:
|
|
Poco::Timespan http_timeout;
|
|
|
|
std::string connection_string;
|
|
|
|
Poco::URI ping_url;
|
|
|
|
Poco::Logger * log = &Poco::Logger::get(BridgeHelperMixin::getName() + "BridgeHelper");
|
|
|
|
std::optional<IdentifierQuotingStyle> quote_style;
|
|
std::optional<bool> is_schema_allowed;
|
|
|
|
protected:
|
|
auto getConnectionString() const
|
|
{
|
|
return connection_string;
|
|
}
|
|
|
|
public:
|
|
using Configuration = Poco::Util::AbstractConfiguration;
|
|
|
|
const Context & context;
|
|
const Configuration & config;
|
|
|
|
static constexpr inline auto DEFAULT_HOST = "localhost";
|
|
static constexpr inline auto DEFAULT_PORT = BridgeHelperMixin::DEFAULT_PORT;
|
|
static constexpr inline auto PING_HANDLER = "/ping";
|
|
static constexpr inline auto MAIN_HANDLER = "/";
|
|
static constexpr inline auto COL_INFO_HANDLER = "/columns_info";
|
|
static constexpr inline auto IDENTIFIER_QUOTE_HANDLER = "/identifier_quote";
|
|
static constexpr inline auto SCHEMA_ALLOWED_HANDLER = "/schema_allowed";
|
|
static constexpr inline auto PING_OK_ANSWER = "Ok.";
|
|
|
|
XDBCBridgeHelper(const Context & global_context_, const Poco::Timespan & http_timeout_, const std::string & connection_string_)
|
|
: http_timeout(http_timeout_), connection_string(connection_string_), context(global_context_), config(context.getConfigRef())
|
|
{
|
|
size_t bridge_port = config.getUInt(BridgeHelperMixin::configPrefix() + ".port", DEFAULT_PORT);
|
|
std::string bridge_host = config.getString(BridgeHelperMixin::configPrefix() + ".host", DEFAULT_HOST);
|
|
|
|
ping_url.setHost(bridge_host);
|
|
ping_url.setPort(bridge_port);
|
|
ping_url.setScheme("http");
|
|
ping_url.setPath(PING_HANDLER);
|
|
}
|
|
|
|
String getName() const override
|
|
{
|
|
return BridgeHelperMixin::getName();
|
|
}
|
|
|
|
IdentifierQuotingStyle getIdentifierQuotingStyle() override
|
|
{
|
|
if (!quote_style.has_value())
|
|
{
|
|
startBridgeSync();
|
|
|
|
auto uri = createBaseURI();
|
|
uri.setPath(IDENTIFIER_QUOTE_HANDLER);
|
|
uri.addQueryParameter("connection_string", getConnectionString());
|
|
|
|
ReadWriteBufferFromHTTP buf(
|
|
uri, Poco::Net::HTTPRequest::HTTP_POST, {}, ConnectionTimeouts::getHTTPTimeouts(context));
|
|
std::string character;
|
|
readStringBinary(character, buf);
|
|
if (character.length() > 1)
|
|
throw Exception("Failed to parse quoting style from '" + character + "' for service " + BridgeHelperMixin::serviceAlias(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT);
|
|
else if (character.length() == 0)
|
|
quote_style = IdentifierQuotingStyle::None;
|
|
else if (character[0] == '`')
|
|
quote_style = IdentifierQuotingStyle::Backticks;
|
|
else if (character[0] == '"')
|
|
quote_style = IdentifierQuotingStyle::DoubleQuotes;
|
|
else
|
|
throw Exception("Can not map quote identifier '" + character + "' to enum value", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT);
|
|
}
|
|
|
|
return *quote_style;
|
|
}
|
|
|
|
bool isSchemaAllowed() override
|
|
{
|
|
if (!is_schema_allowed.has_value())
|
|
{
|
|
startBridgeSync();
|
|
|
|
auto uri = createBaseURI();
|
|
uri.setPath(SCHEMA_ALLOWED_HANDLER);
|
|
uri.addQueryParameter("connection_string", getConnectionString());
|
|
|
|
ReadWriteBufferFromHTTP buf(
|
|
uri, Poco::Net::HTTPRequest::HTTP_POST, {}, ConnectionTimeouts::getHTTPTimeouts(context));
|
|
|
|
bool res;
|
|
readBoolText(res, buf);
|
|
is_schema_allowed = res;
|
|
}
|
|
|
|
return *is_schema_allowed;
|
|
}
|
|
|
|
/**
|
|
* @todo leaky abstraction - used by external API's
|
|
*/
|
|
std::vector<std::pair<std::string, std::string>> getURLParams(const std::string & cols, UInt64 max_block_size) const override
|
|
{
|
|
std::vector<std::pair<std::string, std::string>> result;
|
|
|
|
result.emplace_back("connection_string", connection_string); /// already validated
|
|
result.emplace_back("columns", cols);
|
|
result.emplace_back("max_block_size", std::to_string(max_block_size));
|
|
|
|
return result;
|
|
}
|
|
|
|
/**
|
|
* Performs spawn of external daemon
|
|
*/
|
|
void startBridgeSync() const override
|
|
{
|
|
if (!checkBridgeIsRunning())
|
|
{
|
|
LOG_TRACE(log, "{} is not running, will try to start it", BridgeHelperMixin::serviceAlias());
|
|
startBridge();
|
|
bool started = false;
|
|
|
|
uint64_t milliseconds_to_wait = 10; /// Exponential backoff
|
|
uint64_t counter = 0;
|
|
while (milliseconds_to_wait < 10000)
|
|
{
|
|
++counter;
|
|
LOG_TRACE(log, "Checking {} is running, try {}", BridgeHelperMixin::serviceAlias(), counter);
|
|
if (checkBridgeIsRunning())
|
|
{
|
|
started = true;
|
|
break;
|
|
}
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(milliseconds_to_wait));
|
|
milliseconds_to_wait *= 2;
|
|
}
|
|
|
|
if (!started)
|
|
throw Exception(BridgeHelperMixin::getName() + "BridgeHelper: " + BridgeHelperMixin::serviceAlias() + " is not responding",
|
|
ErrorCodes::EXTERNAL_SERVER_IS_NOT_RESPONDING);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* URI to fetch the data from external service
|
|
*/
|
|
Poco::URI getMainURI() const override
|
|
{
|
|
auto uri = createBaseURI();
|
|
uri.setPath(MAIN_HANDLER);
|
|
return uri;
|
|
}
|
|
|
|
/**
|
|
* URI to retrieve column description from external service
|
|
*/
|
|
Poco::URI getColumnsInfoURI() const override
|
|
{
|
|
auto uri = createBaseURI();
|
|
uri.setPath(COL_INFO_HANDLER);
|
|
return uri;
|
|
}
|
|
|
|
protected:
|
|
Poco::URI createBaseURI() const
|
|
{
|
|
Poco::URI uri;
|
|
uri.setHost(ping_url.getHost());
|
|
uri.setPort(ping_url.getPort());
|
|
uri.setScheme("http");
|
|
return uri;
|
|
}
|
|
|
|
private:
|
|
bool checkBridgeIsRunning() const
|
|
{
|
|
try
|
|
{
|
|
ReadWriteBufferFromHTTP buf(
|
|
ping_url, Poco::Net::HTTPRequest::HTTP_GET, {}, ConnectionTimeouts::getHTTPTimeouts(context));
|
|
return checkString(XDBCBridgeHelper::PING_OK_ANSWER, buf);
|
|
}
|
|
catch (...)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/* Contains logic for instantiation of the bridge instance */
|
|
void startBridge() const
|
|
{
|
|
auto cmd = BridgeHelperMixin::startBridge(config, log, http_timeout);
|
|
context.addXDBCBridgeCommand(std::move(cmd));
|
|
}
|
|
};
|
|
|
|
struct JDBCBridgeMixin
|
|
{
|
|
static constexpr inline auto DEFAULT_PORT = 9019;
|
|
static const String configPrefix()
|
|
{
|
|
return "jdbc_bridge";
|
|
}
|
|
static const String serviceAlias()
|
|
{
|
|
return "clickhouse-jdbc-bridge";
|
|
}
|
|
static const String getName()
|
|
{
|
|
return "JDBC";
|
|
}
|
|
static AccessType getSourceAccessType()
|
|
{
|
|
return AccessType::JDBC;
|
|
}
|
|
|
|
static std::unique_ptr<ShellCommand> startBridge(const Poco::Util::AbstractConfiguration &, const Poco::Logger *, const Poco::Timespan &)
|
|
{
|
|
throw Exception("jdbc-bridge is not running. Please, start it manually", ErrorCodes::EXTERNAL_SERVER_IS_NOT_RESPONDING);
|
|
}
|
|
};
|
|
|
|
struct ODBCBridgeMixin
|
|
{
|
|
static constexpr inline auto DEFAULT_PORT = 9018;
|
|
|
|
static const String configPrefix()
|
|
{
|
|
return "odbc_bridge";
|
|
}
|
|
static const String serviceAlias()
|
|
{
|
|
return "clickhouse-odbc-bridge";
|
|
}
|
|
static const String getName()
|
|
{
|
|
return "ODBC";
|
|
}
|
|
static AccessType getSourceAccessType()
|
|
{
|
|
return AccessType::ODBC;
|
|
}
|
|
|
|
static std::unique_ptr<ShellCommand> startBridge(
|
|
const Poco::Util::AbstractConfiguration & config, Poco::Logger * log, const Poco::Timespan & http_timeout)
|
|
{
|
|
/// Path to executable folder
|
|
Poco::Path path{config.getString("application.dir", "/usr/bin")};
|
|
|
|
std::vector<std::string> cmd_args;
|
|
path.setFileName("clickhouse-odbc-bridge");
|
|
|
|
#if !CLICKHOUSE_SPLIT_BINARY
|
|
cmd_args.push_back("odbc-bridge");
|
|
#endif
|
|
|
|
cmd_args.push_back("--http-port");
|
|
cmd_args.push_back(std::to_string(config.getUInt(configPrefix() + ".port", DEFAULT_PORT)));
|
|
cmd_args.push_back("--listen-host");
|
|
cmd_args.push_back(config.getString(configPrefix() + ".listen_host", XDBCBridgeHelper<ODBCBridgeMixin>::DEFAULT_HOST));
|
|
cmd_args.push_back("--http-timeout");
|
|
cmd_args.push_back(std::to_string(http_timeout.totalMicroseconds()));
|
|
if (config.has("logger." + configPrefix() + "_log"))
|
|
{
|
|
cmd_args.push_back("--log-path");
|
|
cmd_args.push_back(config.getString("logger." + configPrefix() + "_log"));
|
|
}
|
|
if (config.has("logger." + configPrefix() + "_errlog"))
|
|
{
|
|
cmd_args.push_back("--err-log-path");
|
|
cmd_args.push_back(config.getString("logger." + configPrefix() + "_errlog"));
|
|
}
|
|
if (config.has("logger." + configPrefix() + "_stdout"))
|
|
{
|
|
cmd_args.push_back("--stdout-path");
|
|
cmd_args.push_back(config.getString("logger." + configPrefix() + "_stdout"));
|
|
}
|
|
if (config.has("logger." + configPrefix() + "_stderr"))
|
|
{
|
|
cmd_args.push_back("--stderr-path");
|
|
cmd_args.push_back(config.getString("logger." + configPrefix() + "_stderr"));
|
|
}
|
|
if (config.has("logger." + configPrefix() + "_level"))
|
|
{
|
|
cmd_args.push_back("--log-level");
|
|
cmd_args.push_back(config.getString("logger." + configPrefix() + "_level"));
|
|
}
|
|
|
|
LOG_TRACE(log, "Starting {}", serviceAlias());
|
|
|
|
return ShellCommand::executeDirect(path.toString(), cmd_args, true);
|
|
}
|
|
};
|
|
}
|