ClickHouse/dbms/include/DB/Storages/StorageDistributed.h

103 lines
4.1 KiB
C
Raw Normal View History

2012-05-21 20:38:34 +00:00
#pragma once
#include <DB/Storages/IStorage.h>
#include <DB/Client/ConnectionPool.h>
#include <DB/Client/ConnectionPoolWithFailover.h>
2012-07-26 20:16:57 +00:00
#include <DB/Interpreters/Settings.h>
2013-09-23 12:01:19 +00:00
#include <DB/Interpreters/Context.h>
2012-05-21 20:38:34 +00:00
namespace DB
{
/** Распределённая таблица, находящаяся на нескольких серверах.
* Использует данные заданной БД и таблицы на каждом сервере.
2012-05-30 04:45:49 +00:00
*
* Можно передать один адрес, а не несколько.
* В этом случае, таблицу можно считать удалённой, а не распределённой.
2012-05-21 20:38:34 +00:00
*/
class StorageDistributed : public IStorage
{
public:
static StoragePtr create(
2012-05-21 20:38:34 +00:00
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_ = "");
2012-05-21 20:38:34 +00:00
std::string getName() const { return "Distributed"; }
std::string getTableName() const { return name; }
std::string getSignColumnName() const { return sign_column_name; };
2012-12-15 04:39:13 +00:00
bool supportsSampling() const { return true; }
bool supportsFinal() const { return !sign_column_name.empty(); }
bool supportsPrewhere() const { return true; }
2012-05-21 20:38:34 +00:00
const NamesAndTypesList & getColumnsList() const { return *columns; }
NameAndTypePair getColumn(const String &column_name) const;
bool hasColumn(const String &column_name) const;
2012-05-21 20:38:34 +00:00
bool isRemote() const { return true; }
2012-05-21 20:38:34 +00:00
BlockInputStreams read(
const Names & column_names,
ASTPtr query,
const Settings & settings,
2012-05-22 18:32:45 +00:00
QueryProcessingStage::Enum & processed_stage,
2012-05-21 20:38:34 +00:00
size_t max_block_size = DEFAULT_BLOCK_SIZE,
2012-05-30 04:45:49 +00:00
unsigned threads = 1);
2012-05-21 20:38:34 +00:00
void dropImpl() {}
2012-06-18 06:19:13 +00:00
void rename(const String & new_path_to_db, const String & new_name) { name = new_name; }
2013-09-23 12:01:19 +00:00
/// в подтаблицах добавлять и удалять столбы нужно вручную
/// структура подтаблиц не проверяется
void alter(const ASTAlterQuery::Parameters &params);
2012-05-21 20:38:34 +00:00
Block getBlockWithVirtualColumns();
2012-05-21 20:38:34 +00:00
private:
StorageDistributed(
const std::string & name_,
NamesAndTypesListPtr columns_,
const String & remote_database_,
const String & remote_table_,
2013-12-07 16:51:29 +00:00
Cluster & cluster_,
2013-09-23 12:01:19 +00:00
const Context & context_,
const String & sign_column_name_ = "");
/// Создает копию запроса, меняет имена базы данных и таблицы, записавыет значения переменных host и port, если они не пустые.
ASTPtr remakeQuery(ASTPtr query, const String & host, size_t port);
2012-06-18 06:19:13 +00:00
String name;
2012-05-21 20:38:34 +00:00
NamesAndTypesListPtr columns;
String remote_database;
String remote_table;
String sign_column_name;
2012-05-21 20:38:34 +00:00
/// Имя виртуального столбца, куда записывается имя хоста (Например "_host").
String _host_column_name;
/// Имя виртуального столбца, куда записывается номер порта (Например "_port").
String _port_column_name;
2013-09-23 12:01:19 +00:00
const Context & context;
/// Используется только, если таблица должна владеть объектом Cluster, которым больше никто не владеет - для реализации TableFunctionRemote.
SharedPtr<Cluster> owned_cluster;
/// Соединения с удалёнными серверами.
2013-12-07 16:51:29 +00:00
Cluster & cluster;
2012-05-21 20:38:34 +00:00
};
}