mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-12 02:23:14 +00:00
398 lines
14 KiB
C++
398 lines
14 KiB
C++
#pragma once
|
||
|
||
#include <DB/Storages/IStorage.h>
|
||
#include <DB/Storages/MergeTree/MergeTreeData.h>
|
||
#include <DB/Storages/MergeTree/MergeTreeDataMerger.h>
|
||
#include <DB/Storages/MergeTree/MergeTreeDataWriter.h>
|
||
#include <DB/Storages/MergeTree/MergeTreeDataSelectExecutor.h>
|
||
#include <DB/Storages/MergeTree/ReplicatedMergeTreePartsExchange.h>
|
||
#include <zkutil/ZooKeeper.h>
|
||
#include <zkutil/LeaderElection.h>
|
||
#include <statdaemons/threadpool.hpp>
|
||
|
||
namespace DB
|
||
{
|
||
|
||
/** Движок, использующий merge-дерево и реплицируемый через ZooKeeper.
|
||
*/
|
||
class StorageReplicatedMergeTree : public IStorage
|
||
{
|
||
public:
|
||
/** Если !attach, либо создает новую таблицу в ZK, либо добавляет реплику в существующую таблицу.
|
||
*/
|
||
static StoragePtr create(
|
||
const String & zookeeper_path_,
|
||
const String & replica_name_,
|
||
bool attach,
|
||
const String & path_, const String & database_name_, const String & name_,
|
||
NamesAndTypesListPtr columns_,
|
||
Context & context_,
|
||
ASTPtr & primary_expr_ast_,
|
||
const String & date_column_name_,
|
||
const ASTPtr & sampling_expression_, /// nullptr, если семплирование не поддерживается.
|
||
size_t index_granularity_,
|
||
MergeTreeData::Mode mode_ = MergeTreeData::Ordinary,
|
||
const String & sign_column_ = "",
|
||
const MergeTreeSettings & settings_ = MergeTreeSettings());
|
||
|
||
void shutdown() override;
|
||
~StorageReplicatedMergeTree();
|
||
|
||
std::string getName() const override
|
||
{
|
||
return "Replicated" + data.getModePrefix() + "MergeTree";
|
||
}
|
||
|
||
std::string getTableName() const override { return table_name; }
|
||
std::string getSignColumnName() const { return data.getSignColumnName(); }
|
||
bool supportsSampling() const override { return data.supportsSampling(); }
|
||
bool supportsFinal() const override { return data.supportsFinal(); }
|
||
bool supportsPrewhere() const override { return data.supportsPrewhere(); }
|
||
|
||
const NamesAndTypesList & getColumnsList() const override { return data.getColumnsList(); }
|
||
|
||
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) override;
|
||
|
||
BlockOutputStreamPtr write(ASTPtr query) override;
|
||
|
||
bool optimize() override;
|
||
|
||
/** Удаляет реплику из ZooKeeper. Если других реплик нет, удаляет всю таблицу из ZooKeeper.
|
||
*/
|
||
void drop() override;
|
||
|
||
bool supportsIndexForIn() const override { return true; }
|
||
|
||
private:
|
||
friend class ReplicatedMergeTreeBlockOutputStream;
|
||
|
||
/// Добавляет куски в множество currently_merging.
|
||
struct CurrentlyMergingPartsTagger
|
||
{
|
||
Strings parts;
|
||
StorageReplicatedMergeTree & storage;
|
||
|
||
CurrentlyMergingPartsTagger(const Strings & parts_, StorageReplicatedMergeTree & storage_)
|
||
: parts(parts_), storage(storage_)
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(storage.currently_merging_mutex);
|
||
for (const auto & name : parts)
|
||
{
|
||
if (storage.currently_merging.count(name))
|
||
throw Exception("Tagging alreagy tagged part " + name + ". This is a bug.", ErrorCodes::LOGICAL_ERROR);
|
||
}
|
||
storage.currently_merging.insert(parts.begin(), parts.end());
|
||
}
|
||
|
||
~CurrentlyMergingPartsTagger()
|
||
{
|
||
try
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(storage.currently_merging_mutex);
|
||
for (const auto & name : parts)
|
||
{
|
||
if (!storage.currently_merging.count(name))
|
||
throw Exception("Untagging already untagged part " + name + ". This is a bug.", ErrorCodes::LOGICAL_ERROR);
|
||
storage.currently_merging.erase(name);
|
||
}
|
||
}
|
||
catch (...)
|
||
{
|
||
tryLogCurrentException(__PRETTY_FUNCTION__);
|
||
}
|
||
}
|
||
};
|
||
|
||
typedef Poco::SharedPtr<CurrentlyMergingPartsTagger> CurrentlyMergingPartsTaggerPtr;
|
||
|
||
/// Добавляет кусок в множество future_parts.
|
||
struct FuturePartTagger
|
||
{
|
||
String part;
|
||
StorageReplicatedMergeTree & storage;
|
||
|
||
FuturePartTagger(const String & part_, StorageReplicatedMergeTree & storage_)
|
||
: part(part_), storage(storage_)
|
||
{
|
||
if (!storage.future_parts.insert(part).second)
|
||
throw Exception("Tagging already tagged future part " + part + ". This is a bug.", ErrorCodes::LOGICAL_ERROR);
|
||
}
|
||
|
||
~FuturePartTagger()
|
||
{
|
||
try
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(storage.queue_mutex);
|
||
if (!storage.future_parts.erase(part))
|
||
throw Exception("Untagging already untagged future part " + part + ". This is a bug.", ErrorCodes::LOGICAL_ERROR);
|
||
}
|
||
catch (...)
|
||
{
|
||
tryLogCurrentException(__PRETTY_FUNCTION__);
|
||
}
|
||
}
|
||
};
|
||
|
||
typedef Poco::SharedPtr<FuturePartTagger> FuturePartTaggerPtr;
|
||
|
||
struct LogEntry
|
||
{
|
||
enum Type
|
||
{
|
||
GET_PART,
|
||
MERGE_PARTS,
|
||
};
|
||
|
||
String znode_name;
|
||
|
||
Type type;
|
||
String source_replica;
|
||
String new_part_name;
|
||
Strings parts_to_merge;
|
||
|
||
CurrentlyMergingPartsTaggerPtr currently_merging_tagger;
|
||
FuturePartTaggerPtr future_part_tagger;
|
||
|
||
void tagPartsAsCurrentlyMerging(StorageReplicatedMergeTree & storage)
|
||
{
|
||
if (type == MERGE_PARTS)
|
||
currently_merging_tagger = new CurrentlyMergingPartsTagger(parts_to_merge, storage);
|
||
}
|
||
|
||
void tagPartAsFuture(StorageReplicatedMergeTree & storage)
|
||
{
|
||
if (type == MERGE_PARTS || type == GET_PART)
|
||
future_part_tagger = new FuturePartTagger(new_part_name, storage);
|
||
}
|
||
|
||
void writeText(WriteBuffer & out) const;
|
||
void readText(ReadBuffer & in);
|
||
|
||
String toString() const
|
||
{
|
||
String s;
|
||
{
|
||
WriteBufferFromString out(s);
|
||
writeText(out);
|
||
}
|
||
return s;
|
||
}
|
||
|
||
static LogEntry parse(const String & s)
|
||
{
|
||
ReadBufferFromString in(s);
|
||
LogEntry res;
|
||
res.readText(in);
|
||
assertEOF(in);
|
||
return res;
|
||
}
|
||
};
|
||
|
||
typedef std::list<LogEntry> LogEntries;
|
||
|
||
typedef std::set<String> StringSet;
|
||
typedef std::vector<std::thread> Threads;
|
||
|
||
Context & context;
|
||
zkutil::ZooKeeperPtr zookeeper;
|
||
|
||
/// Куски, для которых в очереди есть задание на слияние.
|
||
StringSet currently_merging;
|
||
Poco::FastMutex currently_merging_mutex;
|
||
|
||
/** Очередь того, что нужно сделать на этой реплике, чтобы всех догнать. Берется из ZooKeeper (/replicas/me/queue/).
|
||
* В ZK записи в хронологическом порядке. Здесь - не обязательно.
|
||
*/
|
||
LogEntries queue;
|
||
Poco::FastMutex queue_mutex;
|
||
|
||
/** Куски, которые появятся в результате действий, выполняемых прямо сейчас фоновыми потоками (этих действий нет в очереди).
|
||
* Использовать под залоченным queue_mutex.
|
||
*/
|
||
StringSet future_parts;
|
||
|
||
String table_name;
|
||
String full_path;
|
||
|
||
String zookeeper_path;
|
||
String replica_name;
|
||
String replica_path;
|
||
|
||
/** /replicas/me/is_active.
|
||
*/
|
||
zkutil::EphemeralNodeHolderPtr replica_is_active_node;
|
||
|
||
/** Случайные данные, которые мы записали в /replicas/me/is_active.
|
||
*/
|
||
String active_node_identifier;
|
||
|
||
/** Является ли эта реплика "ведущей". Ведущая реплика выбирает куски для слияния.
|
||
*/
|
||
bool is_leader_node = false;
|
||
|
||
InterserverIOEndpointHolderPtr endpoint_holder;
|
||
|
||
MergeTreeData data;
|
||
MergeTreeDataSelectExecutor reader;
|
||
MergeTreeDataWriter writer;
|
||
MergeTreeDataMerger merger;
|
||
ReplicatedMergeTreePartsFetcher fetcher;
|
||
zkutil::LeaderElectionPtr leader_election;
|
||
|
||
/// Для чтения данных из директории unreplicated.
|
||
std::unique_ptr<MergeTreeData> unreplicated_data;
|
||
std::unique_ptr<MergeTreeDataSelectExecutor> unreplicated_reader;
|
||
std::unique_ptr<MergeTreeDataMerger> unreplicated_merger;
|
||
|
||
/// Поток, следящий за обновлениями в логах всех реплик и загружающий их в очередь.
|
||
std::thread queue_updating_thread;
|
||
|
||
/// Потоки, выполняющие действия из очереди.
|
||
Threads queue_threads;
|
||
|
||
/// Поток, выбирающий куски для слияния.
|
||
std::thread merge_selecting_thread;
|
||
/// Поток, удаляющий информацию о старых блоках из ZooKeeper.
|
||
std::thread clear_old_blocks_thread;
|
||
|
||
/// Поток, обрабатывающий переподключение к ZooKeeper при истечении сессии (очень маловероятное событие).
|
||
std::thread restarting_thread;
|
||
|
||
/// Когда последний раз выбрасывали старые логи из ZooKeeper.
|
||
time_t clear_old_logs_time = 0;
|
||
|
||
Logger * log;
|
||
|
||
/// Нужно ли завершить фоновые потоки (кроме restarting_thread).
|
||
volatile bool shutdown_called = false;
|
||
/// Нужно ли завершить restarting_thread.
|
||
volatile bool permanent_shutdown_called = false;
|
||
|
||
StorageReplicatedMergeTree(
|
||
const String & zookeeper_path_,
|
||
const String & replica_name_,
|
||
bool attach,
|
||
const String & path_, const String & database_name_, const String & name_,
|
||
NamesAndTypesListPtr columns_,
|
||
Context & context_,
|
||
ASTPtr & primary_expr_ast_,
|
||
const String & date_column_name_,
|
||
const ASTPtr & sampling_expression_,
|
||
size_t index_granularity_,
|
||
MergeTreeData::Mode mode_ = MergeTreeData::Ordinary,
|
||
const String & sign_column_ = "",
|
||
const MergeTreeSettings & settings_ = MergeTreeSettings());
|
||
|
||
/// Инициализация.
|
||
|
||
/** Проверяет, что в ZooKeeper в таблице нет данных.
|
||
*/
|
||
bool isTableEmpty();
|
||
|
||
/** Создает минимальный набор нод в ZooKeeper.
|
||
*/
|
||
void createTable();
|
||
void createReplica();
|
||
|
||
/** Отметить в ZooKeeper, что эта реплика сейчас активна.
|
||
*/
|
||
void activateReplica();
|
||
|
||
/** Проверить, что список столбцов и настройки таблицы совпадают с указанными в ZK (/metadata).
|
||
* Если нет - бросить исключение.
|
||
*/
|
||
void checkTableStructure();
|
||
|
||
/** Проверить, что множество кусков соответствует тому, что в ZK (/replicas/me/parts/).
|
||
* Если каких-то кусков, описанных в ZK нет локально, бросить исключение.
|
||
* Если какие-то локальные куски не упоминаются в ZK, удалить их.
|
||
* Но если таких слишком много, на всякий случай бросить исключение - скорее всего, это ошибка конфигурации.
|
||
*/
|
||
void checkParts();
|
||
|
||
/// Запустить или остановить фоновые потоки. Используется для частичной переинициализации при пересоздании сессии в ZooKeeper.
|
||
void startup();
|
||
void partialShutdown();
|
||
|
||
|
||
/** Проверить, что чексумма куска совпадает с чексуммой того же куска на какой-нибудь другой реплике.
|
||
* Если ни у кого нет такого куска, ничего не проверяет.
|
||
* Не очень надежно: если две реплики добавляют кусок почти одновременно, ни одной проверки не произойдет.
|
||
* Кладет в ops действия, добавляющие данные о куске в ZooKeeper.
|
||
*/
|
||
void checkPartAndAddToZooKeeper(MergeTreeData::DataPartPtr part, zkutil::Ops & ops);
|
||
|
||
void clearOldParts();
|
||
|
||
/// Удалить из ZooKeeper старые записи в логе.
|
||
void clearOldLogs();
|
||
|
||
/// Удалить из ZooKeeper старые хеши блоков. Это делает ведущая реплика.
|
||
void clearOldBlocks();
|
||
|
||
/// Выполнение заданий из очереди.
|
||
|
||
/** Кладет в queue записи из ZooKeeper (/replicas/me/queue/).
|
||
*/
|
||
void loadQueue();
|
||
|
||
/** Копирует новые записи из логов всех реплик в очередь этой реплики.
|
||
*/
|
||
void pullLogsToQueue();
|
||
|
||
/** Можно ли сейчас попробовать выполнить это действие. Если нет, нужно оставить его в очереди и попробовать выполнить другое.
|
||
* Вызывается под queue_mutex.
|
||
*/
|
||
bool shouldExecuteLogEntry(const LogEntry & entry);
|
||
|
||
/** Выполнить действие из очереди. Бросает исключение, если что-то не так.
|
||
*/
|
||
void executeLogEntry(const LogEntry & entry);
|
||
|
||
/** В бесконечном цикле обновляет очередь.
|
||
*/
|
||
void queueUpdatingThread();
|
||
|
||
/** В бесконечном цикле выполняет действия из очереди.
|
||
*/
|
||
void queueThread();
|
||
|
||
/// Выбор кусков для слияния.
|
||
|
||
void becomeLeader();
|
||
|
||
/** В бесконечном цикле выбирает куски для слияния и записывает в лог.
|
||
*/
|
||
void mergeSelectingThread();
|
||
|
||
/** В бесконечном цикле вызывает clearOldBlocks.
|
||
*/
|
||
void clearOldBlocksThread();
|
||
|
||
/** В бесконечном цикле проверяет, не протухла ли сессия в ZooKeeper.
|
||
*/
|
||
void restartingThread();
|
||
|
||
/// Вызывается во время выбора кусков для слияния.
|
||
bool canMergeParts(const MergeTreeData::DataPartPtr & left, const MergeTreeData::DataPartPtr & right);
|
||
|
||
/// Обмен кусками.
|
||
|
||
/** Возвращает пустую строку, если куска ни у кого нет.
|
||
*/
|
||
String findReplicaHavingPart(const String & part_name, bool active);
|
||
|
||
/** Скачать указанный кусок с указанной реплики.
|
||
*/
|
||
void fetchPart(const String & part_name, const String & replica_name);
|
||
|
||
};
|
||
|
||
}
|