ClickHouse/dbms/src/Storages/StorageReplicatedMergeTree.cpp

1988 lines
67 KiB
C++
Raw Normal View History

2014-03-21 13:42:14 +00:00
#include <DB/Storages/StorageReplicatedMergeTree.h>
2014-04-02 10:10:37 +00:00
#include <DB/Storages/MergeTree/ReplicatedMergeTreeBlockOutputStream.h>
2014-04-02 13:45:39 +00:00
#include <DB/Storages/MergeTree/ReplicatedMergeTreePartsExchange.h>
2014-07-22 13:49:52 +00:00
#include <DB/Storages/MergeTree/MergeTreePartChecker.h>
2014-03-22 14:44:44 +00:00
#include <DB/Parsers/formatAST.h>
#include <DB/IO/WriteBufferFromOStream.h>
#include <DB/IO/ReadBufferFromString.h>
2014-07-10 10:16:50 +00:00
#include <DB/Interpreters/InterpreterAlterQuery.h>
2014-05-13 10:10:26 +00:00
#include <time.h>
2014-03-21 13:42:14 +00:00
namespace DB
{
2014-04-03 11:48:28 +00:00
2014-07-15 14:37:49 +00:00
const auto ERROR_SLEEP_MS = 1000;
2014-07-02 10:16:49 +00:00
const auto MERGE_SELECTING_SLEEP_MS = 5 * 1000;
2014-07-15 14:37:49 +00:00
const auto CLEANUP_SLEEP_MS = 30 * 1000;
2014-04-03 11:48:28 +00:00
2014-07-25 13:11:02 +00:00
/// Преобразовать число в строку формате суффиксов автоинкрементных нод в ZooKeeper.
static String padIndex(UInt64 index)
{
String index_str = toString(index);
while (index_str.size() < 10)
index_str = '0' + index_str;
return index_str;
}
2014-04-03 11:48:28 +00:00
2014-03-21 13:42:14 +00:00
StorageReplicatedMergeTree::StorageReplicatedMergeTree(
const String & zookeeper_path_,
const String & replica_name_,
2014-03-21 19:17:59 +00:00
bool attach,
2014-05-08 07:12:01 +00:00
const String & path_, const String & database_name_, const String & name_,
NamesAndTypesListPtr columns_,
2014-03-22 14:44:44 +00:00
Context & context_,
2014-03-21 13:42:14 +00:00
ASTPtr & primary_expr_ast_,
const String & date_column_name_,
const ASTPtr & sampling_expression_,
size_t index_granularity_,
MergeTreeData::Mode mode_,
const String & sign_column_,
const MergeTreeSettings & settings_)
:
context(context_), zookeeper(context.getZooKeeper()), database_name(database_name_),
2014-04-04 12:47:57 +00:00
table_name(name_), full_path(path_ + escapeForFileName(table_name) + '/'), zookeeper_path(zookeeper_path_),
2014-04-14 10:19:33 +00:00
replica_name(replica_name_),
2014-03-21 13:42:14 +00:00
data( full_path, columns_, context_, primary_expr_ast_, date_column_name_, sampling_expression_,
2014-07-23 09:15:41 +00:00
index_granularity_, mode_, sign_column_, settings_, database_name_ + "." + table_name, true,
std::bind(&StorageReplicatedMergeTree::enqueuePartForCheck, this, std::placeholders::_1)),
2014-04-04 10:37:33 +00:00
reader(data), writer(data), merger(data), fetcher(data),
2014-07-04 13:59:05 +00:00
log(&Logger::get(database_name_ + "." + table_name + " (StorageReplicatedMergeTree)")),
shutdown_event(false), permanent_shutdown_event(false)
2014-03-21 13:42:14 +00:00
{
2014-05-13 11:24:04 +00:00
if (!zookeeper)
{
2014-07-23 13:58:38 +00:00
if (!attach)
throw Exception("Can't create replicated table without ZooKeeper", ErrorCodes::NO_ZOOKEEPER);
2014-05-13 11:24:04 +00:00
goReadOnly();
return;
}
2014-03-22 14:44:44 +00:00
if (!zookeeper_path.empty() && *zookeeper_path.rbegin() == '/')
zookeeper_path.erase(zookeeper_path.end() - 1);
replica_path = zookeeper_path + "/replicas/" + replica_name;
2014-03-21 19:17:59 +00:00
if (!attach)
{
2014-05-13 10:10:26 +00:00
if (!zookeeper->exists(zookeeper_path))
2014-03-22 14:44:44 +00:00
createTable();
2014-07-10 08:40:59 +00:00
checkTableStructure(false);
2014-07-10 10:16:50 +00:00
createReplica();
2014-03-21 19:17:59 +00:00
}
else
{
2014-07-10 08:40:59 +00:00
bool skip_sanity_checks = false;
if (zookeeper->exists(replica_path + "/flags/force_restore_data"))
{
skip_sanity_checks = true;
zookeeper->remove(replica_path + "/flags/force_restore_data");
2014-07-10 10:16:50 +00:00
LOG_WARNING(log, "Skipping the limits on severity of changes to data parts and columns (flag "
<< replica_path << "/flags/force_restore_data).");
2014-07-10 08:40:59 +00:00
}
checkTableStructure(skip_sanity_checks);
checkParts(skip_sanity_checks);
2014-03-21 19:17:59 +00:00
}
2014-03-22 14:44:44 +00:00
2014-05-26 11:40:22 +00:00
initVirtualParts();
2014-04-03 08:47:59 +00:00
loadQueue();
2014-04-24 10:20:02 +00:00
String unreplicated_path = full_path + "unreplicated/";
if (Poco::File(unreplicated_path).exists())
{
LOG_INFO(log, "Have unreplicated data");
unreplicated_data.reset(new MergeTreeData(unreplicated_path, columns_, context_, primary_expr_ast_,
2014-05-08 07:12:01 +00:00
date_column_name_, sampling_expression_, index_granularity_, mode_, sign_column_, settings_,
2014-07-09 13:39:19 +00:00
database_name_ + "." + table_name + "[unreplicated]", false));
2014-04-24 10:20:02 +00:00
unreplicated_reader.reset(new MergeTreeDataSelectExecutor(*unreplicated_data));
2014-05-08 08:03:03 +00:00
unreplicated_merger.reset(new MergeTreeDataMerger(*unreplicated_data));
2014-04-24 10:20:02 +00:00
}
2014-04-25 13:55:15 +00:00
2014-05-13 10:10:26 +00:00
/// Сгенерируем этому экземпляру случайный идентификатор.
struct timespec times;
if (clock_gettime(CLOCK_THREAD_CPUTIME_ID, &times))
throwFromErrno("Cannot clock_gettime.", ErrorCodes::CANNOT_CLOCK_GETTIME);
active_node_identifier = toString(times.tv_nsec);
2014-04-25 13:55:15 +00:00
restarting_thread = std::thread(&StorageReplicatedMergeTree::restartingThread, this);
2014-03-21 13:42:14 +00:00
}
StoragePtr StorageReplicatedMergeTree::create(
const String & zookeeper_path_,
const String & replica_name_,
2014-03-21 19:17:59 +00:00
bool attach,
2014-05-08 07:12:01 +00:00
const String & path_, const String & database_name_, const String & name_,
NamesAndTypesListPtr columns_,
2014-03-22 14:44:44 +00:00
Context & context_,
2014-03-21 13:42:14 +00:00
ASTPtr & primary_expr_ast_,
const String & date_column_name_,
const ASTPtr & sampling_expression_,
size_t index_granularity_,
MergeTreeData::Mode mode_,
const String & sign_column_,
const MergeTreeSettings & settings_)
{
2014-04-02 07:59:43 +00:00
StorageReplicatedMergeTree * res = new StorageReplicatedMergeTree(zookeeper_path_, replica_name_, attach,
2014-05-08 07:12:01 +00:00
path_, database_name_, name_, columns_, context_, primary_expr_ast_, date_column_name_, sampling_expression_,
2014-04-02 07:59:43 +00:00
index_granularity_, mode_, sign_column_, settings_);
StoragePtr res_ptr = res->thisPtr();
2014-05-13 11:24:04 +00:00
if (!res->is_read_only)
{
String endpoint_name = "ReplicatedMergeTree:" + res->replica_path;
2014-07-22 13:49:52 +00:00
InterserverIOEndpointPtr endpoint = new ReplicatedMergeTreePartsServer(res->data, *res);
2014-05-13 11:24:04 +00:00
res->endpoint_holder = new InterserverIOEndpointHolder(endpoint_name, endpoint, res->context.getInterserverIOHandler());
}
2014-04-02 07:59:43 +00:00
return res_ptr;
2014-03-21 13:42:14 +00:00
}
2014-03-22 14:44:44 +00:00
static String formattedAST(const ASTPtr & ast)
{
if (!ast)
return "";
std::stringstream ss;
formatAST(*ast, ss, 0, false, true);
return ss.str();
}
2014-03-21 19:17:59 +00:00
2014-03-22 14:44:44 +00:00
void StorageReplicatedMergeTree::createTable()
2014-03-21 19:17:59 +00:00
{
2014-05-26 18:27:25 +00:00
LOG_DEBUG(log, "Creating table " << zookeeper_path);
2014-05-13 10:10:26 +00:00
zookeeper->create(zookeeper_path, "", zkutil::CreateMode::Persistent);
2014-03-21 19:17:59 +00:00
2014-07-10 10:16:50 +00:00
/// Запишем метаданные таблицы, чтобы реплики могли сверять с ними параметры таблицы.
2014-03-22 14:44:44 +00:00
std::stringstream metadata;
metadata << "metadata format version: 1" << std::endl;
metadata << "date column: " << data.date_column_name << std::endl;
metadata << "sampling expression: " << formattedAST(data.sampling_expression) << std::endl;
metadata << "index granularity: " << data.index_granularity << std::endl;
metadata << "mode: " << static_cast<int>(data.mode) << std::endl;
metadata << "sign column: " << data.sign_column << std::endl;
metadata << "primary key: " << formattedAST(data.primary_expr_ast) << std::endl;
2014-03-21 19:17:59 +00:00
2014-05-13 10:10:26 +00:00
zookeeper->create(zookeeper_path + "/metadata", metadata.str(), zkutil::CreateMode::Persistent);
2014-07-10 10:16:50 +00:00
zookeeper->create(zookeeper_path + "/columns", data.getColumnsList().toString(), zkutil::CreateMode::Persistent);
2014-03-21 19:17:59 +00:00
2014-07-15 14:37:49 +00:00
zookeeper->create(zookeeper_path + "/log", "", zkutil::CreateMode::Persistent);
2014-05-13 10:10:26 +00:00
zookeeper->create(zookeeper_path + "/blocks", "", zkutil::CreateMode::Persistent);
zookeeper->create(zookeeper_path + "/block_numbers", "", zkutil::CreateMode::Persistent);
zookeeper->create(zookeeper_path + "/leader_election", "", zkutil::CreateMode::Persistent);
zookeeper->create(zookeeper_path + "/temp", "", zkutil::CreateMode::Persistent);
2014-07-25 12:11:06 +00:00
zookeeper->create(zookeeper_path + "/flags", "", zkutil::CreateMode::Persistent);
2014-07-15 14:37:49 +00:00
/// Создадим replicas в последнюю очередь, чтобы нельзя было добавить реплику, пока все остальные ноды не созданы.
2014-07-10 10:16:50 +00:00
zookeeper->create(zookeeper_path + "/replicas", "", zkutil::CreateMode::Persistent);
2014-03-22 14:44:44 +00:00
}
2014-03-21 19:17:59 +00:00
2014-03-22 14:44:44 +00:00
/** Проверить, что список столбцов и настройки таблицы совпадают с указанными в ZK (/metadata).
* Если нет - бросить исключение.
2014-03-21 19:17:59 +00:00
*/
2014-07-10 08:40:59 +00:00
void StorageReplicatedMergeTree::checkTableStructure(bool skip_sanity_checks)
2014-03-22 14:44:44 +00:00
{
2014-05-13 10:10:26 +00:00
String metadata_str = zookeeper->get(zookeeper_path + "/metadata");
2014-03-22 14:44:44 +00:00
ReadBufferFromString buf(metadata_str);
assertString("metadata format version: 1", buf);
assertString("\ndate column: ", buf);
assertString(data.date_column_name, buf);
assertString("\nsampling expression: ", buf);
assertString(formattedAST(data.sampling_expression), buf);
assertString("\nindex granularity: ", buf);
assertString(toString(data.index_granularity), buf);
assertString("\nmode: ", buf);
assertString(toString(static_cast<int>(data.mode)), buf);
assertString("\nsign column: ", buf);
assertString(data.sign_column, buf);
assertString("\nprimary key: ", buf);
2014-07-10 10:16:50 +00:00
/// NOTE: Можно сделать менее строгую проверку совпадения выражений, чтобы таблицы не ломались от небольших изменений
/// в коде formatAST.
2014-03-22 14:44:44 +00:00
assertString(formattedAST(data.primary_expr_ast), buf);
2014-07-15 11:42:06 +00:00
assertString("\n", buf);
2014-07-10 08:40:59 +00:00
assertEOF(buf);
2014-07-10 10:16:50 +00:00
zkutil::Stat stat;
auto columns = NamesAndTypesList::parse(zookeeper->get(zookeeper_path + "/columns", &stat), context.getDataTypeFactory());
columns_version = stat.version;
2014-07-10 08:40:59 +00:00
if (columns != data.getColumnsList())
2014-03-22 14:44:44 +00:00
{
2014-07-10 08:40:59 +00:00
if (data.getColumnsList().sizeOfDifference(columns) <= 2 || skip_sanity_checks)
2014-07-10 10:16:50 +00:00
{
2014-07-10 08:40:59 +00:00
LOG_WARNING(log, "Table structure in ZooKeeper is a little different from local table structure. Assuming ALTER.");
/// Без всяких блокировок, потому что таблица еще не создана.
InterpreterAlterQuery::updateMetadata(database_name, table_name, columns, context);
data.setColumnsList(columns);
2014-07-10 10:16:50 +00:00
}
2014-07-10 08:40:59 +00:00
else
2014-07-10 10:16:50 +00:00
{
2014-07-10 08:40:59 +00:00
throw Exception("Table structure in ZooKeeper is very different from local table structure.",
ErrorCodes::INCOMPATIBLE_COLUMNS);
2014-07-10 10:16:50 +00:00
}
2014-03-22 14:44:44 +00:00
}
}
2014-03-21 19:17:59 +00:00
2014-03-22 14:44:44 +00:00
void StorageReplicatedMergeTree::createReplica()
{
2014-05-26 18:29:29 +00:00
LOG_DEBUG(log, "Creating replica " << replica_path);
2014-05-26 18:27:25 +00:00
2014-05-26 18:14:52 +00:00
/** Запомним список других реплик.
* NOTE: Здесь есть race condition. Если почти одновременно добавить нескольких реплик, сразу же начиная в них писать,
* небольшая часть данных может не реплицироваться.
*/
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
/// Создадим пустую реплику.
2014-05-13 10:10:26 +00:00
zookeeper->create(replica_path, "", zkutil::CreateMode::Persistent);
2014-07-16 08:58:59 +00:00
zookeeper->create(replica_path + "/columns", data.getColumnsList().toString(), zkutil::CreateMode::Persistent);
2014-05-13 10:10:26 +00:00
zookeeper->create(replica_path + "/host", "", zkutil::CreateMode::Persistent);
2014-07-15 14:37:49 +00:00
zookeeper->create(replica_path + "/log_pointer", "", zkutil::CreateMode::Persistent);
2014-05-13 10:10:26 +00:00
zookeeper->create(replica_path + "/queue", "", zkutil::CreateMode::Persistent);
zookeeper->create(replica_path + "/parts", "", zkutil::CreateMode::Persistent);
2014-05-26 18:14:52 +00:00
2014-07-15 14:37:49 +00:00
/** Нужно изменить данные ноды /replicas на что угодно, чтобы поток, удаляющий старые записи в логе,
* споткнулся об это изменение и не удалил записи, которые мы еще не прочитали.
*/
zookeeper->set(zookeeper_path + "/replicas", "last added replica: " + replica_name);
2014-05-26 18:14:52 +00:00
if (replicas.empty())
2014-05-26 18:27:25 +00:00
{
LOG_DEBUG(log, "No other replicas");
2014-05-26 18:14:52 +00:00
return;
2014-05-26 18:27:25 +00:00
}
2014-05-26 18:14:52 +00:00
2014-07-15 14:37:49 +00:00
/// "Эталонная" реплика, у которой мы возьмем информацию о множестве кусков, очередь и указатель на лог.
String source_replica = replicas[rand() % replicas.size()];
2014-05-26 18:14:52 +00:00
2014-05-26 18:27:25 +00:00
LOG_INFO(log, "Will mimic " << source_replica);
2014-05-26 18:14:52 +00:00
String source_path = zookeeper_path + "/replicas/" + source_replica;
/// Порядок следующих трех действий важен. Записи в логе могут продублироваться, но не могут потеряться.
2014-07-15 14:37:49 +00:00
/// Скопируем у эталонной реплики ссылку на лог.
zookeeper->set(replica_path + "/log_pointer", zookeeper->get(source_path + "/log_pointer"));
2014-05-26 18:14:52 +00:00
/// Запомним очередь эталонной реплики.
Strings source_queue_names = zookeeper->getChildren(source_path + "/queue");
std::sort(source_queue_names.begin(), source_queue_names.end());
Strings source_queue;
for (const String & entry_name : source_queue_names)
{
String entry;
if (!zookeeper->tryGet(source_path + "/queue/" + entry_name, entry))
continue;
source_queue.push_back(entry);
}
/// Добавим в очередь задания на получение всех активных кусков, которые есть у эталонной реплики.
Strings parts = zookeeper->getChildren(source_path + "/parts");
ActiveDataPartSet active_parts_set;
for (const String & part : parts)
{
active_parts_set.add(part);
}
Strings active_parts = active_parts_set.getParts();
for (const String & name : active_parts)
{
LogEntry log_entry;
log_entry.type = LogEntry::GET_PART;
log_entry.source_replica = "";
log_entry.new_part_name = name;
zookeeper->create(replica_path + "/queue/queue-", log_entry.toString(), zkutil::CreateMode::PersistentSequential);
}
2014-05-26 18:27:25 +00:00
LOG_DEBUG(log, "Queued " << active_parts.size() << " parts to be fetched");
2014-05-26 18:14:52 +00:00
/// Добавим в очередь содержимое очереди эталонной реплики.
for (const String & entry : source_queue)
{
zookeeper->create(replica_path + "/queue/queue-", entry, zkutil::CreateMode::PersistentSequential);
}
2014-05-26 18:27:25 +00:00
LOG_DEBUG(log, "Copied " << source_queue.size() << " queue entries");
2014-03-22 14:44:44 +00:00
}
2014-03-21 19:17:59 +00:00
2014-03-22 14:44:44 +00:00
void StorageReplicatedMergeTree::activateReplica()
{
std::stringstream host;
host << "host: " << context.getInterserverIOHost() << std::endl;
host << "port: " << context.getInterserverIOPort() << std::endl;
2014-03-21 19:17:59 +00:00
2014-05-13 10:10:26 +00:00
/** Если нода отмечена как активная, но отметка сделана в этом же экземпляре, удалим ее.
* Такое возможно только при истечении сессии в ZooKeeper.
* Здесь есть небольшой race condition (можем удалить не ту ноду, для которой сделали tryGet),
* но он крайне маловероятен при нормальном использовании.
*/
String data;
if (zookeeper->tryGet(replica_path + "/is_active", data) && data == active_node_identifier)
zookeeper->tryRemove(replica_path + "/is_active");
2014-04-03 08:47:59 +00:00
/// Одновременно объявим, что эта реплика активна, и обновим хост.
2014-03-22 14:44:44 +00:00
zkutil::Ops ops;
2014-05-13 10:10:26 +00:00
ops.push_back(new zkutil::Op::Create(replica_path + "/is_active", "", zookeeper->getDefaultACL(), zkutil::CreateMode::Ephemeral));
2014-03-22 14:44:44 +00:00
ops.push_back(new zkutil::Op::SetData(replica_path + "/host", host.str(), -1));
2014-04-03 13:11:11 +00:00
try
{
2014-05-13 10:10:26 +00:00
zookeeper->multi(ops);
2014-04-03 13:11:11 +00:00
}
catch (zkutil::KeeperException & e)
{
if (e.code == ZNODEEXISTS)
2014-04-03 13:11:11 +00:00
throw Exception("Replica " + replica_path + " appears to be already active. If you're sure it's not, "
"try again in a minute or remove znode " + replica_path + "/is_active manually", ErrorCodes::REPLICA_IS_ALREADY_ACTIVE);
throw;
}
2014-03-21 19:17:59 +00:00
2014-05-13 10:10:26 +00:00
replica_is_active_node = zkutil::EphemeralNodeHolder::existing(replica_path + "/is_active", *zookeeper);
2014-03-22 14:44:44 +00:00
}
2014-03-21 19:17:59 +00:00
2014-07-10 08:40:59 +00:00
void StorageReplicatedMergeTree::checkParts(bool skip_sanity_checks)
2014-04-02 07:59:43 +00:00
{
2014-05-13 10:10:26 +00:00
Strings expected_parts_vec = zookeeper->getChildren(replica_path + "/parts");
2014-05-22 10:37:17 +00:00
/// Куски в ZK.
2014-04-02 07:59:43 +00:00
NameSet expected_parts(expected_parts_vec.begin(), expected_parts_vec.end());
2014-04-09 16:32:32 +00:00
MergeTreeData::DataParts parts = data.getAllDataParts();
2014-04-02 07:59:43 +00:00
2014-05-22 10:37:17 +00:00
/// Локальные куски, которых нет в ZK.
2014-04-08 17:45:21 +00:00
MergeTreeData::DataParts unexpected_parts;
2014-05-22 10:37:17 +00:00
2014-04-02 07:59:43 +00:00
for (const auto & part : parts)
{
if (expected_parts.count(part->name))
{
expected_parts.erase(part->name);
}
else
{
2014-04-08 17:45:21 +00:00
unexpected_parts.insert(part);
2014-04-02 07:59:43 +00:00
}
}
2014-05-22 10:37:17 +00:00
/// Какие локальные куски добавить в ZK.
2014-04-08 17:45:21 +00:00
MergeTreeData::DataPartsVector parts_to_add;
2014-05-22 10:37:17 +00:00
/// Какие куски нужно забрать с других реплик.
Strings parts_to_fetch;
2014-04-08 17:45:21 +00:00
for (const String & missing_name : expected_parts)
{
2014-05-22 10:37:17 +00:00
/// Если локально не хватает какого-то куска, но есть покрывающий его кусок, можно заменить в ZK недостающий покрывающим.
2014-07-25 11:38:46 +00:00
auto containing = data.getActiveContainingPart(missing_name);
2014-05-22 10:37:17 +00:00
if (containing)
{
LOG_ERROR(log, "Ignoring missing local part " << missing_name << " because part " << containing->name << " exists");
if (unexpected_parts.count(containing))
{
parts_to_add.push_back(containing);
unexpected_parts.erase(containing);
}
}
else
2014-04-08 17:45:21 +00:00
{
2014-05-22 10:37:17 +00:00
parts_to_fetch.push_back(missing_name);
2014-04-08 17:45:21 +00:00
}
}
2014-04-02 07:59:43 +00:00
2014-05-22 10:37:17 +00:00
for (const String & name : parts_to_fetch)
expected_parts.erase(name);
String sanity_report =
"There are " + toString(unexpected_parts.size()) + " unexpected parts, "
+ toString(parts_to_add.size()) + " unexpectedly merged parts, "
2014-07-24 16:29:37 +00:00
+ toString(expected_parts.size()) + " missing obsolete parts, "
2014-05-22 10:37:17 +00:00
+ toString(parts_to_fetch.size()) + " missing parts";
bool insane =
parts_to_add.size() > 2 ||
2014-04-08 17:45:21 +00:00
unexpected_parts.size() > 2 ||
2014-05-22 10:37:17 +00:00
expected_parts.size() > 20 ||
parts_to_fetch.size() > 2;
2014-07-10 08:40:59 +00:00
if (insane && !skip_sanity_checks)
2014-05-22 10:37:17 +00:00
{
throw Exception("The local set of parts of table " + getTableName() + " doesn't look like the set of parts in ZooKeeper. "
+ sanity_report,
2014-04-02 07:59:43 +00:00
ErrorCodes::TOO_MANY_UNEXPECTED_DATA_PARTS);
2014-04-08 17:45:21 +00:00
}
2014-07-10 10:16:50 +00:00
if (insane)
{
LOG_WARNING(log, sanity_report);
}
2014-04-08 17:45:21 +00:00
/// Добавим в ZK информацию о кусках, покрывающих недостающие куски.
for (MergeTreeData::DataPartPtr part : parts_to_add)
{
2014-04-09 16:32:32 +00:00
LOG_ERROR(log, "Adding unexpected local part to ZooKeeper: " << part->name);
2014-05-22 10:37:17 +00:00
2014-04-08 17:45:21 +00:00
zkutil::Ops ops;
2014-07-15 09:56:17 +00:00
checkPartAndAddToZooKeeper(part, ops);
2014-05-13 10:10:26 +00:00
zookeeper->multi(ops);
2014-04-08 17:45:21 +00:00
}
2014-04-02 07:59:43 +00:00
2014-04-08 17:45:21 +00:00
/// Удалим из ZK информацию о кусках, покрытых только что добавленными.
for (const String & name : expected_parts)
{
2014-05-22 10:37:17 +00:00
LOG_ERROR(log, "Removing unexpectedly merged local part from ZooKeeper: " << name);
zkutil::Ops ops;
2014-07-10 08:40:59 +00:00
ops.push_back(new zkutil::Op::Remove(replica_path + "/parts/" + name + "/columns", -1));
2014-05-22 10:37:17 +00:00
ops.push_back(new zkutil::Op::Remove(replica_path + "/parts/" + name + "/checksums", -1));
ops.push_back(new zkutil::Op::Remove(replica_path + "/parts/" + name, -1));
zookeeper->multi(ops);
}
/// Добавим в очередь задание забрать недостающие куски с других реплик и уберем из ZK информацию, что они у нас есть.
for (const String & name : parts_to_fetch)
{
LOG_ERROR(log, "Removing missing part from ZooKeeper and queueing a fetch: " << name);
LogEntry log_entry;
log_entry.type = LogEntry::GET_PART;
2014-05-26 17:12:46 +00:00
log_entry.source_replica = "";
2014-05-22 10:37:17 +00:00
log_entry.new_part_name = name;
/// Полагаемся, что это происходит до загрузки очереди (loadQueue).
2014-04-08 17:45:21 +00:00
zkutil::Ops ops;
2014-07-10 08:40:59 +00:00
ops.push_back(new zkutil::Op::Remove(replica_path + "/parts/" + name + "/columns", -1));
2014-04-08 17:45:21 +00:00
ops.push_back(new zkutil::Op::Remove(replica_path + "/parts/" + name + "/checksums", -1));
ops.push_back(new zkutil::Op::Remove(replica_path + "/parts/" + name, -1));
2014-05-22 10:37:17 +00:00
ops.push_back(new zkutil::Op::Create(
replica_path + "/queue/queue-", log_entry.toString(), zookeeper->getDefaultACL(), zkutil::CreateMode::PersistentSequential));
2014-05-13 10:10:26 +00:00
zookeeper->multi(ops);
2014-04-08 17:45:21 +00:00
}
/// Удалим лишние локальные куски.
2014-04-02 07:59:43 +00:00
for (MergeTreeData::DataPartPtr part : unexpected_parts)
{
2014-05-22 10:37:17 +00:00
LOG_ERROR(log, "Renaming unexpected part " << part->name << " to ignored_" + part->name);
2014-04-02 10:10:37 +00:00
data.renameAndDetachPart(part, "ignored_");
2014-04-02 07:59:43 +00:00
}
}
2014-03-21 19:17:59 +00:00
2014-05-26 11:40:22 +00:00
void StorageReplicatedMergeTree::initVirtualParts()
{
auto parts = data.getDataParts();
for (const auto & part : parts)
{
virtual_parts.add(part->name);
}
}
2014-07-15 09:56:17 +00:00
void StorageReplicatedMergeTree::checkPartAndAddToZooKeeper(MergeTreeData::DataPartPtr part, zkutil::Ops & ops)
2014-04-08 17:45:21 +00:00
{
check(part->columns);
2014-07-15 09:56:17 +00:00
int expected_columns_version = columns_version;
2014-07-10 10:16:50 +00:00
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
std::random_shuffle(replicas.begin(), replicas.end());
String expected_columns_str = part->columns.toString();
for (const String & replica : replicas)
2014-04-08 17:45:21 +00:00
{
2014-07-10 10:16:50 +00:00
zkutil::Stat stat_before, stat_after;
String columns_str;
if (!zookeeper->tryGet(zookeeper_path + "/replicas/" + replica + "/parts/" + part->name + "/columns", columns_str, &stat_before))
continue;
if (columns_str != expected_columns_str)
{
LOG_INFO(log, "Not checking checksums of part " << part->name << " with replica " << replica
<< " because columns are different");
continue;
}
2014-04-14 13:08:26 +00:00
String checksums_str;
2014-07-10 10:16:50 +00:00
/// Проверим, что версия ноды со столбцами не изменилась, пока мы читали checksums.
/// Это гарантирует, что столбцы и чексуммы относятся к одним и тем же данным.
if (!zookeeper->tryGet(zookeeper_path + "/replicas/" + replica + "/parts/" + part->name + "/checksums", checksums_str) ||
!zookeeper->exists(zookeeper_path + "/replicas/" + replica + "/parts/" + part->name + "/columns", &stat_after) ||
stat_before.version != stat_after.version)
2014-04-14 13:08:26 +00:00
{
2014-07-10 10:16:50 +00:00
LOG_INFO(log, "Not checking checksums of part " << part->name << " with replica " << replica
<< " because part changed while we were reading its checksums");
continue;
2014-04-14 13:08:26 +00:00
}
2014-07-10 10:16:50 +00:00
auto checksums = MergeTreeData::DataPart::Checksums::parse(checksums_str);
checksums.checkEqual(part->checksums, true);
2014-04-08 17:45:21 +00:00
}
ops.push_back(new zkutil::Op::Check(
zookeeper_path + "/columns",
expected_columns_version));
2014-04-08 17:45:21 +00:00
ops.push_back(new zkutil::Op::Create(
replica_path + "/parts/" + part->name,
"",
2014-05-13 10:10:26 +00:00
zookeeper->getDefaultACL(),
2014-04-08 17:45:21 +00:00
zkutil::CreateMode::Persistent));
2014-07-10 10:16:50 +00:00
ops.push_back(new zkutil::Op::Create(
replica_path + "/parts/" + part->name + "/columns",
part->columns.toString(),
zookeeper->getDefaultACL(),
zkutil::CreateMode::Persistent));
2014-04-08 17:45:21 +00:00
ops.push_back(new zkutil::Op::Create(
replica_path + "/parts/" + part->name + "/checksums",
part->checksums.toString(),
2014-05-13 10:10:26 +00:00
zookeeper->getDefaultACL(),
2014-04-08 17:45:21 +00:00
zkutil::CreateMode::Persistent));
}
2014-04-09 15:52:47 +00:00
void StorageReplicatedMergeTree::clearOldParts()
{
2014-07-25 11:15:11 +00:00
MergeTreeData::DataPartsVector parts = data.grabOldParts();
size_t count = parts.size();
2014-04-09 15:52:47 +00:00
2014-07-25 11:15:11 +00:00
if (!count)
return;
try
2014-04-09 15:52:47 +00:00
{
2014-07-25 11:15:11 +00:00
while (!parts.empty())
{
MergeTreeData::DataPartPtr part = parts.back();
zkutil::Ops ops;
ops.push_back(new zkutil::Op::Remove(replica_path + "/parts/" + part->name + "/columns", -1));
ops.push_back(new zkutil::Op::Remove(replica_path + "/parts/" + part->name + "/checksums", -1));
ops.push_back(new zkutil::Op::Remove(replica_path + "/parts/" + part->name, -1));
2014-07-25 12:38:58 +00:00
auto code = zookeeper->tryMulti(ops);
if (code != ZOK)
LOG_WARNING(log, "Couldn't remove " << part->name << " from ZooKeeper: " << zkutil::ZooKeeper::error2string(code));
2014-07-25 11:15:11 +00:00
part->remove();
parts.pop_back();
}
}
catch (...)
{
data.addOldParts(parts);
throw;
2014-04-09 15:52:47 +00:00
}
2014-07-25 11:15:11 +00:00
LOG_DEBUG(log, "Removed " << count << " old parts");
2014-04-09 15:52:47 +00:00
}
2014-04-14 10:19:33 +00:00
void StorageReplicatedMergeTree::clearOldLogs()
{
2014-07-15 14:37:49 +00:00
zkutil::Stat stat;
if (!zookeeper->exists(zookeeper_path + "/log", &stat))
throw Exception(zookeeper_path + "/log doesn't exist", ErrorCodes::NOT_FOUND_NODE);
int children_count = stat.numChildren;
/// Будем ждать, пока накопятся в 1.1 раза больше записей, чем нужно.
if (static_cast<double>(children_count) < data.settings.replicated_logs_to_keep * 1.1)
return;
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas", &stat);
2014-04-14 10:19:33 +00:00
UInt64 min_pointer = std::numeric_limits<UInt64>::max();
for (const String & replica : replicas)
{
2014-07-15 14:37:49 +00:00
String pointer = zookeeper->get(zookeeper_path + "/replicas/" + replica + "/log_pointer");
if (pointer.empty())
2014-04-14 10:19:33 +00:00
return;
min_pointer = std::min(min_pointer, parse<UInt64>(pointer));
}
2014-07-15 14:37:49 +00:00
Strings entries = zookeeper->getChildren(zookeeper_path + "/log");
2014-04-14 10:19:33 +00:00
std::sort(entries.begin(), entries.end());
2014-07-25 13:11:02 +00:00
2014-07-15 14:48:50 +00:00
/// Не будем трогать последние replicated_logs_to_keep записей.
entries.erase(entries.end() - std::min(entries.size(), data.settings.replicated_logs_to_keep), entries.end());
2014-07-25 13:11:02 +00:00
/// Не будем трогать записи, не меньшие min_pointer.
entries.erase(std::lower_bound(entries.begin(), entries.end(), "log-" + padIndex(min_pointer)), entries.end());
if (entries.empty())
return;
2014-04-14 10:19:33 +00:00
2014-07-15 14:37:49 +00:00
zkutil::Ops ops;
2014-07-25 13:11:02 +00:00
for (size_t i = 0; i < entries.size(); ++i)
2014-04-14 10:19:33 +00:00
{
2014-07-25 13:11:02 +00:00
ops.push_back(new zkutil::Op::Remove(zookeeper_path + "/log/" + entries[i], -1));
if (ops.size() > 400 || i + 1 == entries.size())
2014-07-25 12:11:06 +00:00
{
/// Одновременно с очисткой лога проверим, не добавилась ли реплика с тех пор, как мы получили список реплик.
ops.push_back(new zkutil::Op::Check(zookeeper_path + "/replicas", stat.version));
zookeeper->multi(ops);
ops.clear();
}
}
2014-07-25 13:11:02 +00:00
LOG_DEBUG(log, "Removed " << entries.size() << " old log entries: " << entries.front() << " - " << entries.back());
2014-04-14 10:19:33 +00:00
}
void StorageReplicatedMergeTree::clearOldBlocks()
{
zkutil::Stat stat;
2014-05-13 10:10:26 +00:00
if (!zookeeper->exists(zookeeper_path + "/blocks", &stat))
2014-04-14 10:19:33 +00:00
throw Exception(zookeeper_path + "/blocks doesn't exist", ErrorCodes::NOT_FOUND_NODE);
int children_count = stat.numChildren;
2014-04-14 10:56:06 +00:00
2014-04-14 10:19:33 +00:00
/// Чтобы делать "асимптотически" меньше запросов exists, будем ждать, пока накопятся в 1.1 раза больше блоков, чем нужно.
2014-04-14 10:56:06 +00:00
if (static_cast<double>(children_count) < data.settings.replicated_deduplication_window * 1.1)
2014-04-14 10:19:33 +00:00
return;
2014-04-14 10:56:06 +00:00
LOG_TRACE(log, "Clearing about " << static_cast<size_t>(children_count) - data.settings.replicated_deduplication_window
2014-07-15 14:37:49 +00:00
<< " old blocks from ZooKeeper. This might take several minutes.");
2014-04-14 10:56:06 +00:00
2014-05-13 10:10:26 +00:00
Strings blocks = zookeeper->getChildren(zookeeper_path + "/blocks");
2014-04-14 10:19:33 +00:00
std::vector<std::pair<Int64, String> > timed_blocks;
for (const String & block : blocks)
{
zkutil::Stat stat;
2014-05-13 10:10:26 +00:00
zookeeper->exists(zookeeper_path + "/blocks/" + block, &stat);
timed_blocks.push_back(std::make_pair(stat.czxid, block));
2014-04-14 10:19:33 +00:00
}
2014-07-25 12:11:06 +00:00
zkutil::Ops ops;
2014-04-14 10:56:06 +00:00
std::sort(timed_blocks.begin(), timed_blocks.end(), std::greater<std::pair<Int64, String>>());
2014-04-14 10:19:33 +00:00
for (size_t i = data.settings.replicated_deduplication_window; i < timed_blocks.size(); ++i)
{
2014-04-14 10:56:06 +00:00
ops.push_back(new zkutil::Op::Remove(zookeeper_path + "/blocks/" + timed_blocks[i].second + "/number", -1));
2014-07-15 09:56:17 +00:00
ops.push_back(new zkutil::Op::Remove(zookeeper_path + "/blocks/" + timed_blocks[i].second + "/columns", -1));
2014-04-14 10:56:06 +00:00
ops.push_back(new zkutil::Op::Remove(zookeeper_path + "/blocks/" + timed_blocks[i].second + "/checksums", -1));
ops.push_back(new zkutil::Op::Remove(zookeeper_path + "/blocks/" + timed_blocks[i].second, -1));
2014-07-25 16:09:58 +00:00
if (ops.size() > 400 || i + 1 == timed_blocks.size())
2014-07-25 12:11:06 +00:00
{
zookeeper->multi(ops);
ops.clear();
}
2014-04-14 10:19:33 +00:00
}
LOG_TRACE(log, "Cleared " << blocks.size() - data.settings.replicated_deduplication_window << " old blocks from ZooKeeper");
}
2014-04-03 08:47:59 +00:00
void StorageReplicatedMergeTree::loadQueue()
{
Poco::ScopedLock<Poco::FastMutex> lock(queue_mutex);
2014-05-13 10:10:26 +00:00
Strings children = zookeeper->getChildren(replica_path + "/queue");
2014-04-03 08:47:59 +00:00
std::sort(children.begin(), children.end());
for (const String & child : children)
{
2014-05-13 10:10:26 +00:00
String s = zookeeper->get(replica_path + "/queue/" + child);
2014-04-03 11:48:28 +00:00
LogEntry entry = LogEntry::parse(s);
entry.znode_name = child;
2014-05-26 11:40:22 +00:00
entry.addResultToVirtualParts(*this);
2014-04-03 11:48:28 +00:00
queue.push_back(entry);
2014-04-03 08:47:59 +00:00
}
}
2014-07-15 14:37:49 +00:00
void StorageReplicatedMergeTree::pullLogsToQueue(zkutil::EventPtr next_update_event)
2014-04-03 08:47:59 +00:00
{
Poco::ScopedLock<Poco::FastMutex> lock(queue_mutex);
2014-03-21 19:17:59 +00:00
2014-07-15 14:37:49 +00:00
String index_str = zookeeper->get(replica_path + "/log_pointer");
UInt64 index;
2014-04-03 12:49:01 +00:00
2014-07-15 14:37:49 +00:00
if (index_str.empty())
2014-04-03 12:49:01 +00:00
{
2014-07-15 14:37:49 +00:00
/// Если у нас еще нет указателя на лог, поставим указатель на первую запись в нем.
Strings entries = zookeeper->getChildren(zookeeper_path + "/log");
index = entries.empty() ? 0 : parse<UInt64>(std::min_element(entries.begin(), entries.end())->substr(strlen("log-")));
2014-04-03 12:49:01 +00:00
2014-07-23 09:50:29 +00:00
zookeeper->set(replica_path + "/log_pointer", toString(index));
2014-07-15 14:37:49 +00:00
}
else
2014-04-03 08:47:59 +00:00
{
2014-07-15 14:37:49 +00:00
index = parse<UInt64>(index_str);
2014-04-03 12:49:01 +00:00
}
2014-04-03 08:47:59 +00:00
2014-07-25 13:28:49 +00:00
UInt64 first_index = index;
2014-04-04 12:47:57 +00:00
size_t count = 0;
2014-07-15 14:37:49 +00:00
String entry_str;
while (zookeeper->tryGet(zookeeper_path + "/log/log-" + padIndex(index), entry_str))
2014-04-03 12:49:01 +00:00
{
2014-04-04 12:47:57 +00:00
++count;
2014-07-15 14:37:49 +00:00
++index;
2014-04-03 11:48:28 +00:00
2014-07-15 14:37:49 +00:00
LogEntry entry = LogEntry::parse(entry_str);
2014-04-03 08:47:59 +00:00
2014-04-03 12:49:01 +00:00
/// Одновременно добавим запись в очередь и продвинем указатель на лог.
zkutil::Ops ops;
ops.push_back(new zkutil::Op::Create(
2014-07-15 14:37:49 +00:00
replica_path + "/queue/queue-", entry_str, zookeeper->getDefaultACL(), zkutil::CreateMode::PersistentSequential));
2014-04-03 12:49:01 +00:00
ops.push_back(new zkutil::Op::SetData(
2014-07-15 14:37:49 +00:00
replica_path + "/log_pointer", toString(index), -1));
2014-05-13 10:10:26 +00:00
auto results = zookeeper->multi(ops);
2014-04-03 12:49:01 +00:00
String path_created = dynamic_cast<zkutil::Op::Create &>(ops[0]).getPathCreated();
2014-04-04 12:47:57 +00:00
entry.znode_name = path_created.substr(path_created.find_last_of('/') + 1);
2014-05-26 11:40:22 +00:00
entry.addResultToVirtualParts(*this);
2014-04-03 12:49:01 +00:00
queue.push_back(entry);
2014-07-15 14:37:49 +00:00
}
2014-04-03 12:49:01 +00:00
2014-07-15 14:37:49 +00:00
if (next_update_event)
{
if (zookeeper->exists(zookeeper_path + "/log/log-" + padIndex(index), nullptr, next_update_event))
next_update_event->set();
2014-04-03 08:47:59 +00:00
}
2014-04-04 12:47:57 +00:00
2014-07-15 15:51:27 +00:00
if (!count)
return;
2014-07-08 14:41:15 +00:00
if (queue_task_handle)
queue_task_handle->wake();
2014-07-15 14:37:49 +00:00
2014-07-25 13:28:49 +00:00
LOG_DEBUG(log, "Pulled " << count << " entries to queue: log-" << padIndex(first_index) << " - log-" << padIndex(index - 1));
2014-04-03 08:47:59 +00:00
}
2014-04-07 15:45:46 +00:00
bool StorageReplicatedMergeTree::shouldExecuteLogEntry(const LogEntry & entry)
2014-04-03 08:47:59 +00:00
{
2014-07-15 11:12:58 +00:00
if ((entry.type == LogEntry::MERGE_PARTS || entry.type == LogEntry::GET_PART) && future_parts.count(entry.new_part_name))
2014-07-04 11:18:04 +00:00
{
LOG_DEBUG(log, "Not executing log entry for part " << entry.new_part_name <<
" because another log entry for the same part is being processed. This shouldn't happen often.");
return false;
}
2014-04-07 15:45:46 +00:00
if (entry.type == LogEntry::MERGE_PARTS)
{
/** Если какая-то из нужных частей сейчас передается или мерджится, подождем окончания этой операции.
* Иначе, даже если всех нужных частей для мерджа нет, нужно попытаться сделать мердж.
* Если каких-то частей не хватает, вместо мерджа будет попытка скачать кусок.
* Такая ситуация возможна, если получение какого-то куска пофейлилось, и его переместили в конец очереди.
*/
for (const auto & name : entry.parts_to_merge)
{
if (future_parts.count(name))
{
2014-05-07 14:12:15 +00:00
LOG_TRACE(log, "Not merging into part " << entry.new_part_name << " because part " << name << " is not ready yet.");
2014-04-07 15:45:46 +00:00
return false;
}
}
}
return true;
2014-04-03 11:48:28 +00:00
}
2014-07-18 15:41:04 +00:00
bool StorageReplicatedMergeTree::executeLogEntry(const LogEntry & entry, BackgroundProcessingPool::Context & pool_context)
2014-04-03 11:48:28 +00:00
{
if (entry.type == LogEntry::GET_PART ||
entry.type == LogEntry::MERGE_PARTS)
{
/// Если у нас уже есть этот кусок или покрывающий его кусок, ничего делать не нужно.
2014-07-25 11:38:46 +00:00
MergeTreeData::DataPartPtr containing_part = data.getActiveContainingPart(entry.new_part_name);
2014-04-03 11:48:28 +00:00
/// Даже если кусок есть локально, его (в исключительных случаях) может не быть в zookeeper.
2014-05-13 10:10:26 +00:00
if (containing_part && zookeeper->exists(replica_path + "/parts/" + containing_part->name))
2014-04-04 12:47:57 +00:00
{
2014-04-07 15:45:46 +00:00
if (!(entry.type == LogEntry::GET_PART && entry.source_replica == replica_name))
LOG_DEBUG(log, "Skipping action for part " + entry.new_part_name + " - part already exists");
2014-07-18 15:41:04 +00:00
return true;
2014-04-04 12:47:57 +00:00
}
2014-04-03 11:48:28 +00:00
}
2014-04-07 15:45:46 +00:00
if (entry.type == LogEntry::GET_PART && entry.source_replica == replica_name)
2014-05-27 10:03:13 +00:00
LOG_WARNING(log, "Part " << entry.new_part_name << " from own log doesn't exist.");
2014-04-07 15:45:46 +00:00
bool do_fetch = false;
2014-04-04 12:47:57 +00:00
if (entry.type == LogEntry::GET_PART)
{
2014-04-07 15:45:46 +00:00
do_fetch = true;
2014-04-04 12:47:57 +00:00
}
else if (entry.type == LogEntry::MERGE_PARTS)
{
2014-04-04 13:27:47 +00:00
MergeTreeData::DataPartsVector parts;
bool have_all_parts = true;
2014-04-04 13:27:47 +00:00
for (const String & name : entry.parts_to_merge)
{
2014-07-25 11:38:46 +00:00
MergeTreeData::DataPartPtr part = data.getActiveContainingPart(name);
2014-04-07 15:45:46 +00:00
if (!part)
{
have_all_parts = false;
break;
}
if (part->name != name)
{
LOG_ERROR(log, "Log and parts set look inconsistent: " << name << " is covered by " << part->name
<< " but should be merged into " << entry.new_part_name);
have_all_parts = false;
break;
}
2014-04-04 13:27:47 +00:00
parts.push_back(part);
}
2014-04-04 17:20:45 +00:00
2014-04-07 15:45:46 +00:00
if (!have_all_parts)
2014-04-04 13:27:47 +00:00
{
2014-04-07 15:45:46 +00:00
/// Если нет всех нужных кусков, попробуем взять у кого-нибудь уже помердженный кусок.
do_fetch = true;
LOG_DEBUG(log, "Don't have all parts for merge " << entry.new_part_name << "; will try to fetch it instead");
2014-04-04 13:27:47 +00:00
}
2014-04-07 15:45:46 +00:00
else
{
2014-07-02 12:30:38 +00:00
/// Если собираемся сливать большие куски, увеличим счетчик потоков, сливающих большие куски.
for (const auto & part : parts)
{
2014-07-23 15:24:45 +00:00
if (part->size_in_bytes > data.settings.max_bytes_to_merge_parts_small)
2014-07-02 12:30:38 +00:00
{
pool_context.incrementCounter("big merges");
pool_context.incrementCounter("replicated big merges");
break;
}
}
auto table_lock = lockStructure(true);
2014-07-01 15:58:25 +00:00
MergeTreeData::Transaction transaction;
MergeTreeData::DataPartPtr part = merger.mergeParts(parts, entry.new_part_name, &transaction);
2014-04-07 15:45:46 +00:00
zkutil::Ops ops;
2014-07-15 09:56:17 +00:00
checkPartAndAddToZooKeeper(part, ops);
2014-04-07 15:45:46 +00:00
2014-05-13 10:10:26 +00:00
zookeeper->multi(ops);
2014-07-01 15:58:25 +00:00
transaction.commit();
2014-07-02 10:16:49 +00:00
merge_selecting_event.set();
2014-04-07 15:45:46 +00:00
ProfileEvents::increment(ProfileEvents::ReplicatedPartMerges);
}
2014-04-04 12:47:57 +00:00
}
else
{
throw Exception("Unexpected log entry type: " + toString(static_cast<int>(entry.type)));
}
2014-04-07 15:45:46 +00:00
if (do_fetch)
{
2014-07-18 15:41:04 +00:00
String replica;
2014-04-07 15:45:46 +00:00
try
{
2014-07-18 15:41:04 +00:00
replica = findReplicaHavingPart(entry.new_part_name, true);
2014-04-08 17:45:21 +00:00
if (replica.empty())
{
ProfileEvents::increment(ProfileEvents::ReplicatedPartFailedFetches);
throw Exception("No active replica has part " + entry.new_part_name, ErrorCodes::NO_REPLICA_HAS_PART);
}
2014-04-07 15:45:46 +00:00
fetchPart(entry.new_part_name, replica);
if (entry.type == LogEntry::MERGE_PARTS)
ProfileEvents::increment(ProfileEvents::ReplicatedPartFetchesOfMerged);
}
catch (...)
{
/** Если не получилось скачать кусок, нужный для какого-то мерджа, лучше не пытаться получить другие куски для этого мерджа,
* а попытаться сразу получить помердженный кусок. Чтобы так получилось, переместим действия для получения остальных кусков
* для этого мерджа в конец очереди.
*
*/
try
{
Poco::ScopedLock<Poco::FastMutex> lock(queue_mutex);
/// Найдем действие по объединению этого куска с другими. Запомним других.
StringSet parts_for_merge;
LogEntries::iterator merge_entry;
for (LogEntries::iterator it = queue.begin(); it != queue.end(); ++it)
{
if (it->type == LogEntry::MERGE_PARTS)
{
if (std::find(it->parts_to_merge.begin(), it->parts_to_merge.end(), entry.new_part_name)
!= it->parts_to_merge.end())
{
parts_for_merge = StringSet(it->parts_to_merge.begin(), it->parts_to_merge.end());
merge_entry = it;
break;
}
}
}
if (!parts_for_merge.empty())
{
/// Переместим в конец очереди действия, получающие parts_for_merge.
for (LogEntries::iterator it = queue.begin(); it != queue.end();)
{
auto it0 = it;
++it;
if (it0 == merge_entry)
break;
if ((it0->type == LogEntry::MERGE_PARTS || it0->type == LogEntry::GET_PART)
&& parts_for_merge.count(it0->new_part_name))
{
queue.splice(queue.end(), queue, it0, it);
}
}
2014-07-18 15:41:04 +00:00
/** Если этого куска ни у кого нет, но в очереди упоминается мердж с его участием, то наверно этот кусок такой старый,
* что его все померджили и удалили. Не будем бросать исключение, чтобы queueTask лишний раз не спала.
*/
if (replica.empty())
{
LOG_INFO(log, "No replica has part " << entry.new_part_name << ". Will fetch merged part instead.");
return false;
}
2014-04-07 15:45:46 +00:00
}
2014-07-23 09:15:41 +00:00
/// Если ни у кого нет куска, и в очереди нет слияний с его участием, проверим, есть ли у кого-то покрывающий его.
if (replica.empty())
enqueuePartForCheck(entry.new_part_name);
2014-04-07 15:45:46 +00:00
}
catch (...)
{
tryLogCurrentException(__PRETTY_FUNCTION__);
}
throw;
}
}
2014-07-18 15:41:04 +00:00
return true;
2014-04-03 11:48:28 +00:00
}
void StorageReplicatedMergeTree::queueUpdatingThread()
{
while (!shutdown_called)
{
2014-04-04 10:37:33 +00:00
try
{
2014-07-15 14:37:49 +00:00
pullLogsToQueue(queue_updating_event);
2014-04-09 15:52:47 +00:00
2014-07-15 14:37:49 +00:00
queue_updating_event->wait();
2014-04-04 10:37:33 +00:00
}
catch (...)
{
tryLogCurrentException(__PRETTY_FUNCTION__);
2014-07-15 14:37:49 +00:00
queue_updating_event->tryWait(ERROR_SLEEP_MS);
}
2014-04-03 11:48:28 +00:00
}
2014-07-15 14:37:49 +00:00
LOG_DEBUG(log, "queue updating thread finished");
2014-04-03 08:47:59 +00:00
}
2014-03-21 19:17:59 +00:00
2014-07-02 12:30:38 +00:00
bool StorageReplicatedMergeTree::queueTask(BackgroundProcessingPool::Context & pool_context)
2014-04-03 11:48:28 +00:00
{
2014-07-02 12:30:38 +00:00
LogEntry entry;
bool have_work = false;
2014-04-03 11:48:28 +00:00
2014-07-02 12:30:38 +00:00
try
{
Poco::ScopedLock<Poco::FastMutex> lock(queue_mutex);
bool empty = queue.empty();
if (!empty)
2014-04-03 11:48:28 +00:00
{
2014-07-02 12:30:38 +00:00
for (LogEntries::iterator it = queue.begin(); it != queue.end(); ++it)
2014-04-03 11:48:28 +00:00
{
2014-07-02 12:30:38 +00:00
if (shouldExecuteLogEntry(*it))
2014-04-07 15:45:46 +00:00
{
2014-07-02 12:30:38 +00:00
entry = *it;
entry.tagPartAsFuture(*this);
queue.erase(it);
have_work = true;
break;
2014-04-07 15:45:46 +00:00
}
2014-04-03 11:48:28 +00:00
}
}
2014-07-02 12:30:38 +00:00
}
catch (...)
{
tryLogCurrentException(__PRETTY_FUNCTION__);
}
2014-03-21 19:17:59 +00:00
2014-07-02 12:30:38 +00:00
if (!have_work)
return false;
2014-03-21 19:17:59 +00:00
2014-07-18 15:41:04 +00:00
bool exception = true;
2014-07-02 12:30:38 +00:00
bool success = false;
2014-04-02 07:59:43 +00:00
2014-07-02 12:30:38 +00:00
try
{
2014-07-18 15:41:04 +00:00
success = executeLogEntry(entry, pool_context);
2014-04-07 15:45:46 +00:00
2014-07-18 16:05:57 +00:00
if (success)
{
auto code = zookeeper->tryRemove(replica_path + "/queue/" + entry.znode_name);
if (code != ZOK)
LOG_ERROR(log, "Couldn't remove " << replica_path + "/queue/" + entry.znode_name << ": "
<< zkutil::ZooKeeper::error2string(code) + ". There must be a bug somewhere. Ignoring it.");
}
2014-04-03 11:48:28 +00:00
2014-07-18 15:41:04 +00:00
exception = false;
2014-07-02 12:30:38 +00:00
}
catch (Exception & e)
{
if (e.code() == ErrorCodes::NO_REPLICA_HAS_PART)
2014-07-18 15:41:04 +00:00
/// Если ни у кого нет нужного куска, наверно, просто не все реплики работают; не будем писать в лог с уровнем Error.
2014-07-02 12:30:38 +00:00
LOG_INFO(log, e.displayText());
else
2014-04-04 10:37:33 +00:00
tryLogCurrentException(__PRETTY_FUNCTION__);
2014-07-02 12:30:38 +00:00
}
catch (...)
{
tryLogCurrentException(__PRETTY_FUNCTION__);
}
2014-04-03 11:48:28 +00:00
2014-07-02 12:30:38 +00:00
if (!success)
{
/// Добавим действие, которое не получилось выполнить, в конец очереди.
entry.future_part_tagger = nullptr;
Poco::ScopedLock<Poco::FastMutex> lock(queue_mutex);
queue.push_back(entry);
2014-04-03 11:48:28 +00:00
}
2014-07-02 12:30:38 +00:00
2014-07-18 15:41:04 +00:00
/// Если не было исключения, не нужно спать.
return !exception;
2014-04-03 11:48:28 +00:00
}
2014-04-04 10:37:33 +00:00
void StorageReplicatedMergeTree::mergeSelectingThread()
{
2014-07-25 14:31:50 +00:00
bool need_pull = true;
2014-04-04 10:37:33 +00:00
while (!shutdown_called && is_leader_node)
{
2014-04-07 18:14:39 +00:00
bool success = false;
2014-04-04 10:37:33 +00:00
2014-04-07 18:14:39 +00:00
try
2014-04-04 10:37:33 +00:00
{
2014-07-25 14:31:50 +00:00
if (need_pull)
{
/// Нужно загрузить новую запись в очередь перед тем, как выбирать куски для слияния.
/// (чтобы кусок добавился в virtual_parts).
pullLogsToQueue();
need_pull = false;
}
2014-04-07 18:14:39 +00:00
size_t merges_queued = 0;
2014-07-03 14:22:50 +00:00
/// Есть ли в очереди или в фоновом потоке мердж крупных кусков.
2014-07-02 12:30:38 +00:00
bool has_big_merge = context.getBackgroundPool().getCounter("replicated big merges") > 0;
2014-04-04 10:37:33 +00:00
2014-07-02 12:30:38 +00:00
if (!has_big_merge)
2014-04-07 18:14:39 +00:00
{
Poco::ScopedLock<Poco::FastMutex> lock(queue_mutex);
2014-04-04 10:37:33 +00:00
2014-04-07 18:14:39 +00:00
for (const auto & entry : queue)
2014-05-26 11:40:22 +00:00
{
2014-04-07 18:14:39 +00:00
if (entry.type == LogEntry::MERGE_PARTS)
2014-05-26 11:40:22 +00:00
{
2014-04-07 18:14:39 +00:00
++merges_queued;
2014-05-26 11:40:22 +00:00
if (!has_big_merge)
{
for (const String & name : entry.parts_to_merge)
{
2014-07-25 11:38:46 +00:00
MergeTreeData::DataPartPtr part = data.getActiveContainingPart(name);
2014-05-26 11:40:22 +00:00
if (!part || part->name != name)
continue;
2014-07-23 15:24:45 +00:00
if (part->size_in_bytes > data.settings.max_bytes_to_merge_parts_small)
2014-05-26 11:40:22 +00:00
{
has_big_merge = true;
break;
}
}
}
}
}
2014-04-07 18:14:39 +00:00
}
2014-04-04 10:37:33 +00:00
2014-07-02 10:16:49 +00:00
do
2014-04-07 18:14:39 +00:00
{
2014-07-02 12:30:38 +00:00
if (merges_queued >= data.settings.max_replicated_merges_in_queue)
2014-07-02 10:16:49 +00:00
break;
2014-04-04 10:37:33 +00:00
2014-07-02 10:16:49 +00:00
MergeTreeData::DataPartsVector parts;
2014-04-04 12:47:57 +00:00
2014-04-04 17:20:45 +00:00
String merged_name;
auto can_merge = std::bind(
&StorageReplicatedMergeTree::canMergeParts, this, std::placeholders::_1, std::placeholders::_2);
2014-07-02 10:16:49 +00:00
if (!merger.selectPartsToMerge(parts, merged_name, MergeTreeDataMerger::NO_LIMIT,
false, false, has_big_merge, can_merge) &&
!merger.selectPartsToMerge(parts, merged_name, MergeTreeDataMerger::NO_LIMIT,
2014-05-14 17:51:37 +00:00
true, false, has_big_merge, can_merge))
2014-07-02 10:16:49 +00:00
break;
2014-04-04 17:20:45 +00:00
2014-07-02 10:16:49 +00:00
LogEntry entry;
entry.type = LogEntry::MERGE_PARTS;
entry.source_replica = replica_name;
entry.new_part_name = merged_name;
2014-04-04 17:20:45 +00:00
2014-07-02 10:16:49 +00:00
for (const auto & part : parts)
{
entry.parts_to_merge.push_back(part->name);
2014-04-04 10:37:33 +00:00
}
2014-07-25 14:31:50 +00:00
need_pull = true;
2014-07-02 10:16:49 +00:00
2014-07-25 14:31:50 +00:00
zookeeper->create(zookeeper_path + "/log/log-", entry.toString(), zkutil::CreateMode::PersistentSequential);
2014-05-07 13:58:20 +00:00
String month_name = parts[0]->name.substr(0, 6);
for (size_t i = 0; i + 1 < parts.size(); ++i)
2014-04-04 17:20:45 +00:00
{
2014-05-07 13:58:20 +00:00
/// Уберем больше не нужные отметки о несуществующих блоках.
for (UInt64 number = parts[i]->right + 1; number <= parts[i + 1]->left - 1; ++number)
{
2014-07-22 13:49:52 +00:00
String path = zookeeper_path + "/block_numbers/" + month_name + "/block-" + padIndex(number);
2014-04-04 10:37:33 +00:00
2014-05-13 10:10:26 +00:00
zookeeper->tryRemove(path);
2014-05-07 13:58:20 +00:00
}
2014-04-04 17:20:45 +00:00
}
2014-07-02 10:16:49 +00:00
success = true;
2014-04-04 10:37:33 +00:00
}
2014-07-02 10:16:49 +00:00
while(false);
2014-04-04 10:37:33 +00:00
}
catch (...)
{
tryLogCurrentException(__PRETTY_FUNCTION__);
}
2014-04-14 10:56:06 +00:00
if (shutdown_called || !is_leader_node)
2014-04-04 10:37:33 +00:00
break;
2014-04-14 10:56:06 +00:00
if (!success)
2014-07-02 10:16:49 +00:00
merge_selecting_event.tryWait(MERGE_SELECTING_SLEEP_MS);
2014-04-14 10:56:06 +00:00
}
2014-07-15 14:37:49 +00:00
LOG_DEBUG(log, "merge selecting thread finished");
2014-04-14 10:56:06 +00:00
}
2014-07-15 14:37:49 +00:00
void StorageReplicatedMergeTree::cleanupThread()
2014-04-14 10:56:06 +00:00
{
2014-07-15 14:37:49 +00:00
while (!shutdown_called)
2014-04-14 10:56:06 +00:00
{
try
2014-04-14 10:19:33 +00:00
{
2014-07-15 14:37:49 +00:00
clearOldParts();
if (is_leader_node)
{
clearOldLogs();
clearOldBlocks();
}
2014-04-14 10:19:33 +00:00
}
2014-04-14 10:56:06 +00:00
catch (...)
{
tryLogCurrentException(__PRETTY_FUNCTION__);
}
2014-04-14 10:19:33 +00:00
2014-07-15 14:37:49 +00:00
shutdown_event.tryWait(CLEANUP_SLEEP_MS);
2014-04-04 10:37:33 +00:00
}
2014-07-15 14:37:49 +00:00
LOG_DEBUG(log, "cleanup thread finished");
2014-04-04 10:37:33 +00:00
}
2014-07-15 15:51:27 +00:00
void StorageReplicatedMergeTree::alterThread()
{
bool force_recheck_parts = true;
while (!shutdown_called)
{
try
{
zkutil::Stat stat;
String columns_str = zookeeper->get(zookeeper_path + "/columns", &stat, alter_thread_event);
NamesAndTypesList columns = NamesAndTypesList::parse(columns_str, context.getDataTypeFactory());
bool changed = false;
/// Проверим, что описание столбцов изменилось.
/// Чтобы не останавливать лишний раз все запросы в таблицу, проверим сначала под локом на чтение.
{
auto table_lock = lockStructure(false);
if (columns != data.getColumnsList())
changed = true;
}
/// Если описание столбцов изменилось, обновим структуру таблицы локально.
if (changed)
{
auto table_lock = lockStructureForAlter();
if (columns != data.getColumnsList())
{
LOG_INFO(log, "Columns list changed in ZooKeeper. Applying changes locally.");
InterpreterAlterQuery::updateMetadata(database_name, table_name, columns, context);
data.setColumnsList(columns);
2014-07-16 09:32:15 +00:00
if (unreplicated_data)
unreplicated_data->setColumnsList(columns);
2014-07-15 15:51:27 +00:00
columns_version = stat.version;
LOG_INFO(log, "Applied changes to table.");
}
else
{
changed = false;
}
}
/// Обновим куски.
if (changed || force_recheck_parts)
{
if (changed)
LOG_INFO(log, "ALTER-ing parts");
int changed_parts = 0;
2014-07-16 09:32:15 +00:00
auto parts = data.getDataParts();
2014-07-15 15:51:27 +00:00
for (const MergeTreeData::DataPartPtr & part : parts)
{
/// Обновим кусок и запишем результат во временные файлы.
2014-07-23 12:44:23 +00:00
/// TODO: Можно пропускать проверку на слишком большие изменения, если в ZooKeeper есть, например,
/// нода /flags/force_alter.
2014-07-15 15:51:27 +00:00
auto transaction = data.alterDataPart(part, columns);
if (!transaction)
continue;
++changed_parts;
/// Обновим метаданные куска в ZooKeeper.
zkutil::Ops ops;
ops.push_back(new zkutil::Op::SetData(replica_path + "/parts/" + part->name + "/columns", part->columns.toString(), -1));
ops.push_back(new zkutil::Op::SetData(replica_path + "/parts/" + part->name + "/checksums", part->checksums.toString(), -1));
zookeeper->multi(ops);
/// Применим изменения файлов.
transaction->commit();
}
2014-07-16 09:32:15 +00:00
/// То же самое для нереплицируемых данных.
if (unreplicated_data)
{
parts = unreplicated_data->getDataParts();
for (const MergeTreeData::DataPartPtr & part : parts)
{
auto transaction = unreplicated_data->alterDataPart(part, columns);
if (!transaction)
continue;
++changed_parts;
transaction->commit();
}
}
2014-07-16 08:58:59 +00:00
zookeeper->set(replica_path + "/columns", columns.toString());
2014-07-15 15:51:27 +00:00
if (changed || changed_parts != 0)
LOG_INFO(log, "ALTER-ed " << changed_parts << " parts");
force_recheck_parts = false;
}
alter_thread_event->wait();
}
catch (...)
{
tryLogCurrentException(__PRETTY_FUNCTION__);
force_recheck_parts = true;
alter_thread_event->tryWait(ERROR_SLEEP_MS);
}
}
LOG_DEBUG(log, "alter thread finished");
}
2014-07-22 13:49:52 +00:00
void StorageReplicatedMergeTree::removePartAndEnqueueFetch(const String & part_name)
{
String part_path = replica_path + "/parts/" + part_name;
LogEntry log_entry;
log_entry.type = LogEntry::GET_PART;
log_entry.source_replica = "";
log_entry.new_part_name = part_name;
zkutil::Ops ops;
ops.push_back(new zkutil::Op::Create(
replica_path + "/queue/queue-", log_entry.toString(), zookeeper->getDefaultACL(),
zkutil::CreateMode::PersistentSequential));
ops.push_back(new zkutil::Op::Remove(part_path + "/checksums", -1));
ops.push_back(new zkutil::Op::Remove(part_path + "/columns", -1));
ops.push_back(new zkutil::Op::Remove(part_path, -1));
auto results = zookeeper->multi(ops);
{
Poco::ScopedLock<Poco::FastMutex> lock(queue_mutex);
String path_created = dynamic_cast<zkutil::Op::Create &>(ops[0]).getPathCreated();
log_entry.znode_name = path_created.substr(path_created.find_last_of('/') + 1);
log_entry.addResultToVirtualParts(*this);
queue.push_back(log_entry);
}
}
void StorageReplicatedMergeTree::enqueuePartForCheck(const String & name)
{
Poco::ScopedLock<Poco::FastMutex> lock(parts_to_check_mutex);
if (parts_to_check_set.count(name))
return;
parts_to_check_queue.push_back(name);
parts_to_check_set.insert(name);
parts_to_check_event.set();
}
2014-07-21 10:05:56 +00:00
void StorageReplicatedMergeTree::partCheckThread()
{
while (!shutdown_called)
{
try
{
/// Достанем из очереди кусок для проверки.
String part_name;
{
Poco::ScopedLock<Poco::FastMutex> lock(parts_to_check_mutex);
if (parts_to_check_queue.empty())
{
if (!parts_to_check_set.empty())
{
LOG_ERROR(log, "Non-empty parts_to_check_set with empty parts_to_check_queue. This is a bug.");
parts_to_check_set.clear();
}
}
else
{
part_name = parts_to_check_queue.front();
}
}
if (part_name.empty())
{
parts_to_check_event.wait();
continue;
}
LOG_WARNING(log, "Checking part " << part_name);
2014-07-23 09:15:41 +00:00
ProfileEvents::increment(ProfileEvents::ReplicatedPartChecks);
2014-07-21 10:05:56 +00:00
2014-07-25 11:38:46 +00:00
auto part = data.getActiveContainingPart(part_name);
2014-07-21 10:05:56 +00:00
String part_path = replica_path + "/parts/" + part_name;
/// Этого или покрывающего куска у нас нет.
if (!part)
{
/// Если кусок есть в ZooKeeper, удалим его оттуда и добавим в очередь задание скачать его.
if (zookeeper->exists(part_path))
{
LOG_WARNING(log, "Part " << part_name << " exists in ZooKeeper but not locally. "
"Removing from ZooKeeper and queueing a fetch.");
2014-07-23 09:15:41 +00:00
ProfileEvents::increment(ProfileEvents::ReplicatedPartChecksFailed);
2014-07-21 10:05:56 +00:00
2014-07-22 13:49:52 +00:00
removePartAndEnqueueFetch(part_name);
2014-07-21 10:05:56 +00:00
}
/// Если куска нет в ZooKeeper, проверим есть ли он хоть у кого-то.
else
{
2014-07-23 11:02:22 +00:00
ActiveDataPartSet::Part part_info;
ActiveDataPartSet::parsePartName(part_name, part_info);
2014-07-21 10:05:56 +00:00
2014-07-23 11:02:22 +00:00
/** Будем проверять только куски, не полученные в результате слияния.
* Для кусков, полученных в результате слияния такая проверка была бы некорректной,
* потому что слитого куска может еще ни у кого не быть.
*/
2014-07-25 16:09:58 +00:00
if (part_info.left == part_info.right)
2014-07-22 13:49:52 +00:00
{
2014-07-23 11:02:22 +00:00
LOG_WARNING(log, "Checking if anyone has part covering " << part_name << ".");
bool found = false;
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
for (const String & replica : replicas)
2014-07-22 13:49:52 +00:00
{
2014-07-23 11:02:22 +00:00
Strings parts = zookeeper->getChildren(zookeeper_path + "/replicas/" + replica + "/parts");
for (const String & part_on_replica : parts)
2014-07-22 13:49:52 +00:00
{
2014-07-23 11:02:22 +00:00
if (part_on_replica == part_name || ActiveDataPartSet::contains(part_on_replica, part_name))
{
found = true;
LOG_WARNING(log, "Found part " << part_on_replica << " on " << replica);
break;
}
2014-07-22 13:49:52 +00:00
}
2014-07-23 11:02:22 +00:00
if (found)
break;
2014-07-22 13:49:52 +00:00
}
2014-07-23 11:02:22 +00:00
if (!found)
2014-07-22 13:49:52 +00:00
{
2014-07-25 16:09:58 +00:00
LOG_ERROR(log, "No replica has part covering " << part_name);
2014-07-23 11:02:22 +00:00
ProfileEvents::increment(ProfileEvents::ReplicatedPartChecksFailed);
/** Если ни у кого нет такого куска, удалим его из нашей очереди.
*
* Еще можно было бы добавить его в block_numbers, чтобы он не мешал слияниям,
* но если так сделать, ZooKeeper почему-то пропустит один номер для автоинкремента,
2014-07-23 12:44:23 +00:00
* и в номерах блоков все равно останется дырка.
* TODO: можно это исправить, сделав две директории block_numbers: для автоинкрементных и ручных нод.
2014-07-22 13:49:52 +00:00
*/
2014-07-23 11:02:22 +00:00
2014-07-25 16:09:58 +00:00
bool was_in_queue = false;
2014-07-22 13:49:52 +00:00
{
2014-07-23 11:02:22 +00:00
Poco::ScopedLock<Poco::FastMutex> lock(queue_mutex);
/** NOTE: Не удалятся записи в очереди, которые сейчас выполняются.
* Они пофейлятся и положат кусок снова в очередь на проверку.
* Расчитываем, что это редкая ситуация.
*/
for (LogEntries::iterator it = queue.begin(); it != queue.end(); )
2014-07-22 13:49:52 +00:00
{
2014-07-23 11:02:22 +00:00
if (it->new_part_name == part_name)
{
zookeeper->remove(replica_path + "/queue/" + it->znode_name);
queue.erase(it++);
2014-07-25 16:09:58 +00:00
was_in_queue = true;
2014-07-23 11:02:22 +00:00
}
else
{
++it;
}
2014-07-22 13:49:52 +00:00
}
}
2014-07-25 16:09:58 +00:00
if (was_in_queue)
/** Такая ситуация возможна, если на всех репликах, где был кусок, он испортился.
* Например, у реплики, которая только что его записала, отключили питание, и данные не записались из кеша на диск.
*/
LOG_ERROR(log, "Part " << part_name << " is lost forever. Say goodbye to a piece of data!");
2014-07-22 13:49:52 +00:00
}
}
2014-07-21 10:05:56 +00:00
}
}
/// У нас есть этот кусок, и он активен.
else if (part->name == part_name)
{
/// Если кусок есть в ZooKeeper, сверим его данные с его чексуммами, а их с ZooKeeper.
if (zookeeper->exists(replica_path + "/parts/" + part_name))
{
2014-07-22 13:49:52 +00:00
LOG_WARNING(log, "Checking data of part " << part_name << ".");
try
{
auto zk_checksums = MergeTreeData::DataPart::Checksums::parse(
zookeeper->get(replica_path + "/parts/" + part_name + "/checksums"));
zk_checksums.checkEqual(part->checksums, true);
auto zk_columns = NamesAndTypesList::parse(
zookeeper->get(replica_path + "/parts/" + part_name + "/columns"), context.getDataTypeFactory());
if (part->columns != zk_columns)
throw Exception("Columns of local part " + part_name + " are different from ZooKeeper");
MergeTreePartChecker::checkDataPart(
data.getFullPath() + part_name, data.index_granularity, true, context.getDataTypeFactory());
2014-07-21 10:05:56 +00:00
2014-07-22 13:49:52 +00:00
LOG_INFO(log, "Part " << part_name << " looks good.");
}
catch (...)
{
tryLogCurrentException(__PRETTY_FUNCTION__);
2014-07-23 09:15:41 +00:00
LOG_ERROR(log, "Part " << part_name << " looks broken. Removing it and queueing a fetch.");
ProfileEvents::increment(ProfileEvents::ReplicatedPartChecksFailed);
2014-07-22 13:49:52 +00:00
removePartAndEnqueueFetch(part_name);
/// Удалим кусок локально.
2014-07-23 16:31:34 +00:00
data.renameAndDetachPart(part, "broken_");
2014-07-22 13:49:52 +00:00
}
2014-07-21 10:05:56 +00:00
}
/// Если куска нет в ZooKeeper, удалим его локально.
else
{
2014-07-23 09:15:41 +00:00
ProfileEvents::increment(ProfileEvents::ReplicatedPartChecksFailed);
2014-07-25 11:15:11 +00:00
LOG_ERROR(log, "Unexpected part " << part_name << ". Removing.");
data.renameAndDetachPart(part, "unexpected_");
2014-07-21 10:05:56 +00:00
}
}
else
{
/// Если у нас есть покрывающий кусок, игнорируем все проблемы с этим куском.
/// В худшем случае в лог еще old_parts_lifetime секунд будут валиться ошибки, пока кусок не удалится как старый.
}
/// Удалим кусок из очереди.
{
Poco::ScopedLock<Poco::FastMutex> lock(parts_to_check_mutex);
if (parts_to_check_queue.empty() || parts_to_check_queue.front() != part_name)
{
LOG_ERROR(log, "Someone changed parts_to_check_queue.front(). This is a bug.");
}
else
{
parts_to_check_queue.pop_front();
parts_to_check_set.erase(part_name);
}
}
}
catch (...)
{
tryLogCurrentException(__PRETTY_FUNCTION__);
parts_to_check_event.tryWait(ERROR_SLEEP_MS);
}
}
}
2014-04-04 10:37:33 +00:00
bool StorageReplicatedMergeTree::canMergeParts(const MergeTreeData::DataPartPtr & left, const MergeTreeData::DataPartPtr & right)
{
2014-05-26 11:40:22 +00:00
/// Если какой-то из кусков уже собираются слить в больший, не соглашаемся его сливать.
if (virtual_parts.getContainingPart(left->name) != left->name ||
virtual_parts.getContainingPart(right->name) != right->name)
2014-04-04 10:37:33 +00:00
return false;
2014-06-30 13:57:20 +00:00
/// Если о каком-то из кусков нет информации в ZK, не будем сливать.
if (!zookeeper->exists(replica_path + "/parts/" + left->name) ||
!zookeeper->exists(replica_path + "/parts/" + right->name))
return false;
2014-05-07 13:58:20 +00:00
String month_name = left->name.substr(0, 6);
2014-04-04 10:37:33 +00:00
/// Можно слить куски, если все номера между ними заброшены - не соответствуют никаким блокам.
for (UInt64 number = left->right + 1; number <= right->left - 1; ++number)
{
2014-07-22 13:49:52 +00:00
String path = zookeeper_path + "/block_numbers/" + month_name + "/block-" + padIndex(number);
2014-04-04 10:37:33 +00:00
2014-05-13 10:10:26 +00:00
if (AbandonableLockInZooKeeper::check(path, *zookeeper) != AbandonableLockInZooKeeper::ABANDONED)
2014-04-04 10:37:33 +00:00
return false;
}
return true;
}
void StorageReplicatedMergeTree::becomeLeader()
{
2014-04-04 12:47:57 +00:00
LOG_INFO(log, "Became leader");
2014-04-04 10:37:33 +00:00
is_leader_node = true;
merge_selecting_thread = std::thread(&StorageReplicatedMergeTree::mergeSelectingThread, this);
}
2014-04-08 17:45:21 +00:00
String StorageReplicatedMergeTree::findReplicaHavingPart(const String & part_name, bool active)
2014-04-03 11:48:28 +00:00
{
2014-05-13 10:10:26 +00:00
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
2014-04-03 11:48:28 +00:00
/// Из реплик, у которых есть кусок, выберем одну равновероятно.
std::random_shuffle(replicas.begin(), replicas.end());
for (const String & replica : replicas)
{
2014-05-13 10:10:26 +00:00
if (zookeeper->exists(zookeeper_path + "/replicas/" + replica + "/parts/" + part_name) &&
(!active || zookeeper->exists(zookeeper_path + "/replicas/" + replica + "/is_active")))
2014-04-03 11:48:28 +00:00
return replica;
}
2014-04-08 17:45:21 +00:00
return "";
2014-04-03 11:48:28 +00:00
}
void StorageReplicatedMergeTree::fetchPart(const String & part_name, const String & replica_name)
{
2014-04-04 12:47:57 +00:00
LOG_DEBUG(log, "Fetching part " << part_name << " from " << replica_name);
2014-04-03 11:48:28 +00:00
auto table_lock = lockStructure(true);
String host;
int port;
2014-05-13 10:10:26 +00:00
String host_port_str = zookeeper->get(zookeeper_path + "/replicas/" + replica_name + "/host");
2014-04-03 11:48:28 +00:00
ReadBufferFromString buf(host_port_str);
assertString("host: ", buf);
readString(host, buf);
assertString("\nport: ", buf);
readText(port, buf);
assertString("\n", buf);
assertEOF(buf);
2014-04-03 12:49:01 +00:00
MergeTreeData::MutableDataPartPtr part = fetcher.fetchPart(part_name, zookeeper_path + "/replicas/" + replica_name, host, port);
2014-07-01 15:58:25 +00:00
MergeTreeData::Transaction transaction;
auto removed_parts = data.renameTempPartAndReplace(part, nullptr, &transaction);
2014-04-07 15:45:46 +00:00
2014-04-03 11:48:28 +00:00
zkutil::Ops ops;
2014-07-15 09:56:17 +00:00
checkPartAndAddToZooKeeper(part, ops);
2014-04-07 18:14:39 +00:00
2014-05-13 10:10:26 +00:00
zookeeper->multi(ops);
2014-07-01 15:58:25 +00:00
transaction.commit();
2014-07-02 10:16:49 +00:00
merge_selecting_event.set();
2014-04-09 15:52:47 +00:00
2014-04-07 18:14:39 +00:00
for (const auto & removed_part : removed_parts)
{
LOG_DEBUG(log, "Part " << removed_part->name << " is rendered obsolete by fetching part " << part_name);
ProfileEvents::increment(ProfileEvents::ObsoleteReplicatedParts);
}
2014-04-07 15:45:46 +00:00
ProfileEvents::increment(ProfileEvents::ReplicatedPartFetches);
2014-06-30 13:57:20 +00:00
LOG_DEBUG(log, "Fetched part " << part_name << " from " << replica_name);
2014-04-03 11:48:28 +00:00
}
2014-03-21 19:17:59 +00:00
2014-03-22 14:44:44 +00:00
void StorageReplicatedMergeTree::shutdown()
{
2014-04-25 13:55:15 +00:00
if (permanent_shutdown_called)
2014-05-26 11:40:22 +00:00
{
if (restarting_thread.joinable())
restarting_thread.join();
2014-03-22 14:44:44 +00:00
return;
2014-05-26 11:40:22 +00:00
}
2014-07-15 14:37:49 +00:00
2014-04-25 13:55:15 +00:00
permanent_shutdown_called = true;
2014-07-04 13:59:05 +00:00
permanent_shutdown_event.set();
2014-04-25 13:55:15 +00:00
restarting_thread.join();
2014-07-15 14:37:49 +00:00
endpoint_holder = nullptr;
2014-04-25 13:55:15 +00:00
}
void StorageReplicatedMergeTree::partialShutdown()
{
2014-04-04 10:37:33 +00:00
leader_election = nullptr;
2014-03-22 14:44:44 +00:00
shutdown_called = true;
2014-07-04 13:59:05 +00:00
shutdown_event.set();
2014-07-15 14:37:49 +00:00
merge_selecting_event.set();
queue_updating_event->set();
alter_thread_event->set();
2014-07-16 08:58:59 +00:00
alter_query_event->set();
2014-07-22 13:49:52 +00:00
parts_to_check_event.set();
2014-03-22 14:44:44 +00:00
replica_is_active_node = nullptr;
2014-05-08 08:03:03 +00:00
merger.cancelAll();
if (unreplicated_merger)
unreplicated_merger->cancelAll();
2014-04-03 11:48:28 +00:00
LOG_TRACE(log, "Waiting for threads to finish");
2014-04-04 10:37:33 +00:00
if (is_leader_node)
2014-04-14 10:56:06 +00:00
{
2014-04-22 10:57:19 +00:00
is_leader_node = false;
2014-05-13 11:24:04 +00:00
if (merge_selecting_thread.joinable())
merge_selecting_thread.join();
2014-04-14 10:56:06 +00:00
}
2014-05-13 11:24:04 +00:00
if (queue_updating_thread.joinable())
queue_updating_thread.join();
2014-07-15 14:37:49 +00:00
if (cleanup_thread.joinable())
cleanup_thread.join();
if (alter_thread.joinable())
alter_thread.join();
2014-07-22 13:49:52 +00:00
if (part_check_thread.joinable())
part_check_thread.join();
2014-07-08 14:41:15 +00:00
if (queue_task_handle)
context.getBackgroundPool().removeTask(queue_task_handle);
2014-07-02 12:30:38 +00:00
queue_task_handle.reset();
2014-04-03 11:48:28 +00:00
LOG_TRACE(log, "Threads finished");
2014-03-22 14:44:44 +00:00
}
2014-05-13 11:24:04 +00:00
void StorageReplicatedMergeTree::goReadOnly()
{
LOG_INFO(log, "Going to read-only mode");
is_read_only = true;
permanent_shutdown_called = true;
2014-07-04 13:59:05 +00:00
permanent_shutdown_event.set();
2014-05-13 11:24:04 +00:00
2014-07-15 14:37:49 +00:00
partialShutdown();
2014-05-13 11:24:04 +00:00
}
2014-04-22 10:57:19 +00:00
void StorageReplicatedMergeTree::startup()
{
shutdown_called = false;
2014-07-04 13:59:05 +00:00
shutdown_event.reset();
2014-04-22 10:57:19 +00:00
2014-05-13 10:10:26 +00:00
merger.uncancelAll();
if (unreplicated_merger)
unreplicated_merger->uncancelAll();
2014-04-22 10:57:19 +00:00
activateReplica();
2014-05-13 10:10:26 +00:00
leader_election = new zkutil::LeaderElection(zookeeper_path + "/leader_election", *zookeeper,
2014-04-22 10:57:19 +00:00
std::bind(&StorageReplicatedMergeTree::becomeLeader, this), replica_name);
queue_updating_thread = std::thread(&StorageReplicatedMergeTree::queueUpdatingThread, this);
2014-07-15 14:37:49 +00:00
cleanup_thread = std::thread(&StorageReplicatedMergeTree::cleanupThread, this);
2014-07-15 15:51:27 +00:00
alter_thread = std::thread(&StorageReplicatedMergeTree::alterThread, this);
2014-07-22 13:49:52 +00:00
part_check_thread = std::thread(&StorageReplicatedMergeTree::partCheckThread, this);
2014-07-08 14:41:15 +00:00
queue_task_handle = context.getBackgroundPool().addTask(
std::bind(&StorageReplicatedMergeTree::queueTask, this, std::placeholders::_1));
2014-04-22 10:57:19 +00:00
}
2014-04-25 13:55:15 +00:00
void StorageReplicatedMergeTree::restartingThread()
{
2014-05-13 10:10:26 +00:00
try
2014-04-25 13:55:15 +00:00
{
2014-05-13 10:10:26 +00:00
startup();
while (!permanent_shutdown_called)
2014-04-25 13:55:15 +00:00
{
2014-05-13 10:10:26 +00:00
if (zookeeper->expired())
{
LOG_WARNING(log, "ZooKeeper session has expired. Switching to a new session.");
2014-04-25 13:55:15 +00:00
2014-05-13 10:10:26 +00:00
/// Запретим писать в таблицу, пока подменяем zookeeper.
2014-07-15 11:12:58 +00:00
LOG_TRACE(log, "Locking INSERTs");
2014-05-13 10:10:26 +00:00
auto structure_lock = lockDataForAlter();
2014-07-15 11:12:58 +00:00
LOG_TRACE(log, "Locked INSERTs");
2014-04-25 13:55:15 +00:00
2014-05-13 10:10:26 +00:00
partialShutdown();
2014-04-25 13:55:15 +00:00
2014-05-13 10:10:26 +00:00
zookeeper = context.getZooKeeper();
startup();
}
2014-04-25 13:55:15 +00:00
2014-07-04 13:59:05 +00:00
permanent_shutdown_event.tryWait(60 * 1000);
2014-05-13 10:10:26 +00:00
}
}
catch (...)
{
tryLogCurrentException("StorageReplicatedMergeTree::restartingThread");
2014-05-13 11:24:04 +00:00
LOG_ERROR(log, "Exception in restartingThread. The storage will be read-only until server restart.");
goReadOnly();
2014-07-15 14:37:49 +00:00
LOG_DEBUG(log, "restarting thread finished");
2014-05-13 10:10:26 +00:00
return;
2014-04-25 13:55:15 +00:00
}
2014-05-13 10:10:26 +00:00
try
{
endpoint_holder = nullptr;
partialShutdown();
}
catch (...)
{
tryLogCurrentException("StorageReplicatedMergeTree::restartingThread");
}
2014-07-15 14:37:49 +00:00
LOG_DEBUG(log, "restarting thread finished");
2014-04-25 13:55:15 +00:00
}
2014-03-22 14:44:44 +00:00
StorageReplicatedMergeTree::~StorageReplicatedMergeTree()
{
try
{
shutdown();
}
catch(...)
{
tryLogCurrentException("~StorageReplicatedMergeTree");
}
}
BlockInputStreams StorageReplicatedMergeTree::read(
const Names & column_names,
ASTPtr query,
const Settings & settings,
QueryProcessingStage::Enum & processed_stage,
size_t max_block_size,
unsigned threads)
{
2014-04-24 10:20:02 +00:00
BlockInputStreams res = reader.read(column_names, query, settings, processed_stage, max_block_size, threads);
if (unreplicated_reader)
{
BlockInputStreams res2 = unreplicated_reader->read(column_names, query, settings, processed_stage, max_block_size, threads);
res.insert(res.begin(), res2.begin(), res2.end());
}
return res;
2014-03-22 14:44:44 +00:00
}
2014-04-02 10:10:37 +00:00
BlockOutputStreamPtr StorageReplicatedMergeTree::write(ASTPtr query)
{
2014-05-13 11:24:04 +00:00
if (is_read_only)
throw Exception("Table is in read only mode", ErrorCodes::TABLE_IS_READ_ONLY);
2014-04-02 10:10:37 +00:00
String insert_id;
if (ASTInsertQuery * insert = typeid_cast<ASTInsertQuery *>(&*query))
2014-04-02 10:10:37 +00:00
insert_id = insert->insert_id;
return new ReplicatedMergeTreeBlockOutputStream(*this, insert_id);
}
2014-03-22 14:44:44 +00:00
2014-05-08 08:03:03 +00:00
bool StorageReplicatedMergeTree::optimize()
{
/// Померджим какие-нибудь куски из директории unreplicated.
/// TODO: Мерджить реплицируемые куски тоже.
/// TODO: Не давать вызывать это из нескольких потоков сразу: один кусок может принять участие в нескольких несовместимых слияниях.
2014-05-08 08:03:03 +00:00
if (!unreplicated_data)
return false;
unreplicated_data->clearOldParts();
MergeTreeData::DataPartsVector parts;
String merged_name;
auto always_can_merge = [](const MergeTreeData::DataPartPtr &a, const MergeTreeData::DataPartPtr &b) { return true; };
if (!unreplicated_merger->selectPartsToMerge(parts, merged_name, 0, true, true, false, always_can_merge))
return false;
unreplicated_merger->mergeParts(parts, merged_name);
return true;
}
2014-07-16 08:58:59 +00:00
void StorageReplicatedMergeTree::alter(const AlterCommands & params, const String & database_name, const String & table_name, Context & context)
{
LOG_DEBUG(log, "Doing ALTER");
NamesAndTypesList new_columns;
String new_columns_str;
int new_columns_version;
zkutil::Stat stat;
{
auto table_lock = lockStructureForAlter();
data.checkAlter(params);
new_columns = data.getColumnsList();
params.apply(new_columns);
new_columns_str = new_columns.toString();
/// Делаем ALTER.
zookeeper->set(zookeeper_path + "/columns", new_columns_str, -1, &stat);
new_columns_version = stat.version;
}
LOG_DEBUG(log, "Updated columns in ZooKeeper. Waiting for replicas to apply changes.");
/// Ждем, пока все реплики обновят данные.
/// Подпишемся на изменения столбцов, чтобы перестать ждать, если кто-то еще сделает ALTER.
if (!zookeeper->exists(zookeeper_path + "/columns", &stat, alter_query_event))
throw Exception(zookeeper_path + "/columns doesn't exist", ErrorCodes::NOT_FOUND_NODE);
if (stat.version != new_columns_version)
{
LOG_WARNING(log, zookeeper_path + "/columns changed before this ALTER finished; "
"overlapping ALTER-s are fine but use caution with nontransitive changes");
return;
}
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
for (const String & replica : replicas)
{
LOG_DEBUG(log, "Waiting for " << replica << " to apply changes");
while (!shutdown_called)
{
String replica_columns_str;
/// Реплику могли успеть удалить.
if (!zookeeper->tryGet(zookeeper_path + "/replicas/" + replica + "/columns", replica_columns_str, &stat))
{
LOG_WARNING(log, replica << " was removed");
break;
}
int replica_columns_version = stat.version;
if (replica_columns_str == new_columns_str)
break;
if (!zookeeper->exists(zookeeper_path + "/columns", &stat))
throw Exception(zookeeper_path + "/columns doesn't exist", ErrorCodes::NOT_FOUND_NODE);
if (stat.version != new_columns_version)
{
LOG_WARNING(log, zookeeper_path + "/columns changed before ALTER finished; "
"overlapping ALTER-s are fine but use caution with nontransitive changes");
return;
}
2014-07-17 10:44:17 +00:00
if (!zookeeper->exists(zookeeper_path + "/replicas/" + replica + "/columns", &stat, alter_query_event))
2014-07-16 08:58:59 +00:00
{
LOG_WARNING(log, replica << " was removed");
break;
}
if (stat.version != replica_columns_version)
continue;
alter_query_event->wait();
}
if (shutdown_called)
break;
}
LOG_DEBUG(log, "ALTER finished");
}
2014-03-22 14:44:44 +00:00
void StorageReplicatedMergeTree::drop()
{
2014-07-23 13:58:38 +00:00
if (!zookeeper)
throw Exception("Can't drop replicated table without ZooKeeper", ErrorCodes::NO_ZOOKEEPER);
2014-04-03 11:48:28 +00:00
shutdown();
2014-05-27 12:08:40 +00:00
LOG_INFO(log, "Removing replica " << replica_path);
2014-03-22 14:44:44 +00:00
replica_is_active_node = nullptr;
2014-07-07 09:51:42 +00:00
zookeeper->tryRemoveRecursive(replica_path);
2014-07-07 11:39:06 +00:00
/// Проверяем, что zookeeper_path существует: его могла удалить другая реплика после выполнения предыдущей строки.
2014-07-07 09:51:42 +00:00
Strings replicas;
if (zookeeper->tryGetChildren(zookeeper_path + "/replicas", replicas) == ZOK && replicas.empty())
2014-05-27 12:08:40 +00:00
{
LOG_INFO(log, "Removing table " << zookeeper_path << " (this might take several minutes)");
2014-07-07 09:51:42 +00:00
zookeeper->tryRemoveRecursive(zookeeper_path);
2014-05-27 12:08:40 +00:00
}
2014-07-02 12:30:38 +00:00
data.dropAllData();
2014-03-22 14:44:44 +00:00
}
2014-04-02 10:10:37 +00:00
void StorageReplicatedMergeTree::LogEntry::writeText(WriteBuffer & out) const
{
writeString("format version: 1\n", out);
2014-04-07 15:45:46 +00:00
writeString("source replica: ", out);
writeString(source_replica, out);
writeString("\n", out);
2014-04-02 10:10:37 +00:00
switch (type)
{
case GET_PART:
writeString("get\n", out);
writeString(new_part_name, out);
break;
case MERGE_PARTS:
writeString("merge\n", out);
for (const String & s : parts_to_merge)
{
writeString(s, out);
writeString("\n", out);
}
writeString("into\n", out);
writeString(new_part_name, out);
break;
}
writeString("\n", out);
}
void StorageReplicatedMergeTree::LogEntry::readText(ReadBuffer & in)
{
String type_str;
assertString("format version: 1\n", in);
2014-04-07 17:39:45 +00:00
assertString("source replica: ", in);
2014-04-07 15:45:46 +00:00
readString(source_replica, in);
2014-04-02 10:10:37 +00:00
assertString("\n", in);
2014-04-07 17:39:45 +00:00
readString(type_str, in);
assertString("\n", in);
2014-04-02 10:10:37 +00:00
if (type_str == "get")
{
type = GET_PART;
readString(new_part_name, in);
}
else if (type_str == "merge")
{
type = MERGE_PARTS;
while (true)
{
String s;
readString(s, in);
assertString("\n", in);
if (s == "into")
break;
parts_to_merge.push_back(s);
}
readString(new_part_name, in);
}
assertString("\n", in);
}
2014-03-21 13:42:14 +00:00
}