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

103 lines
3.9 KiB
C++
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#pragma once
#include <DB/Storages/IStorage.h>
#include <DB/Client/ConnectionPool.h>
#include <DB/Client/ConnectionPoolWithFailover.h>
#include <DB/Interpreters/Settings.h>
namespace DB
{
/** Распределённая таблица, находящаяся на нескольких серверах.
* Использует данные заданной БД и таблицы на каждом сервере.
*
* Можно передать один адрес, а не несколько.
* В этом случае, таблицу можно считать удалённой, а не распределённой.
*/
class StorageDistributed : public IStorage
{
public:
/// Массив шардов. Каждый шард - адреса одного сервера.
typedef std::vector<Poco::Net::SocketAddress> Addresses;
/// Массив шардов. Для каждого шарда - массив адресов реплик (серверов, считающихся идентичными).
typedef std::vector<Addresses> AddressesWithFailover;
static StoragePtr create(
const std::string & name_, /// Имя таблицы.
NamesAndTypesListPtr columns_, /// Список столбцов.
const Addresses & addresses, /// Адреса удалённых серверов.
const String & remote_database_, /// БД на удалённых серверах.
const String & remote_table_, /// Имя таблицы на удалённых серверах.
const DataTypeFactory & data_type_factory_,
const Settings & settings,
const String & sign_column_name_ = "");
/// Использовать реплики для отказоустойчивости.
static StoragePtr create(
const std::string & name_, /// Имя таблицы.
NamesAndTypesListPtr columns_, /// Список столбцов.
const AddressesWithFailover & addresses, /// Адреса удалённых серверов с учётом реплик.
const String & remote_database_, /// БД на удалённых серверах.
const String & remote_table_, /// Имя таблицы на удалённых серверах.
const DataTypeFactory & data_type_factory_,
const Settings & settings,
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(); }
const NamesAndTypesList & getColumnsList() const { return *columns; }
bool isRemote() const { return true; }
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 dropImpl() {}
void rename(const String & new_path_to_db, const String & new_name) { name = new_name; }
private:
StorageDistributed(
const std::string & name_,
NamesAndTypesListPtr columns_,
const Addresses & addresses,
const String & remote_database_,
const String & remote_table_,
const DataTypeFactory & data_type_factory_,
const Settings & settings,
const String & sign_column_name_ = "");
/// Использовать реплики для отказоустойчивости.
StorageDistributed(
const std::string & name_,
NamesAndTypesListPtr columns_,
const AddressesWithFailover & addresses,
const String & remote_database_,
const String & remote_table_,
const DataTypeFactory & data_type_factory_,
const Settings & settings,
const String & sign_column_name_ = "");
String name;
NamesAndTypesListPtr columns;
String remote_database;
String remote_table;
const DataTypeFactory & data_type_factory;
String sign_column_name;
/// Соединения с удалёнными серверами.
ConnectionPools pools;
};
}