mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-11 18:14:03 +00:00
109 lines
4.7 KiB
C++
109 lines
4.7 KiB
C++
#pragma once
|
||
|
||
#include <DB/Storages/IStorage.h>
|
||
#include <DB/Client/ConnectionPool.h>
|
||
#include <DB/Client/ConnectionPoolWithFailover.h>
|
||
#include <DB/Interpreters/Settings.h>
|
||
#include <DB/Interpreters/Context.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
/** Распределённая таблица, находящаяся на нескольких серверах.
|
||
* Использует данные заданной БД и таблицы на каждом сервере.
|
||
*
|
||
* Можно передать один адрес, а не несколько.
|
||
* В этом случае, таблицу можно считать удалённой, а не распределённой.
|
||
*/
|
||
class StorageDistributed : public IStorage
|
||
{
|
||
public:
|
||
static StoragePtr create(
|
||
const std::string & name_, /// Имя таблицы.
|
||
NamesAndTypesListPtr columns_, /// Список столбцов.
|
||
const String & remote_database_, /// БД на удалённых серверах.
|
||
const String & remote_table_, /// Имя таблицы на удалённых серверах.
|
||
const String & cluster_name,
|
||
Context & context_,
|
||
const String & sign_column_name_ = "");
|
||
|
||
static StoragePtr create(
|
||
const std::string & name_, /// Имя таблицы.
|
||
NamesAndTypesListPtr columns_, /// Список столбцов.
|
||
const String & remote_database_, /// БД на удалённых серверах.
|
||
const String & remote_table_, /// Имя таблицы на удалённых серверах.
|
||
SharedPtr<Cluster> & owned_cluster_,
|
||
Context & context_,
|
||
const String & sign_column_name_ = "");
|
||
|
||
std::string getName() const { return "Distributed"; }
|
||
std::string getTableName() const { return name; }
|
||
std::string getSignColumnName() const { return sign_column_name; };
|
||
bool supportsSampling() const { return true; }
|
||
bool supportsFinal() const { return !sign_column_name.empty(); }
|
||
bool supportsPrewhere() const { return true; }
|
||
|
||
const NamesAndTypesList & getColumnsList() const { return *columns; }
|
||
NameAndTypePair getColumn(const String &column_name) const;
|
||
bool hasColumn(const String &column_name) const;
|
||
|
||
bool isRemote() const { return true; }
|
||
/// Сохранить временные таблицы, чтобы при следующем вызове метода read переслать их на удаленные сервера
|
||
void storeExternalTables(const Tables & tables_) { external_tables = tables_; }
|
||
|
||
BlockInputStreams read(
|
||
const Names & column_names,
|
||
ASTPtr query,
|
||
const Settings & settings,
|
||
QueryProcessingStage::Enum & processed_stage,
|
||
size_t max_block_size = DEFAULT_BLOCK_SIZE,
|
||
unsigned threads = 1);
|
||
|
||
void drop() override {}
|
||
void rename(const String & new_path_to_db, const String & new_name) { name = new_name; }
|
||
/// в подтаблицах добавлять и удалять столбы нужно вручную
|
||
/// структура подтаблиц не проверяется
|
||
void alter(const ASTAlterQuery::Parameters ¶ms);
|
||
|
||
Block getBlockWithVirtualColumns();
|
||
|
||
private:
|
||
StorageDistributed(
|
||
const std::string & name_,
|
||
NamesAndTypesListPtr columns_,
|
||
const String & remote_database_,
|
||
const String & remote_table_,
|
||
Cluster & cluster_,
|
||
const Context & context_,
|
||
const String & sign_column_name_ = "");
|
||
|
||
/// Создает копию запроса, меняет имена базы данных и таблицы, записавыет значения переменных host и port, если они не пустые.
|
||
ASTPtr remakeQuery(ASTPtr query, const String & host, size_t port);
|
||
|
||
String name;
|
||
NamesAndTypesListPtr columns;
|
||
String remote_database;
|
||
String remote_table;
|
||
String sign_column_name;
|
||
|
||
/// Имя виртуального столбца, куда записывается имя хоста (Например "_host").
|
||
String _host_column_name;
|
||
/// Имя виртуального столбца, куда записывается номер порта (Например "_port").
|
||
String _port_column_name;
|
||
|
||
const Context & context;
|
||
|
||
/// Временные таблицы, которые необходимо отправить на сервер. Переменная очищается после каждого вызова метода read
|
||
/// Для подготовки к отправке нужно использовтаь метод storeExternalTables
|
||
Tables external_tables;
|
||
|
||
/// Используется только, если таблица должна владеть объектом Cluster, которым больше никто не владеет - для реализации TableFunctionRemote.
|
||
SharedPtr<Cluster> owned_cluster;
|
||
|
||
/// Соединения с удалёнными серверами.
|
||
Cluster & cluster;
|
||
};
|
||
|
||
}
|