mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 11:22:12 +00:00
3750 lines
145 KiB
C++
3750 lines
145 KiB
C++
#include <Common/ZooKeeper/Types.h>
|
|
#include <Common/ZooKeeper/KeeperException.h>
|
|
|
|
#include <Common/FieldVisitors.h>
|
|
|
|
#include <Storages/ColumnsDescription.h>
|
|
#include <Storages/StorageReplicatedMergeTree.h>
|
|
#include <Storages/MergeTree/MergeTreeDataPart.h>
|
|
#include <Storages/MergeTree/ReplicatedMergeTreeBlockOutputStream.h>
|
|
#include <Storages/MergeTree/ReplicatedMergeTreeQuorumEntry.h>
|
|
#include <Storages/MergeTree/MergeList.h>
|
|
#include <Storages/MergeTree/ReplicatedMergeTreeAddress.h>
|
|
|
|
#include <Databases/IDatabase.h>
|
|
|
|
#include <Parsers/formatAST.h>
|
|
#include <Parsers/ASTOptimizeQuery.h>
|
|
#include <Parsers/ASTLiteral.h>
|
|
|
|
#include <IO/ReadBufferFromString.h>
|
|
#include <IO/Operators.h>
|
|
#include <IO/ConnectionTimeouts.h>
|
|
|
|
#include <Interpreters/InterpreterAlterQuery.h>
|
|
#include <Interpreters/PartLog.h>
|
|
|
|
#include <DataStreams/RemoteBlockInputStream.h>
|
|
#include <DataStreams/NullBlockOutputStream.h>
|
|
#include <DataStreams/copyData.h>
|
|
|
|
#include <Common/Macros.h>
|
|
#include <Storages/VirtualColumnUtils.h>
|
|
#include <Common/formatReadable.h>
|
|
#include <Common/setThreadName.h>
|
|
#include <Common/escapeForFileName.h>
|
|
#include <Common/StringUtils/StringUtils.h>
|
|
#include <Common/typeid_cast.h>
|
|
|
|
#include <Poco/DirectoryIterator.h>
|
|
|
|
#include <common/ThreadPool.h>
|
|
|
|
#include <ext/range.h>
|
|
#include <ext/scope_guard.h>
|
|
|
|
#include <cfenv>
|
|
#include <ctime>
|
|
#include <thread>
|
|
#include <future>
|
|
|
|
|
|
namespace ProfileEvents
|
|
{
|
|
extern const Event ReplicatedPartMerges;
|
|
extern const Event ReplicatedPartFailedFetches;
|
|
extern const Event ReplicatedPartFetchesOfMerged;
|
|
extern const Event ObsoleteReplicatedParts;
|
|
extern const Event ReplicatedPartFetches;
|
|
extern const Event DataAfterMergeDiffersFromReplica;
|
|
}
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int NO_ZOOKEEPER;
|
|
extern const int INCORRECT_DATA;
|
|
extern const int INCOMPATIBLE_COLUMNS;
|
|
extern const int REPLICA_IS_ALREADY_EXIST;
|
|
extern const int NO_SUCH_REPLICA;
|
|
extern const int NO_REPLICA_HAS_PART;
|
|
extern const int LOGICAL_ERROR;
|
|
extern const int TOO_MANY_UNEXPECTED_DATA_PARTS;
|
|
extern const int ABORTED;
|
|
extern const int REPLICA_IS_NOT_IN_QUORUM;
|
|
extern const int TABLE_IS_READ_ONLY;
|
|
extern const int NOT_FOUND_NODE;
|
|
extern const int NO_ACTIVE_REPLICAS;
|
|
extern const int LEADERSHIP_CHANGED;
|
|
extern const int TABLE_IS_READ_ONLY;
|
|
extern const int TABLE_WAS_NOT_DROPPED;
|
|
extern const int PARTITION_ALREADY_EXISTS;
|
|
extern const int TOO_MANY_RETRIES_TO_FETCH_PARTS;
|
|
extern const int RECEIVED_ERROR_FROM_REMOTE_IO_SERVER;
|
|
extern const int PARTITION_DOESNT_EXIST;
|
|
extern const int CHECKSUM_DOESNT_MATCH;
|
|
extern const int BAD_SIZE_OF_FILE_IN_DATA_PART;
|
|
extern const int UNEXPECTED_FILE_IN_DATA_PART;
|
|
extern const int NO_FILE_IN_DATA_PART;
|
|
extern const int UNFINISHED;
|
|
extern const int METADATA_MISMATCH;
|
|
extern const int RECEIVED_ERROR_TOO_MANY_REQUESTS;
|
|
extern const int TOO_MANY_FETCHES;
|
|
extern const int BAD_DATA_PART_NAME;
|
|
extern const int PART_IS_TEMPORARILY_LOCKED;
|
|
extern const int INCORRECT_FILE_NAME;
|
|
extern const int CANNOT_ASSIGN_OPTIMIZE;
|
|
}
|
|
|
|
|
|
static const auto QUEUE_UPDATE_ERROR_SLEEP_MS = 1 * 1000;
|
|
static const auto MERGE_SELECTING_SLEEP_MS = 5 * 1000;
|
|
|
|
|
|
/** There are three places for each part, where it should be
|
|
* 1. In the RAM, MergeTreeData::data_parts, all_data_parts.
|
|
* 2. In the filesystem (FS), the directory with the data of the table.
|
|
* 3. in ZooKeeper (ZK).
|
|
*
|
|
* When adding a part, it must be added immediately to these three places.
|
|
* This is done like this
|
|
* - [FS] first write the part into a temporary directory on the filesystem;
|
|
* - [FS] rename the temporary part to the result on the filesystem;
|
|
* - [RAM] immediately afterwards add it to the `data_parts`, and remove from `data_parts` any parts covered by this one;
|
|
* - [RAM] also set the `Transaction` object, which in case of an exception (in next point),
|
|
* rolls back the changes in `data_parts` (from the previous point) back;
|
|
* - [ZK] then send a transaction (multi) to add a part to ZooKeeper (and some more actions);
|
|
* - [FS, ZK] by the way, removing the covered (old) parts from filesystem, from ZooKeeper and from `all_data_parts`
|
|
* is delayed, after a few minutes.
|
|
*
|
|
* There is no atomicity here.
|
|
* It could be possible to achieve atomicity using undo/redo logs and a flag in `DataPart` when it is completely ready.
|
|
* But it would be inconvenient - I would have to write undo/redo logs for each `Part` in ZK, and this would increase already large number of interactions.
|
|
*
|
|
* Instead, we are forced to work in a situation where at any time
|
|
* (from another thread, or after server restart), there may be an unfinished transaction.
|
|
* (note - for this the part should be in RAM)
|
|
* From these cases the most frequent one is when the part is already in the data_parts, but it's not yet in ZooKeeper.
|
|
* This case must be distinguished from the case where such a situation is achieved due to some kind of damage to the state.
|
|
*
|
|
* Do this with the threshold for the time.
|
|
* If the part is young enough, its lack in ZooKeeper will be perceived optimistically - as if it just did not have time to be added there
|
|
* - as if the transaction has not yet been executed, but will soon be executed.
|
|
* And if the part is old, its absence in ZooKeeper will be perceived as an unfinished transaction that needs to be rolled back.
|
|
*
|
|
* PS. Perhaps it would be better to add a flag to the DataPart that a part is inserted into ZK.
|
|
* But here it's too easy to get confused with the consistency of this flag.
|
|
*/
|
|
extern const int MAX_AGE_OF_LOCAL_PART_THAT_WASNT_ADDED_TO_ZOOKEEPER = 5 * 60;
|
|
|
|
|
|
/** For randomized selection of replicas. */
|
|
/// avoid error: non-local variable 'DB::rng' declared '__thread' needs dynamic initialization
|
|
#ifndef __APPLE__
|
|
thread_local
|
|
#endif
|
|
pcg64 rng{randomSeed()};
|
|
|
|
|
|
void StorageReplicatedMergeTree::setZooKeeper(zkutil::ZooKeeperPtr zookeeper)
|
|
{
|
|
std::lock_guard<std::mutex> lock(current_zookeeper_mutex);
|
|
current_zookeeper = zookeeper;
|
|
}
|
|
|
|
zkutil::ZooKeeperPtr StorageReplicatedMergeTree::tryGetZooKeeper()
|
|
{
|
|
std::lock_guard<std::mutex> lock(current_zookeeper_mutex);
|
|
return current_zookeeper;
|
|
}
|
|
|
|
zkutil::ZooKeeperPtr StorageReplicatedMergeTree::getZooKeeper()
|
|
{
|
|
auto res = tryGetZooKeeper();
|
|
if (!res)
|
|
throw Exception("Cannot get ZooKeeper", ErrorCodes::NO_ZOOKEEPER);
|
|
return res;
|
|
}
|
|
|
|
|
|
StorageReplicatedMergeTree::StorageReplicatedMergeTree(
|
|
const String & zookeeper_path_,
|
|
const String & replica_name_,
|
|
bool attach,
|
|
const String & path_, const String & database_name_, const String & name_,
|
|
const ColumnsDescription & columns_,
|
|
Context & context_,
|
|
const ASTPtr & primary_expr_ast_,
|
|
const ASTPtr & secondary_sorting_expr_list_,
|
|
const String & date_column_name,
|
|
const ASTPtr & partition_expr_ast_,
|
|
const ASTPtr & sampling_expression_,
|
|
const MergeTreeData::MergingParams & merging_params_,
|
|
const MergeTreeSettings & settings_,
|
|
bool has_force_restore_data_flag)
|
|
: context(context_),
|
|
current_zookeeper(context.getZooKeeper()), database_name(database_name_),
|
|
table_name(name_), full_path(path_ + escapeForFileName(table_name) + '/'),
|
|
zookeeper_path(context.getMacros()->expand(zookeeper_path_)),
|
|
replica_name(context.getMacros()->expand(replica_name_)),
|
|
data(database_name, table_name,
|
|
full_path, columns_,
|
|
context_, primary_expr_ast_, secondary_sorting_expr_list_, date_column_name, partition_expr_ast_,
|
|
sampling_expression_, merging_params_,
|
|
settings_, true, attach,
|
|
[this] (const std::string & name) { enqueuePartForCheck(name); }),
|
|
reader(data), writer(data), merger(data, context.getBackgroundPool()), queue(data.format_version),
|
|
fetcher(data),
|
|
shutdown_event(false), part_check_thread(*this),
|
|
log(&Logger::get(database_name + "." + table_name + " (StorageReplicatedMergeTree)"))
|
|
{
|
|
if (path_.empty())
|
|
throw Exception("ReplicatedMergeTree storages require data path", ErrorCodes::INCORRECT_FILE_NAME);
|
|
|
|
if (!zookeeper_path.empty() && zookeeper_path.back() == '/')
|
|
zookeeper_path.resize(zookeeper_path.size() - 1);
|
|
/// If zookeeper chroot prefix is used, path should starts with '/', because chroot concatenates without it.
|
|
if (!zookeeper_path.empty() && zookeeper_path.front() != '/')
|
|
zookeeper_path = "/" + zookeeper_path;
|
|
replica_path = zookeeper_path + "/replicas/" + replica_name;
|
|
|
|
bool skip_sanity_checks = false;
|
|
|
|
try
|
|
{
|
|
if (current_zookeeper && current_zookeeper->exists(replica_path + "/flags/force_restore_data"))
|
|
{
|
|
skip_sanity_checks = true;
|
|
current_zookeeper->remove(replica_path + "/flags/force_restore_data");
|
|
|
|
LOG_WARNING(log, "Skipping the limits on severity of changes to data parts and columns (flag "
|
|
<< replica_path << "/flags/force_restore_data).");
|
|
}
|
|
else if (has_force_restore_data_flag)
|
|
{
|
|
skip_sanity_checks = true;
|
|
|
|
LOG_WARNING(log, "Skipping the limits on severity of changes to data parts and columns (flag force_restore_data).");
|
|
}
|
|
}
|
|
catch (const zkutil::KeeperException & e)
|
|
{
|
|
/// Failed to connect to ZK (this became known when trying to perform the first operation).
|
|
if (e.code == ZooKeeperImpl::ZooKeeper::ZCONNECTIONLOSS)
|
|
{
|
|
tryLogCurrentException(__PRETTY_FUNCTION__);
|
|
current_zookeeper = nullptr;
|
|
}
|
|
else
|
|
throw;
|
|
}
|
|
|
|
data.loadDataParts(skip_sanity_checks);
|
|
|
|
if (!current_zookeeper)
|
|
{
|
|
if (!attach)
|
|
throw Exception("Can't create replicated table without ZooKeeper", ErrorCodes::NO_ZOOKEEPER);
|
|
|
|
/// Do not activate the replica. It will be readonly.
|
|
LOG_ERROR(log, "No ZooKeeper: table will be in readonly mode.");
|
|
is_readonly = true;
|
|
return;
|
|
}
|
|
|
|
if (!attach)
|
|
{
|
|
if (!data.getDataParts().empty())
|
|
throw Exception("Data directory for table already containing data parts - probably it was unclean DROP table or manual intervention. You must either clear directory by hand or use ATTACH TABLE instead of CREATE TABLE if you need to use that parts.", ErrorCodes::INCORRECT_DATA);
|
|
|
|
createTableIfNotExists();
|
|
|
|
checkTableStructure(false, false);
|
|
createReplica();
|
|
}
|
|
else
|
|
{
|
|
checkTableStructure(skip_sanity_checks, true);
|
|
checkParts(skip_sanity_checks);
|
|
|
|
/// Temporary directories contain unfinalized results of Merges or Fetches (after forced restart)
|
|
/// and don't allow to reinitialize them, so delete each of them immediately
|
|
data.clearOldTemporaryDirectories(0);
|
|
}
|
|
|
|
createNewZooKeeperNodes();
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::createNewZooKeeperNodes()
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
/// Working with quorum.
|
|
zookeeper->createIfNotExists(zookeeper_path + "/quorum", "");
|
|
zookeeper->createIfNotExists(zookeeper_path + "/quorum/last_part", "");
|
|
zookeeper->createIfNotExists(zookeeper_path + "/quorum/failed_parts", "");
|
|
|
|
/// Tracking lag of replicas.
|
|
zookeeper->createIfNotExists(replica_path + "/min_unprocessed_insert_time", "");
|
|
zookeeper->createIfNotExists(replica_path + "/max_processed_insert_time", "");
|
|
}
|
|
|
|
|
|
static String formattedAST(const ASTPtr & ast)
|
|
{
|
|
if (!ast)
|
|
return "";
|
|
std::stringstream ss;
|
|
formatAST(*ast, ss, false, true);
|
|
return ss.str();
|
|
}
|
|
|
|
|
|
namespace
|
|
{
|
|
/** The basic parameters of table engine for saving in ZooKeeper.
|
|
* Lets you verify that they match local ones.
|
|
*/
|
|
struct TableMetadata
|
|
{
|
|
const MergeTreeData & data;
|
|
|
|
explicit TableMetadata(const MergeTreeData & data_)
|
|
: data(data_) {}
|
|
|
|
void write(WriteBuffer & out) const
|
|
{
|
|
out << "metadata format version: 1" << "\n"
|
|
<< "date column: ";
|
|
|
|
if (data.format_version < MERGE_TREE_DATA_MIN_FORMAT_VERSION_WITH_CUSTOM_PARTITIONING)
|
|
out << data.minmax_idx_columns[data.minmax_idx_date_column_pos] << "\n";
|
|
else
|
|
out << "\n";
|
|
|
|
out << "sampling expression: " << formattedAST(data.sampling_expression) << "\n"
|
|
<< "index granularity: " << data.index_granularity << "\n"
|
|
<< "mode: " << static_cast<int>(data.merging_params.mode) << "\n"
|
|
<< "sign column: " << data.merging_params.sign_column << "\n"
|
|
<< "primary key: " << formattedAST(data.primary_expr_ast) << "\n";
|
|
|
|
if (data.format_version >= MERGE_TREE_DATA_MIN_FORMAT_VERSION_WITH_CUSTOM_PARTITIONING)
|
|
{
|
|
out << "data format version: " << data.format_version.toUnderType() << "\n";
|
|
out << "partition key: " << formattedAST(data.partition_expr_ast) << "\n";
|
|
}
|
|
}
|
|
|
|
String toString() const
|
|
{
|
|
WriteBufferFromOwnString out;
|
|
write(out);
|
|
return out.str();
|
|
}
|
|
|
|
void check(ReadBuffer & in) const
|
|
{
|
|
/// TODO Can be made less cumbersome.
|
|
|
|
in >> "metadata format version: 1";
|
|
|
|
in >> "\ndate column: ";
|
|
String read_date_column;
|
|
in >> read_date_column;
|
|
|
|
if (data.format_version < MERGE_TREE_DATA_MIN_FORMAT_VERSION_WITH_CUSTOM_PARTITIONING)
|
|
{
|
|
const String & local_date_column = data.minmax_idx_columns[data.minmax_idx_date_column_pos];
|
|
if (local_date_column != read_date_column)
|
|
throw Exception("Existing table metadata in ZooKeeper differs in date index column."
|
|
" Stored in ZooKeeper: " + read_date_column + ", local: " + local_date_column,
|
|
ErrorCodes::METADATA_MISMATCH);
|
|
}
|
|
else if (!read_date_column.empty())
|
|
throw Exception(
|
|
"Existing table metadata in ZooKeeper differs in date index column."
|
|
" Stored in ZooKeeper: " + read_date_column + ", local is custom-partitioned.",
|
|
ErrorCodes::METADATA_MISMATCH);
|
|
|
|
in >> "\nsampling expression: ";
|
|
String read_sample_expression;
|
|
String local_sample_expression = formattedAST(data.sampling_expression);
|
|
in >> read_sample_expression;
|
|
|
|
if (read_sample_expression != local_sample_expression)
|
|
throw Exception("Existing table metadata in ZooKeeper differs in sample expression."
|
|
" Stored in ZooKeeper: " + read_sample_expression + ", local: " + local_sample_expression,
|
|
ErrorCodes::METADATA_MISMATCH);
|
|
|
|
in >> "\nindex granularity: ";
|
|
size_t read_index_granularity = 0;
|
|
in >> read_index_granularity;
|
|
|
|
if (read_index_granularity != data.index_granularity)
|
|
throw Exception("Existing table metadata in ZooKeeper differs in index granularity."
|
|
" Stored in ZooKeeper: " + DB::toString(read_index_granularity) + ", local: " + DB::toString(data.index_granularity),
|
|
ErrorCodes::METADATA_MISMATCH);
|
|
|
|
in >> "\nmode: ";
|
|
int read_mode = 0;
|
|
in >> read_mode;
|
|
|
|
if (read_mode != static_cast<int>(data.merging_params.mode))
|
|
throw Exception("Existing table metadata in ZooKeeper differs in mode of merge operation."
|
|
" Stored in ZooKeeper: " + DB::toString(read_mode) + ", local: "
|
|
+ DB::toString(static_cast<int>(data.merging_params.mode)),
|
|
ErrorCodes::METADATA_MISMATCH);
|
|
|
|
in >> "\nsign column: ";
|
|
String read_sign_column;
|
|
in >> read_sign_column;
|
|
|
|
if (read_sign_column != data.merging_params.sign_column)
|
|
throw Exception("Existing table metadata in ZooKeeper differs in sign column."
|
|
" Stored in ZooKeeper: " + read_sign_column + ", local: " + data.merging_params.sign_column,
|
|
ErrorCodes::METADATA_MISMATCH);
|
|
|
|
in >> "\nprimary key: ";
|
|
String read_primary_key;
|
|
String local_primary_key = formattedAST(data.primary_expr_ast);
|
|
in >> read_primary_key;
|
|
|
|
/// NOTE: You can make a less strict check of match expressions so that tables do not break from small changes
|
|
/// in formatAST code.
|
|
if (read_primary_key != local_primary_key)
|
|
throw Exception("Existing table metadata in ZooKeeper differs in primary key."
|
|
" Stored in ZooKeeper: " + read_primary_key + ", local: " + local_primary_key,
|
|
ErrorCodes::METADATA_MISMATCH);
|
|
|
|
in >> "\n";
|
|
MergeTreeDataFormatVersion read_data_format_version;
|
|
if (in.eof())
|
|
read_data_format_version = 0;
|
|
else
|
|
{
|
|
in >> "data format version: ";
|
|
in >> read_data_format_version.toUnderType();
|
|
}
|
|
|
|
if (read_data_format_version != data.format_version)
|
|
throw Exception("Existing table metadata in ZooKeeper differs in data format version."
|
|
" Stored in ZooKeeper: " + DB::toString(read_data_format_version.toUnderType()) +
|
|
", local: " + DB::toString(data.format_version.toUnderType()),
|
|
ErrorCodes::METADATA_MISMATCH);
|
|
|
|
if (data.format_version >= MERGE_TREE_DATA_MIN_FORMAT_VERSION_WITH_CUSTOM_PARTITIONING)
|
|
{
|
|
in >> "\npartition key: ";
|
|
String read_partition_key;
|
|
String local_partition_key = formattedAST(data.partition_expr_ast);
|
|
in >> read_partition_key;
|
|
|
|
if (read_partition_key != local_partition_key)
|
|
throw Exception(
|
|
"Existing table metadata in ZooKeeper differs in partition key expression."
|
|
" Stored in ZooKeeper: " + read_partition_key + ", local: " + local_partition_key,
|
|
ErrorCodes::METADATA_MISMATCH);
|
|
|
|
in >> "\n";
|
|
}
|
|
|
|
assertEOF(in);
|
|
}
|
|
|
|
void check(const String & s) const
|
|
{
|
|
ReadBufferFromString in(s);
|
|
check(in);
|
|
}
|
|
};
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::createTableIfNotExists()
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
if (zookeeper->exists(zookeeper_path))
|
|
return;
|
|
|
|
LOG_DEBUG(log, "Creating table " << zookeeper_path);
|
|
|
|
zookeeper->createAncestors(zookeeper_path);
|
|
|
|
/// We write metadata of table so that the replicas can check table parameters with them.
|
|
String metadata = TableMetadata(data).toString();
|
|
|
|
zkutil::Requests ops;
|
|
ops.emplace_back(zkutil::makeCreateRequest(zookeeper_path, "",
|
|
zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(zookeeper_path + "/metadata", metadata,
|
|
zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(zookeeper_path + "/columns", getColumns().toString(),
|
|
zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(zookeeper_path + "/log", "",
|
|
zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(zookeeper_path + "/blocks", "",
|
|
zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(zookeeper_path + "/block_numbers", "",
|
|
zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(zookeeper_path + "/nonincrement_block_numbers", "",
|
|
zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(zookeeper_path + "/leader_election", "",
|
|
zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(zookeeper_path + "/temp", "",
|
|
zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(zookeeper_path + "/replicas", "",
|
|
zkutil::CreateMode::Persistent));
|
|
|
|
zkutil::Responses responses;
|
|
auto code = zookeeper->tryMulti(ops, responses);
|
|
if (code && code != ZooKeeperImpl::ZooKeeper::ZNODEEXISTS)
|
|
throw zkutil::KeeperException(code);
|
|
}
|
|
|
|
|
|
/** Verify that list of columns and table settings match those specified in ZK (/ metadata).
|
|
* If not, throw an exception.
|
|
*/
|
|
void StorageReplicatedMergeTree::checkTableStructure(bool skip_sanity_checks, bool allow_alter)
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
String metadata_str = zookeeper->get(zookeeper_path + "/metadata");
|
|
TableMetadata(data).check(metadata_str);
|
|
|
|
zkutil::Stat stat;
|
|
auto columns_from_zk = ColumnsDescription::parse(zookeeper->get(zookeeper_path + "/columns", &stat));
|
|
columns_version = stat.version;
|
|
|
|
const ColumnsDescription & old_columns = getColumns();
|
|
if (columns_from_zk != old_columns)
|
|
{
|
|
if (allow_alter &&
|
|
(skip_sanity_checks ||
|
|
old_columns.ordinary.sizeOfDifference(columns_from_zk.ordinary) +
|
|
old_columns.materialized.sizeOfDifference(columns_from_zk.materialized) <= 2))
|
|
{
|
|
LOG_WARNING(log, "Table structure in ZooKeeper is a little different from local table structure. Assuming ALTER.");
|
|
|
|
/// Without any locks, because table has not been created yet.
|
|
context.getDatabase(database_name)->alterTable(context, table_name, columns_from_zk, {});
|
|
|
|
setColumns(std::move(columns_from_zk));
|
|
}
|
|
else
|
|
{
|
|
throw Exception("Table structure in ZooKeeper is too different from local table structure",
|
|
ErrorCodes::INCOMPATIBLE_COLUMNS);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/** If necessary, restore a part, replica itself adds a record for its receipt.
|
|
* What time should I put for this entry in the queue? Time is taken into account when calculating lag of replica.
|
|
* For these purposes, it makes sense to use creation time of missing part
|
|
* (that is, in calculating lag, it will be taken into account how old is the part we need to recover).
|
|
*/
|
|
static time_t tryGetPartCreateTime(zkutil::ZooKeeperPtr & zookeeper, const String & replica_path, const String & part_name)
|
|
{
|
|
time_t res = 0;
|
|
|
|
/// We get creation time of part, if it still exists (was not merged, for example).
|
|
zkutil::Stat stat;
|
|
String unused;
|
|
if (zookeeper->tryGet(replica_path + "/parts/" + part_name, unused, &stat))
|
|
res = stat.ctime / 1000;
|
|
|
|
return res;
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::createReplica()
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
LOG_DEBUG(log, "Creating replica " << replica_path);
|
|
|
|
/// Create an empty replica. We'll create `columns` node at the end - we'll use it as a sign that replica creation is complete.
|
|
zkutil::Requests ops;
|
|
ops.emplace_back(zkutil::makeCreateRequest(replica_path, "", zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(replica_path + "/host", "", zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(replica_path + "/log_pointer", "", zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(replica_path + "/queue", "", zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(replica_path + "/parts", "", zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(replica_path + "/flags", "", zkutil::CreateMode::Persistent));
|
|
|
|
try
|
|
{
|
|
zookeeper->multi(ops);
|
|
}
|
|
catch (const zkutil::KeeperException & e)
|
|
{
|
|
if (e.code == ZooKeeperImpl::ZooKeeper::ZNODEEXISTS)
|
|
throw Exception("Replica " + replica_path + " already exists.", ErrorCodes::REPLICA_IS_ALREADY_EXIST);
|
|
|
|
throw;
|
|
}
|
|
|
|
/** You need to change the data of nodes/replicas to anything, so that the thread that removes old entries in the log,
|
|
* stumbled over this change and does not delete the entries we have not yet read.
|
|
*/
|
|
zookeeper->set(zookeeper_path + "/replicas", "last added replica: " + replica_name);
|
|
|
|
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
|
|
|
|
/** "Reference" replica, from which we take information about the set of parts, queue and pointer to the log.
|
|
* Take random replica created earlier than this.
|
|
*/
|
|
String source_replica;
|
|
|
|
zkutil::Stat stat;
|
|
zookeeper->exists(replica_path, &stat);
|
|
auto my_create_time = stat.czxid;
|
|
|
|
std::shuffle(replicas.begin(), replicas.end(), rng);
|
|
for (const String & replica : replicas)
|
|
{
|
|
if (!zookeeper->exists(zookeeper_path + "/replicas/" + replica, &stat))
|
|
throw Exception("Replica " + zookeeper_path + "/replicas/" + replica + " was removed from right under our feet.",
|
|
ErrorCodes::NO_SUCH_REPLICA);
|
|
if (stat.czxid < my_create_time)
|
|
{
|
|
source_replica = replica;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (source_replica.empty())
|
|
{
|
|
LOG_INFO(log, "This is the first replica");
|
|
}
|
|
else
|
|
{
|
|
LOG_INFO(log, "Will mimic " << source_replica);
|
|
|
|
String source_path = zookeeper_path + "/replicas/" + source_replica;
|
|
|
|
/** If the reference/master replica is not yet fully created, let's wait.
|
|
* NOTE: If something went wrong while creating it, we can hang around forever.
|
|
* You can create an ephemeral node at the time of creation to make sure that the replica is created, and not abandoned.
|
|
* The same can be done for the table. You can automatically delete a replica/table node,
|
|
* if you see that it was not created up to the end, and the one who created it died.
|
|
*/
|
|
while (!zookeeper->exists(source_path + "/columns"))
|
|
{
|
|
LOG_INFO(log, "Waiting for replica " << source_path << " to be fully created");
|
|
|
|
zkutil::EventPtr event = std::make_shared<Poco::Event>();
|
|
if (zookeeper->exists(source_path + "/columns", nullptr, event))
|
|
{
|
|
LOG_WARNING(log, "Oops, a watch has leaked");
|
|
break;
|
|
}
|
|
|
|
event->wait();
|
|
}
|
|
|
|
/// The order of the following three actions is important. Entries in the log can be duplicated, but they can not be lost.
|
|
|
|
/// Copy reference to the log from `reference/master` replica.
|
|
zookeeper->set(replica_path + "/log_pointer", zookeeper->get(source_path + "/log_pointer"));
|
|
|
|
/// Let's remember the queue of the reference/master replica.
|
|
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);
|
|
}
|
|
|
|
/// Add to the queue jobs to receive all the active parts that the reference/master replica has.
|
|
Strings parts = zookeeper->getChildren(source_path + "/parts");
|
|
ActiveDataPartSet active_parts_set(data.format_version, parts);
|
|
|
|
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;
|
|
log_entry.create_time = tryGetPartCreateTime(zookeeper, source_path, name);
|
|
|
|
zookeeper->create(replica_path + "/queue/queue-", log_entry.toString(), zkutil::CreateMode::PersistentSequential);
|
|
}
|
|
LOG_DEBUG(log, "Queued " << active_parts.size() << " parts to be fetched");
|
|
|
|
/// Add content of the reference/master replica queue to the queue.
|
|
for (const String & entry : source_queue)
|
|
{
|
|
zookeeper->create(replica_path + "/queue/queue-", entry, zkutil::CreateMode::PersistentSequential);
|
|
}
|
|
|
|
/// It will then be loaded into the queue variable in `queue.initialize` method.
|
|
|
|
LOG_DEBUG(log, "Copied " << source_queue.size() << " queue entries");
|
|
}
|
|
|
|
zookeeper->create(replica_path + "/columns", getColumns().toString(), zkutil::CreateMode::Persistent);
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::checkParts(bool skip_sanity_checks)
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
Strings expected_parts_vec = zookeeper->getChildren(replica_path + "/parts");
|
|
|
|
/// Parts in ZK.
|
|
NameSet expected_parts(expected_parts_vec.begin(), expected_parts_vec.end());
|
|
|
|
/// There are no PreCommitted parts at startup.
|
|
auto parts = data.getDataParts({MergeTreeDataPartState::Committed, MergeTreeDataPartState::Outdated});
|
|
|
|
/// Local parts that are not in ZK.
|
|
MergeTreeData::DataParts unexpected_parts;
|
|
|
|
for (const auto & part : parts)
|
|
{
|
|
if (expected_parts.count(part->name))
|
|
expected_parts.erase(part->name);
|
|
else
|
|
unexpected_parts.insert(part);
|
|
}
|
|
|
|
/// Which local parts to added into ZK.
|
|
MergeTreeData::DataPartsVector parts_to_add;
|
|
UInt64 parts_to_add_rows = 0;
|
|
|
|
/// Which parts should be taken from other replicas.
|
|
Strings parts_to_fetch;
|
|
|
|
for (const String & missing_name : expected_parts)
|
|
{
|
|
/// If locally some part is missing, but there is a part covering it, you can replace it in ZK with the covering one.
|
|
auto containing = data.getActiveContainingPart(missing_name);
|
|
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);
|
|
parts_to_add_rows += containing->rows_count;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
LOG_ERROR(log, "Fetching missing part " << missing_name);
|
|
parts_to_fetch.push_back(missing_name);
|
|
}
|
|
}
|
|
|
|
for (const String & name : parts_to_fetch)
|
|
expected_parts.erase(name);
|
|
|
|
|
|
/** To check the adequacy, for the parts that are in the FS, but not in ZK, we will only consider not the most recent parts.
|
|
* Because unexpected new parts usually arise only because they did not have time to enroll in ZK with a rough restart of the server.
|
|
* It also occurs from deduplicated parts that did not have time to retire.
|
|
*/
|
|
size_t unexpected_parts_nonnew = 0;
|
|
UInt64 unexpected_parts_nonnew_rows = 0;
|
|
UInt64 unexpected_parts_rows = 0;
|
|
for (const auto & part : unexpected_parts)
|
|
{
|
|
if (part->info.level > 0)
|
|
{
|
|
++unexpected_parts_nonnew;
|
|
unexpected_parts_nonnew_rows += part->rows_count;
|
|
}
|
|
|
|
unexpected_parts_rows += part->rows_count;
|
|
}
|
|
|
|
/// Additional helpful statistics
|
|
auto get_blocks_count_in_data_part = [&] (const String & part_name) -> UInt64
|
|
{
|
|
MergeTreePartInfo part_info;
|
|
if (MergeTreePartInfo::tryParsePartName(part_name, &part_info, data.format_version))
|
|
return part_info.getBlocksCount();
|
|
|
|
LOG_ERROR(log, "Unexpected part name: " << part_name);
|
|
return 0;
|
|
};
|
|
|
|
UInt64 parts_to_fetch_blocks = 0;
|
|
for (const String & name : parts_to_fetch)
|
|
parts_to_fetch_blocks += get_blocks_count_in_data_part(name);
|
|
|
|
UInt64 expected_parts_blocks = 0;
|
|
for (const String & name : expected_parts)
|
|
expected_parts_blocks += get_blocks_count_in_data_part(name);
|
|
|
|
std::stringstream sanity_report;
|
|
sanity_report << "There are "
|
|
<< unexpected_parts.size() << " unexpected parts with " << unexpected_parts_rows << " rows ("
|
|
<< unexpected_parts_nonnew << " of them is not just-written with " << unexpected_parts_rows << " rows), "
|
|
<< parts_to_add.size() << " unexpectedly merged parts with " << parts_to_add_rows << " rows, "
|
|
<< expected_parts.size() << " missing obsolete parts (with " << expected_parts_blocks << " blocks), "
|
|
<< parts_to_fetch.size() << " missing parts (with " << parts_to_fetch_blocks << " blocks).";
|
|
|
|
/** We can automatically synchronize data,
|
|
* if the ratio of the total number of errors to the total number of parts (minimum - on the local filesystem or in ZK)
|
|
* is no more than some threshold (for example 50%).
|
|
*
|
|
* A large ratio of mismatches in the data on the filesystem and the expected data
|
|
* may indicate a configuration error (the server accidentally connected as a replica not from right shard).
|
|
* In this case, the protection mechanism does not allow the server to start.
|
|
*/
|
|
|
|
UInt64 total_rows_on_filesystem = 0;
|
|
for (const auto & part : parts)
|
|
total_rows_on_filesystem += part->rows_count;
|
|
|
|
UInt64 total_suspicious_rows = parts_to_add_rows + unexpected_parts_rows;
|
|
UInt64 total_suspicious_rows_no_new = parts_to_add_rows + unexpected_parts_nonnew_rows;
|
|
|
|
bool insane = total_suspicious_rows > total_rows_on_filesystem * data.settings.replicated_max_ratio_of_wrong_parts;
|
|
|
|
if (insane && !skip_sanity_checks)
|
|
{
|
|
std::stringstream why;
|
|
why << "The local set of parts of table " << database_name << "." << table_name << " doesn't look like the set of parts "
|
|
<< "in ZooKeeper: "
|
|
<< formatReadableQuantity(total_suspicious_rows) << " rows of " << formatReadableQuantity(total_rows_on_filesystem)
|
|
<< " total rows in filesystem are suspicious.";
|
|
|
|
throw Exception(why.str() + " " + sanity_report.str(), ErrorCodes::TOO_MANY_UNEXPECTED_DATA_PARTS);
|
|
}
|
|
|
|
if (total_suspicious_rows_no_new > 0)
|
|
LOG_WARNING(log, sanity_report.str());
|
|
|
|
/// Add information to the ZK about the parts that cover the missing parts.
|
|
for (const MergeTreeData::DataPartPtr & part : parts_to_add)
|
|
{
|
|
LOG_ERROR(log, "Adding unexpected local part to ZooKeeper: " << part->name);
|
|
|
|
zkutil::Requests ops;
|
|
checkPartChecksumsAndAddCommitOps(zookeeper, part, ops);
|
|
zookeeper->multi(ops);
|
|
}
|
|
|
|
/// Remove from ZK information about the parts covered by the newly added ones.
|
|
{
|
|
for (const String & name : expected_parts)
|
|
LOG_ERROR(log, "Removing unexpectedly merged local part from ZooKeeper: " << name);
|
|
|
|
removePartsFromZooKeeper(zookeeper, Strings(expected_parts.begin(), expected_parts.end()));
|
|
}
|
|
|
|
/// Add to the queue job to pick up the missing parts from other replicas and remove from ZK the information that we have them.
|
|
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;
|
|
log_entry.source_replica = "";
|
|
log_entry.new_part_name = name;
|
|
log_entry.create_time = tryGetPartCreateTime(zookeeper, replica_path, name);
|
|
|
|
/// We assume that this occurs before the queue is loaded (queue.initialize).
|
|
zkutil::Requests ops;
|
|
removePartFromZooKeeper(name, ops);
|
|
ops.emplace_back(zkutil::makeCreateRequest(
|
|
replica_path + "/queue/queue-", log_entry.toString(), zkutil::CreateMode::PersistentSequential));
|
|
zookeeper->multi(ops);
|
|
}
|
|
|
|
/// Remove extra local parts.
|
|
for (const MergeTreeData::DataPartPtr & part : unexpected_parts)
|
|
{
|
|
LOG_ERROR(log, "Renaming unexpected part " << part->name << " to ignored_" + part->name);
|
|
data.renameAndDetachPart(part, "ignored_", true);
|
|
}
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::checkPartChecksumsAndAddCommitOps(const zkutil::ZooKeeperPtr & zookeeper,
|
|
const MergeTreeData::DataPartPtr & part, zkutil::Requests & ops, String part_name, NameSet * absent_replicas_paths)
|
|
{
|
|
if (part_name.empty())
|
|
part_name = part->name;
|
|
|
|
check(part->columns);
|
|
int expected_columns_version = columns_version;
|
|
|
|
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
|
|
std::shuffle(replicas.begin(), replicas.end(), rng);
|
|
String expected_columns_str = part->columns.toString();
|
|
bool has_been_alredy_added = false;
|
|
|
|
for (const String & replica : replicas)
|
|
{
|
|
zkutil::Stat stat_before, stat_after;
|
|
String current_part_path = zookeeper_path + "/replicas/" + replica + "/parts/" + part_name;
|
|
|
|
String columns_str;
|
|
if (!zookeeper->tryGet(current_part_path + "/columns", columns_str, &stat_before))
|
|
{
|
|
if (absent_replicas_paths)
|
|
absent_replicas_paths->emplace(current_part_path);
|
|
|
|
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;
|
|
}
|
|
|
|
String checksums_str;
|
|
/// Let's check that the node's version with the columns did not change while we were reading the checksums.
|
|
/// This ensures that the columns and the checksum refer to the same data.
|
|
if (!zookeeper->tryGet(current_part_path + "/checksums", checksums_str) ||
|
|
!zookeeper->exists(current_part_path + "/columns", &stat_after) ||
|
|
stat_before.version != stat_after.version)
|
|
{
|
|
LOG_INFO(log, "Not checking checksums of part " << part_name << " with replica " << replica
|
|
<< " because part changed while we were reading its checksums");
|
|
continue;
|
|
}
|
|
|
|
auto zk_checksums = MinimalisticDataPartChecksums::deserializeFrom(checksums_str);
|
|
zk_checksums.checkEqual(part->checksums, true);
|
|
|
|
if (replica == replica_name)
|
|
has_been_alredy_added = true;
|
|
}
|
|
|
|
if (!has_been_alredy_added)
|
|
{
|
|
String part_path = replica_path + "/parts/" + part_name;
|
|
|
|
ops.emplace_back(zkutil::makeCheckRequest(
|
|
zookeeper_path + "/columns", expected_columns_version));
|
|
ops.emplace_back(zkutil::makeCreateRequest(
|
|
part_path, "", zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(
|
|
part_path + "/columns", part->columns.toString(), zkutil::CreateMode::Persistent));
|
|
ops.emplace_back(zkutil::makeCreateRequest(
|
|
part_path + "/checksums", getChecksumsForZooKeeper(part->checksums), zkutil::CreateMode::Persistent));
|
|
}
|
|
else
|
|
{
|
|
LOG_WARNING(log, "checkPartAndAddToZooKeeper: node " << replica_path + "/parts/" + part_name << " already exists."
|
|
<< " Will not commit any nodes.");
|
|
}
|
|
}
|
|
|
|
MergeTreeData::DataPartsVector StorageReplicatedMergeTree::checkPartChecksumsAndCommit(MergeTreeData::Transaction & transaction,
|
|
const MergeTreeData::DataPartPtr & part)
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
while (true)
|
|
{
|
|
zkutil::Requests ops;
|
|
NameSet absent_part_paths_on_replicas;
|
|
|
|
/// Checksums are checked here and `ops` is filled. In fact, the part is added to ZK just below, when executing `multi`.
|
|
checkPartChecksumsAndAddCommitOps(zookeeper, part, ops, part->name, &absent_part_paths_on_replicas);
|
|
|
|
/// Do not commit if the part is obsolete, we have just briefly checked its checksums
|
|
if (transaction.isEmpty())
|
|
return {};
|
|
|
|
/// Will check that the part did not suddenly appear on skipped replicas
|
|
if (!absent_part_paths_on_replicas.empty())
|
|
{
|
|
zkutil::Requests new_ops;
|
|
for (const String & part_path : absent_part_paths_on_replicas)
|
|
{
|
|
new_ops.emplace_back(zkutil::makeCreateRequest(part_path, "", zkutil::CreateMode::Persistent));
|
|
new_ops.emplace_back(zkutil::makeRemoveRequest(part_path, -1));
|
|
}
|
|
|
|
/// Add check ops at the beginning
|
|
new_ops.insert(new_ops.end(), ops.begin(), ops.end());
|
|
ops = std::move(new_ops);
|
|
}
|
|
|
|
try
|
|
{
|
|
zookeeper->multi(ops);
|
|
return transaction.commit();
|
|
}
|
|
catch (const zkutil::KeeperMultiException & e)
|
|
{
|
|
size_t num_check_ops = 2 * absent_part_paths_on_replicas.size();
|
|
size_t failed_op_index = e.failed_op_index;
|
|
|
|
if (failed_op_index < num_check_ops && e.code == ZooKeeperImpl::ZooKeeper::ZNODEEXISTS)
|
|
{
|
|
LOG_INFO(log, "The part " << e.getPathForFirstFailedOp() << " on a replica suddenly appeared, will recheck checksums");
|
|
}
|
|
else
|
|
throw;
|
|
}
|
|
}
|
|
}
|
|
|
|
String StorageReplicatedMergeTree::getChecksumsForZooKeeper(const MergeTreeDataPartChecksums & checksums)
|
|
{
|
|
return MinimalisticDataPartChecksums::getSerializedString(checksums,
|
|
static_cast<bool>(data.settings.use_minimalistic_checksums_in_zookeeper));
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::pullLogsToQueue(zkutil::EventPtr next_update_event)
|
|
{
|
|
if (queue.pullLogsToQueue(getZooKeeper(), next_update_event))
|
|
{
|
|
if (queue_task_handle)
|
|
queue_task_handle->wake();
|
|
}
|
|
}
|
|
|
|
|
|
bool StorageReplicatedMergeTree::executeLogEntry(const LogEntry & entry)
|
|
{
|
|
if (entry.type == LogEntry::ATTACH_PART)
|
|
{
|
|
LOG_ERROR(log, "Log entries of type ATTACH_PART are obsolete. Skipping.");
|
|
return true;
|
|
}
|
|
|
|
if (entry.type == LogEntry::DROP_RANGE)
|
|
{
|
|
executeDropRange(entry);
|
|
return true;
|
|
}
|
|
|
|
if (entry.type == LogEntry::CLEAR_COLUMN)
|
|
{
|
|
executeClearColumnInPartition(entry);
|
|
return true;
|
|
}
|
|
|
|
if (entry.type == LogEntry::GET_PART ||
|
|
entry.type == LogEntry::MERGE_PARTS)
|
|
{
|
|
/// If we already have this part or a part covering it, we do not need to do anything.
|
|
/// The part may be still in the PreCommitted -> Committed transition so we first search
|
|
/// among PreCommitted parts to definitely find the desired part if it exists.
|
|
MergeTreeData::DataPartPtr existing_part = data.getPartIfExists(entry.new_part_name, {MergeTreeDataPartState::PreCommitted});
|
|
if (!existing_part)
|
|
existing_part = data.getActiveContainingPart(entry.new_part_name);
|
|
|
|
/// Even if the part is locally, it (in exceptional cases) may not be in ZooKeeper. Let's check that it is there.
|
|
if (existing_part && getZooKeeper()->exists(replica_path + "/parts/" + existing_part->name))
|
|
{
|
|
if (!(entry.type == LogEntry::GET_PART && entry.source_replica == replica_name))
|
|
{
|
|
LOG_DEBUG(log, "Skipping action for part " << entry.new_part_name << " because part " + existing_part->name + " already exists.");
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
|
|
if (entry.type == LogEntry::GET_PART && entry.source_replica == replica_name)
|
|
LOG_WARNING(log, "Part " << entry.new_part_name << " from own log doesn't exist.");
|
|
|
|
/// Perhaps we don't need this part, because during write with quorum, the quorum has failed (see below about `/quorum/failed_parts`).
|
|
if (entry.quorum && getZooKeeper()->exists(zookeeper_path + "/quorum/failed_parts/" + entry.new_part_name))
|
|
{
|
|
LOG_DEBUG(log, "Skipping action for part " << entry.new_part_name << " because quorum for that part was failed.");
|
|
return true; /// NOTE Deletion from `virtual_parts` is not done, but it is only necessary for merge.
|
|
}
|
|
|
|
bool do_fetch = false;
|
|
if (entry.type == LogEntry::GET_PART)
|
|
{
|
|
do_fetch = true;
|
|
}
|
|
else if (entry.type == LogEntry::MERGE_PARTS)
|
|
{
|
|
tryExecuteMerge(entry, do_fetch);
|
|
}
|
|
else
|
|
{
|
|
throw Exception("Unexpected log entry type: " + toString(static_cast<int>(entry.type)));
|
|
}
|
|
|
|
if (do_fetch)
|
|
return executeFetch(entry);
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::tryExecuteMerge(const StorageReplicatedMergeTree::LogEntry & entry, bool & do_fetch)
|
|
{
|
|
/// The caller has already decided to make the fetch
|
|
if (do_fetch)
|
|
return;
|
|
|
|
// Log source part names just in case
|
|
{
|
|
std::stringstream log_message;
|
|
log_message << "Executing log entry to merge parts ";
|
|
for (auto i : ext::range(0, entry.parts_to_merge.size()))
|
|
log_message << (i != 0 ? ", " : "") << entry.parts_to_merge[i];
|
|
log_message << " to " << entry.new_part_name;
|
|
|
|
LOG_TRACE(log, log_message.rdbuf());
|
|
}
|
|
|
|
MergeTreeData::DataPartsVector parts;
|
|
bool have_all_parts = true;
|
|
for (const String & name : entry.parts_to_merge)
|
|
{
|
|
MergeTreeData::DataPartPtr part = data.getActiveContainingPart(name);
|
|
if (!part)
|
|
{
|
|
have_all_parts = false;
|
|
break;
|
|
}
|
|
if (part->name != name)
|
|
{
|
|
LOG_WARNING(log, "Part " << name << " is covered by " << part->name
|
|
<< " but should be merged into " << entry.new_part_name << ". This shouldn't happen often.");
|
|
have_all_parts = false;
|
|
break;
|
|
}
|
|
parts.push_back(part);
|
|
}
|
|
|
|
if (!have_all_parts)
|
|
{
|
|
/// If you do not have all the necessary parts, try to take some already merged part from someone.
|
|
do_fetch = true;
|
|
LOG_DEBUG(log, "Don't have all parts for merge " << entry.new_part_name << "; will try to fetch it instead");
|
|
}
|
|
else if (entry.create_time + data.settings.prefer_fetch_merged_part_time_threshold.totalSeconds() <= time(nullptr))
|
|
{
|
|
/// If entry is old enough, and have enough size, and part are exists in any replica,
|
|
/// then prefer fetching of merged part from replica.
|
|
|
|
size_t sum_parts_bytes_on_disk = 0;
|
|
for (const auto & part : parts)
|
|
sum_parts_bytes_on_disk += part->bytes_on_disk;
|
|
|
|
if (sum_parts_bytes_on_disk >= data.settings.prefer_fetch_merged_part_size_threshold)
|
|
{
|
|
String replica = findReplicaHavingPart(entry.new_part_name, true); /// NOTE excessive ZK requests for same data later, may remove.
|
|
if (!replica.empty())
|
|
{
|
|
do_fetch = true;
|
|
LOG_DEBUG(log, "Prefer to fetch " << entry.new_part_name << " from replica " << replica);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (do_fetch)
|
|
return;
|
|
|
|
/// Start to make the main work
|
|
|
|
size_t estimated_space_for_merge = MergeTreeDataMerger::estimateDiskSpaceForMerge(parts);
|
|
|
|
/// Can throw an exception.
|
|
DiskSpaceMonitor::ReservationPtr reserved_space = DiskSpaceMonitor::reserve(full_path, estimated_space_for_merge);
|
|
|
|
auto table_lock = lockStructure(false, __PRETTY_FUNCTION__);
|
|
|
|
MergeList::EntryPtr merge_entry = context.getMergeList().insert(database_name, table_name, entry.new_part_name, parts);
|
|
MergeTreeData::Transaction transaction;
|
|
size_t aio_threshold = context.getSettings().min_bytes_to_use_direct_io;
|
|
|
|
MergeTreeDataMerger::FuturePart future_merged_part(parts);
|
|
if (future_merged_part.name != entry.new_part_name)
|
|
{
|
|
throw Exception("Future merged part name `" + future_merged_part.name + "` differs from part name in log entry: `"
|
|
+ entry.new_part_name + "`", ErrorCodes::BAD_DATA_PART_NAME);
|
|
}
|
|
|
|
/// Logging
|
|
Stopwatch stopwatch;
|
|
ExecutionStatus execution_status;
|
|
MergeTreeData::MutableDataPartPtr part;
|
|
|
|
auto write_part_log = [&] (const ExecutionStatus & execution_status)
|
|
{
|
|
try
|
|
{
|
|
auto part_log = context.getPartLog(database_name);
|
|
if (!part_log)
|
|
return;
|
|
|
|
PartLogElement part_log_elem;
|
|
|
|
part_log_elem.event_type = PartLogElement::MERGE_PARTS;
|
|
part_log_elem.event_time = time(nullptr);
|
|
/// TODO: Stop stopwatch in outer code to exclude ZK timings and so on
|
|
part_log_elem.duration_ms = stopwatch.elapsed() / 1000000;
|
|
|
|
part_log_elem.database_name = database_name;
|
|
part_log_elem.table_name = table_name;
|
|
part_log_elem.part_name = entry.new_part_name;
|
|
|
|
if (part)
|
|
part_log_elem.bytes_compressed_on_disk = part->bytes_on_disk;
|
|
|
|
part_log_elem.source_part_names.reserve(parts.size());
|
|
for (const auto & source_part : parts)
|
|
part_log_elem.source_part_names.push_back(source_part->name);
|
|
|
|
part_log_elem.rows_read = (*merge_entry)->bytes_read_uncompressed;
|
|
part_log_elem.bytes_read_uncompressed = (*merge_entry)->bytes_read_uncompressed;
|
|
|
|
part_log_elem.rows = (*merge_entry)->rows_written;
|
|
part_log_elem.bytes_uncompressed = (*merge_entry)->bytes_written_uncompressed;
|
|
|
|
part_log_elem.error = static_cast<UInt16>(execution_status.code);
|
|
part_log_elem.exception = execution_status.message;
|
|
|
|
part_log->add(part_log_elem);
|
|
}
|
|
catch (...)
|
|
{
|
|
tryLogCurrentException(log, __PRETTY_FUNCTION__);
|
|
}
|
|
};
|
|
|
|
try
|
|
{
|
|
part = merger.mergePartsToTemporaryPart(
|
|
future_merged_part, *merge_entry, aio_threshold, entry.create_time, reserved_space.get(), entry.deduplicate);
|
|
|
|
merger.renameMergedTemporaryPart(part, parts, &transaction);
|
|
|
|
try
|
|
{
|
|
checkPartChecksumsAndCommit(transaction, part);
|
|
}
|
|
catch (const Exception & e)
|
|
{
|
|
if (MergeTreeDataPartChecksums::isBadChecksumsErrorCode(e.code()))
|
|
{
|
|
do_fetch = true;
|
|
transaction.rollback();
|
|
|
|
ProfileEvents::increment(ProfileEvents::DataAfterMergeDiffersFromReplica);
|
|
|
|
LOG_ERROR(log, getCurrentExceptionMessage(false) << ". "
|
|
"Data after merge is not byte-identical to data on another replicas. "
|
|
"There could be several reasons: "
|
|
"1. Using newer version of compression library after server update. "
|
|
"2. Using another compression method. "
|
|
"3. Non-deterministic compression algorithm (highly unlikely). "
|
|
"4. Non-deterministic merge algorithm due to logical error in code. "
|
|
"5. Data corruption in memory due to bug in code. "
|
|
"6. Data corruption in memory due to hardware issue. "
|
|
"7. Manual modification of source data after server startup. "
|
|
"8. Manual modification of checksums stored in ZooKeeper. "
|
|
"We will download merged part from replica to force byte-identical result.");
|
|
|
|
write_part_log(ExecutionStatus::fromCurrentException());
|
|
return;
|
|
}
|
|
|
|
throw;
|
|
}
|
|
|
|
/** Removing old chunks from ZK and from the disk is delayed - see ReplicatedMergeTreeCleanupThread, clearOldParts.
|
|
*/
|
|
|
|
/** With `ZCONNECTIONLOSS` or `ZOPERATIONTIMEOUT`, we can inadvertently roll back local changes to the parts.
|
|
* This is not a problem, because in this case the merge will remain in the queue, and we will try again.
|
|
*/
|
|
merge_selecting_event.set();
|
|
ProfileEvents::increment(ProfileEvents::ReplicatedPartMerges);
|
|
|
|
write_part_log({});
|
|
}
|
|
catch (...)
|
|
{
|
|
write_part_log(ExecutionStatus::fromCurrentException());
|
|
throw;
|
|
}
|
|
}
|
|
|
|
|
|
bool StorageReplicatedMergeTree::executeFetch(const StorageReplicatedMergeTree::LogEntry & entry)
|
|
{
|
|
String replica = findReplicaHavingCoveringPart(entry, true);
|
|
|
|
static std::atomic_uint total_fetches {0};
|
|
if (data.settings.replicated_max_parallel_fetches && total_fetches >= data.settings.replicated_max_parallel_fetches)
|
|
{
|
|
throw Exception("Too many total fetches from replicas, maximum: " + data.settings.replicated_max_parallel_fetches.toString(),
|
|
ErrorCodes::TOO_MANY_FETCHES);
|
|
}
|
|
|
|
++total_fetches;
|
|
SCOPE_EXIT({--total_fetches;});
|
|
|
|
if (data.settings.replicated_max_parallel_fetches_for_table && current_table_fetches >= data.settings.replicated_max_parallel_fetches_for_table)
|
|
{
|
|
throw Exception("Too many fetches from replicas for table, maximum: " + data.settings.replicated_max_parallel_fetches_for_table.toString(),
|
|
ErrorCodes::TOO_MANY_FETCHES);
|
|
}
|
|
|
|
++current_table_fetches;
|
|
SCOPE_EXIT({--current_table_fetches;});
|
|
|
|
try
|
|
{
|
|
if (replica.empty())
|
|
{
|
|
/** If a part is to be written with a quorum and the quorum is not reached yet,
|
|
* then (due to the fact that a part is impossible to download right now),
|
|
* the quorum entry should be considered unsuccessful.
|
|
* TODO Complex code, extract separately.
|
|
*/
|
|
if (entry.quorum)
|
|
{
|
|
if (entry.type != LogEntry::GET_PART)
|
|
throw Exception("Logical error: log entry with quorum but type is not GET_PART", ErrorCodes::LOGICAL_ERROR);
|
|
|
|
LOG_DEBUG(log, "No active replica has part " << entry.new_part_name << " which needs to be written with quorum."
|
|
" Will try to mark that quorum as failed.");
|
|
|
|
/** Atomically:
|
|
* - if replicas do not become active;
|
|
* - if there is a `quorum` node with this part;
|
|
* - delete `quorum` node;
|
|
* - set `nonincrement_block_numbers` to resolve merges through the number of the lost part;
|
|
* - add a part to the list `quorum/failed_parts`;
|
|
* - if the part is not already removed from the list for deduplication `blocks/block_num`, then delete it;
|
|
*
|
|
* If something changes, then we will nothing - we'll get here again next time.
|
|
*/
|
|
|
|
/** We collect the `host` node versions from the replicas.
|
|
* When the replica becomes active, it changes the value of host in the same transaction (with the creation of `is_active`).
|
|
* This will ensure that the replicas do not become active.
|
|
*/
|
|
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
|
|
|
|
zkutil::Requests ops;
|
|
|
|
for (size_t i = 0, size = replicas.size(); i < size; ++i)
|
|
{
|
|
zkutil::Stat stat;
|
|
String path = zookeeper_path + "/replicas/" + replicas[i] + "/host";
|
|
zookeeper->get(path, &stat);
|
|
ops.emplace_back(zkutil::makeCheckRequest(path, stat.version));
|
|
}
|
|
|
|
/// We verify that while we were collecting versions, the replica with the necessary part did not come alive.
|
|
replica = findReplicaHavingPart(entry.new_part_name, true);
|
|
|
|
/// Also during this time a completely new replica could be created.
|
|
/// But if a part does not appear on the old, then it can not be on the new one either.
|
|
|
|
if (replica.empty())
|
|
{
|
|
zkutil::Stat quorum_stat;
|
|
String quorum_path = zookeeper_path + "/quorum/status";
|
|
String quorum_str = zookeeper->get(quorum_path, &quorum_stat);
|
|
ReplicatedMergeTreeQuorumEntry quorum_entry;
|
|
quorum_entry.fromString(quorum_str);
|
|
|
|
if (quorum_entry.part_name == entry.new_part_name)
|
|
{
|
|
ops.emplace_back(zkutil::makeRemoveRequest(quorum_path, quorum_stat.version));
|
|
|
|
auto part_info = MergeTreePartInfo::fromPartName(entry.new_part_name, data.format_version);
|
|
|
|
if (part_info.min_block != part_info.max_block)
|
|
throw Exception("Logical error: log entry with quorum for part covering more than one block number",
|
|
ErrorCodes::LOGICAL_ERROR);
|
|
|
|
zookeeper->createIfNotExists(zookeeper_path + "/nonincrement_block_numbers/" + part_info.partition_id, "");
|
|
|
|
ops.emplace_back(zkutil::makeCreateRequest(
|
|
zookeeper_path + "/nonincrement_block_numbers/" + part_info.partition_id + "/block-" + padIndex(part_info.min_block),
|
|
"",
|
|
zkutil::CreateMode::Persistent));
|
|
|
|
ops.emplace_back(zkutil::makeCreateRequest(
|
|
zookeeper_path + "/quorum/failed_parts/" + entry.new_part_name,
|
|
"",
|
|
zkutil::CreateMode::Persistent));
|
|
|
|
/// Deleting from `blocks`.
|
|
if (!entry.block_id.empty() && zookeeper->exists(zookeeper_path + "/blocks/" + entry.block_id))
|
|
ops.emplace_back(zkutil::makeRemoveRequest(zookeeper_path + "/blocks/" + entry.block_id, -1));
|
|
|
|
zkutil::Responses responses;
|
|
auto code = zookeeper->tryMulti(ops, responses);
|
|
|
|
if (code == ZooKeeperImpl::ZooKeeper::ZOK)
|
|
{
|
|
LOG_DEBUG(log, "Marked quorum for part " << entry.new_part_name << " as failed.");
|
|
return true; /// NOTE Deletion from `virtual_parts` is not done, but it is only necessary for merges.
|
|
}
|
|
else if (code == ZooKeeperImpl::ZooKeeper::ZBADVERSION || code == ZooKeeperImpl::ZooKeeper::ZNONODE || code == ZooKeeperImpl::ZooKeeper::ZNODEEXISTS)
|
|
{
|
|
LOG_DEBUG(log, "State was changed or isn't expected when trying to mark quorum for part "
|
|
<< entry.new_part_name << " as failed. Code: " << zkutil::ZooKeeper::error2string(code));
|
|
}
|
|
else
|
|
throw zkutil::KeeperException(code);
|
|
}
|
|
else
|
|
{
|
|
LOG_WARNING(log, "No active replica has part " << entry.new_part_name
|
|
<< ", but that part needs quorum and /quorum/status contains entry about another part " << quorum_entry.part_name
|
|
<< ". It means that part was successfully written to " << entry.quorum
|
|
<< " replicas, but then all of them goes offline."
|
|
<< " Or it is a bug.");
|
|
}
|
|
}
|
|
}
|
|
|
|
if (replica.empty())
|
|
{
|
|
ProfileEvents::increment(ProfileEvents::ReplicatedPartFailedFetches);
|
|
throw Exception("No active replica has part " + entry.new_part_name + " or covering part", ErrorCodes::NO_REPLICA_HAS_PART);
|
|
}
|
|
}
|
|
|
|
try
|
|
{
|
|
if (!fetchPart(entry.actual_new_part_name, zookeeper_path + "/replicas/" + replica, false, entry.quorum))
|
|
return false;
|
|
}
|
|
catch (Exception & e)
|
|
{
|
|
/// No stacktrace, just log message
|
|
if (e.code() == ErrorCodes::RECEIVED_ERROR_TOO_MANY_REQUESTS)
|
|
e.addMessage("Too busy replica. Will try later.");
|
|
throw;
|
|
}
|
|
|
|
if (entry.type == LogEntry::MERGE_PARTS)
|
|
ProfileEvents::increment(ProfileEvents::ReplicatedPartFetchesOfMerged);
|
|
}
|
|
catch (...)
|
|
{
|
|
/** If you can not download the part you need for some merge, it's better not to try to get other parts for this merge,
|
|
* but try to get already merged part. To do this, move the action to get the remaining parts
|
|
* for this merge at the end of the queue.
|
|
*/
|
|
try
|
|
{
|
|
auto parts_for_merge = queue.moveSiblingPartsForMergeToEndOfQueue(entry.new_part_name);
|
|
|
|
if (!parts_for_merge.empty() && replica.empty())
|
|
{
|
|
LOG_INFO(log, "No active replica has part " << entry.new_part_name << ". Will fetch merged part instead.");
|
|
return false;
|
|
}
|
|
|
|
/** If no active replica has a part, and there is no merge in the queue with its participation,
|
|
* check to see if any (active or inactive) replica has such a part or covering it.
|
|
*/
|
|
if (replica.empty())
|
|
enqueuePartForCheck(entry.new_part_name);
|
|
}
|
|
catch (...)
|
|
{
|
|
tryLogCurrentException(__PRETTY_FUNCTION__);
|
|
}
|
|
|
|
throw;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::executeDropRange(const StorageReplicatedMergeTree::LogEntry & entry)
|
|
{
|
|
LOG_INFO(log, (entry.detach ? "Detaching" : "Removing") << " parts inside " << entry.new_part_name << ".");
|
|
|
|
queue.removeGetsAndMergesInRange(getZooKeeper(), entry.new_part_name);
|
|
|
|
LOG_DEBUG(log, (entry.detach ? "Detaching" : "Removing") << " parts.");
|
|
size_t removed_parts = 0;
|
|
|
|
auto entry_part_info = MergeTreePartInfo::fromPartName(entry.new_part_name, data.format_version);
|
|
|
|
/// Delete the parts contained in the range to be deleted.
|
|
/// It's important that no old parts remain (after the merge), because otherwise,
|
|
/// after adding a new replica, this new replica downloads them, but does not delete them.
|
|
/// And, if you do not, the parts will come to life after the server is restarted.
|
|
/// Therefore, we use all data parts.
|
|
auto parts = data.getDataParts({MergeTreeDataPartState::PreCommitted, MergeTreeDataPartState::Committed, MergeTreeDataPartState::Outdated});
|
|
|
|
for (const auto & part : parts)
|
|
{
|
|
if (!entry_part_info.contains(part->info))
|
|
continue;
|
|
|
|
LOG_DEBUG(log, "Removing part " << part->name);
|
|
++removed_parts;
|
|
|
|
/// If you do not need to delete a part, it's more reliable to move the directory before making changes to ZooKeeper.
|
|
if (entry.detach)
|
|
data.renameAndDetachPart(part);
|
|
|
|
zkutil::Requests ops;
|
|
zkutil::Responses responses;
|
|
removePartFromZooKeeper(part->name, ops);
|
|
auto code = getZooKeeper()->tryMulti(ops, responses);
|
|
|
|
/// If the part is already removed (for example, because it was never added to ZK due to crash,
|
|
/// see ReplicatedMergeTreeBlockOutputStream), then Ok.
|
|
if (code && code != ZooKeeperImpl::ZooKeeper::ZNONODE)
|
|
throw zkutil::KeeperException(code);
|
|
|
|
/// If the part needs to be removed, it is more reliable to delete the directory after the changes in ZooKeeper.
|
|
if (!entry.detach)
|
|
data.removePartsFromWorkingSet({part}, true);
|
|
}
|
|
|
|
LOG_INFO(log, (entry.detach ? "Detached " : "Removed ") << removed_parts << " parts inside " << entry.new_part_name << ".");
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::executeClearColumnInPartition(const LogEntry & entry)
|
|
{
|
|
LOG_INFO(log, "Clear column " << entry.column_name << " in parts inside " << entry.new_part_name << " range");
|
|
|
|
/// Assume optimistic scenario, i.e. conflicts are very rare
|
|
/// So, if conflicts are found, throw an exception and will retry execution later
|
|
queue.disableMergesAndFetchesInRange(entry);
|
|
|
|
auto entry_part_info = MergeTreePartInfo::fromPartName(entry.new_part_name, data.format_version);
|
|
|
|
/// We don't change table structure, only data in some parts
|
|
/// To disable reading from these parts, we will sequentially acquire write lock for each part inside alterDataPart()
|
|
/// If we will lock the whole table here, a deadlock can occur. For example, if use use Buffer table (CLICKHOUSE-3238)
|
|
auto lock_read_structure = lockStructure(false, __PRETTY_FUNCTION__);
|
|
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
AlterCommand alter_command;
|
|
alter_command.type = AlterCommand::DROP_COLUMN;
|
|
alter_command.column_name = entry.column_name;
|
|
|
|
auto new_columns = getColumns();
|
|
alter_command.apply(new_columns);
|
|
|
|
size_t modified_parts = 0;
|
|
auto parts = data.getDataParts();
|
|
auto columns_for_parts = new_columns.getAllPhysical();
|
|
for (const auto & part : parts)
|
|
{
|
|
if (!entry_part_info.contains(part->info))
|
|
continue;
|
|
|
|
LOG_DEBUG(log, "Clearing column " << entry.column_name << " in part " << part->name);
|
|
|
|
auto transaction = data.alterDataPart(part, columns_for_parts, data.primary_expr_ast, false);
|
|
if (!transaction)
|
|
continue;
|
|
|
|
/// Update part metadata in ZooKeeper.
|
|
zkutil::Requests ops;
|
|
ops.emplace_back(zkutil::makeSetRequest(
|
|
replica_path + "/parts/" + part->name + "/columns", transaction->getNewColumns().toString(), -1));
|
|
ops.emplace_back(zkutil::makeSetRequest(
|
|
replica_path + "/parts/" + part->name + "/checksums", getChecksumsForZooKeeper(transaction->getNewChecksums()), -1));
|
|
|
|
zookeeper->multi(ops);
|
|
|
|
transaction->commit();
|
|
++modified_parts;
|
|
}
|
|
|
|
LOG_DEBUG(log, "Cleared column " << entry.column_name << " in " << modified_parts << " parts");
|
|
|
|
/// Recalculate columns size (not only for the modified column)
|
|
data.recalculateColumnSizes();
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::queueUpdatingThread()
|
|
{
|
|
setThreadName("ReplMTQueueUpd");
|
|
|
|
bool update_in_progress = false;
|
|
while (!shutdown_called)
|
|
{
|
|
if (!update_in_progress)
|
|
{
|
|
last_queue_update_start_time.store(time(nullptr));
|
|
update_in_progress = true;
|
|
}
|
|
try
|
|
{
|
|
pullLogsToQueue(queue_updating_event);
|
|
last_queue_update_finish_time.store(time(nullptr));
|
|
update_in_progress = false;
|
|
queue_updating_event->wait();
|
|
}
|
|
catch (...)
|
|
{
|
|
tryLogCurrentException(__PRETTY_FUNCTION__);
|
|
queue_updating_event->tryWait(QUEUE_UPDATE_ERROR_SLEEP_MS);
|
|
}
|
|
}
|
|
|
|
LOG_DEBUG(log, "Queue updating thread finished");
|
|
}
|
|
|
|
|
|
bool StorageReplicatedMergeTree::queueTask()
|
|
{
|
|
/// This object will mark the element of the queue as running.
|
|
ReplicatedMergeTreeQueue::SelectedEntry selected;
|
|
|
|
try
|
|
{
|
|
selected = queue.selectEntryToProcess(merger, data);
|
|
}
|
|
catch (...)
|
|
{
|
|
tryLogCurrentException(__PRETTY_FUNCTION__);
|
|
}
|
|
|
|
LogEntryPtr & entry = selected.first;
|
|
|
|
if (!entry)
|
|
return false;
|
|
|
|
time_t prev_attempt_time = entry->last_attempt_time;
|
|
|
|
bool res = queue.processEntry([this]{ return getZooKeeper(); }, entry, [&](LogEntryPtr & entry)
|
|
{
|
|
try
|
|
{
|
|
return executeLogEntry(*entry);
|
|
}
|
|
catch (const Exception & e)
|
|
{
|
|
if (e.code() == ErrorCodes::NO_REPLICA_HAS_PART)
|
|
{
|
|
/// If no one has the right part, probably not all replicas work; We will not write to log with Error level.
|
|
LOG_INFO(log, e.displayText());
|
|
}
|
|
else if (e.code() == ErrorCodes::ABORTED)
|
|
{
|
|
/// Interrupted merge or downloading a part is not an error.
|
|
LOG_INFO(log, e.message());
|
|
}
|
|
else if (e.code() == ErrorCodes::PART_IS_TEMPORARILY_LOCKED)
|
|
{
|
|
/// Part cannot be added temporarily
|
|
LOG_INFO(log, e.displayText());
|
|
}
|
|
else
|
|
tryLogCurrentException(__PRETTY_FUNCTION__);
|
|
|
|
/** This exception will be written to the queue element, and it can be looked up using `system.replication_queue` table.
|
|
* The thread that performs this action will sleep a few seconds after the exception.
|
|
* See `queue.processEntry` function.
|
|
*/
|
|
throw;
|
|
}
|
|
catch (...)
|
|
{
|
|
tryLogCurrentException(__PRETTY_FUNCTION__);
|
|
throw;
|
|
}
|
|
});
|
|
|
|
/// We will go to sleep if the processing fails and if we have already processed this record recently.
|
|
bool need_sleep = !res && (entry->last_attempt_time - prev_attempt_time < 10);
|
|
|
|
/// If there was no exception, you do not need to sleep.
|
|
return !need_sleep;
|
|
}
|
|
|
|
|
|
namespace
|
|
{
|
|
bool canMergePartsAccordingToZooKeeperInfo(
|
|
const MergeTreeData::DataPartPtr & left,
|
|
const MergeTreeData::DataPartPtr & right,
|
|
zkutil::ZooKeeperPtr && zookeeper, const String & zookeeper_path, const MergeTreeData & data, String * out_reason = nullptr)
|
|
{
|
|
const String & partition_id = left->info.partition_id;
|
|
|
|
/// You can not merge parts, among which is a part for which the quorum is unsatisfied.
|
|
/// Note: theoretically, this could be resolved. But this will make logic more complex.
|
|
String quorum_node_value;
|
|
if (zookeeper->tryGet(zookeeper_path + "/quorum/status", quorum_node_value))
|
|
{
|
|
ReplicatedMergeTreeQuorumEntry quorum_entry;
|
|
quorum_entry.fromString(quorum_node_value);
|
|
|
|
auto part_info = MergeTreePartInfo::fromPartName(quorum_entry.part_name, data.format_version);
|
|
|
|
if (part_info.min_block != part_info.max_block)
|
|
throw Exception("Logical error: part written with quorum covers more than one block numbers", ErrorCodes::LOGICAL_ERROR);
|
|
|
|
if (left->info.max_block <= part_info.min_block && right->info.min_block >= part_info.max_block)
|
|
{
|
|
if (out_reason)
|
|
*out_reason = "Quorum status condition is unsatisfied";
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/// Won't merge last_part even if quorum is satisfied, because we gonna check if replica has this part
|
|
/// on SELECT execution.
|
|
String quorum_last_part;
|
|
if (zookeeper->tryGet(zookeeper_path + "/quorum/last_part", quorum_last_part) && quorum_last_part.empty() == false)
|
|
{
|
|
auto part_info = MergeTreePartInfo::fromPartName(quorum_last_part, data.format_version);
|
|
|
|
if (part_info.min_block != part_info.max_block)
|
|
throw Exception("Logical error: part written with quorum covers more than one block numbers", ErrorCodes::LOGICAL_ERROR);
|
|
|
|
if (left->info.max_block <= part_info.min_block && right->info.min_block >= part_info.max_block)
|
|
{
|
|
if (out_reason)
|
|
*out_reason = "Quorum 'last part' condition is unsatisfied";
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/// You can merge the parts, if all the numbers between them are abandoned - do not correspond to any blocks.
|
|
for (Int64 number = left->info.max_block + 1; number <= right->info.min_block - 1; ++number)
|
|
{
|
|
String path1 = zookeeper_path + "/block_numbers/" + partition_id + "/block-" + padIndex(number);
|
|
String path2 = zookeeper_path + "/nonincrement_block_numbers/" + partition_id + "/block-" + padIndex(number);
|
|
|
|
if (AbandonableLockInZooKeeper::check(path1, *zookeeper) != AbandonableLockInZooKeeper::ABANDONED &&
|
|
AbandonableLockInZooKeeper::check(path2, *zookeeper) != AbandonableLockInZooKeeper::ABANDONED)
|
|
{
|
|
if (out_reason)
|
|
*out_reason = "Block " + toString(number) + " in gap between merging parts " + left->name + " and "
|
|
+ right->name + " is not abandoned";
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
/// If any of the parts is already going to be merged into a larger one, do not agree to merge it.
|
|
bool partsWillNotBeMergedOrDisabled(const MergeTreeData::DataPartPtr & left, const MergeTreeData::DataPartPtr & right,
|
|
ReplicatedMergeTreeQueue & queue, String * out_reason = nullptr)
|
|
{
|
|
auto set_reason = [&out_reason] (const String & part_name)
|
|
{
|
|
if (out_reason)
|
|
*out_reason = "Part " + part_name + " cannot be merged yet, a merge has already assigned for it or it is temporarily disabled";
|
|
return false;
|
|
};
|
|
|
|
if (queue.partWillBeMergedOrMergesDisabled(left->name))
|
|
return set_reason(left->name);
|
|
|
|
if (left.get() != right.get() && queue.partWillBeMergedOrMergesDisabled(right->name))
|
|
return set_reason(right->name);
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
/** It can take a long time to determine whether it is possible to merge two adjacent parts.
|
|
* Two adjacent parts can be merged if all block numbers between their numbers are not used (abandoned).
|
|
* This means that another part can not be inserted between these parts.
|
|
*
|
|
* But if the numbers of adjacent blocks differ much (usually if there are many "abandoned" blocks between them),
|
|
* then too many read requests are made to ZooKeeper to find out if it's possible to merge them.
|
|
*
|
|
* Let's use a statement that if a couple of parts were possible to merge, and their merge is not yet planned,
|
|
* then now they can be merged, and we will remember this state,
|
|
* not to send multiple identical requests to ZooKeeper.
|
|
*/
|
|
|
|
/** Cache for function, that returns bool.
|
|
* If function returned true, cache it forever.
|
|
* If function returned false, cache it for exponentially growing time.
|
|
* Not thread safe.
|
|
*/
|
|
template <typename Key>
|
|
struct CachedMergingPredicate
|
|
{
|
|
using clock = std::chrono::steady_clock;
|
|
|
|
struct Expiration
|
|
{
|
|
static constexpr clock::duration min_delay = std::chrono::seconds(1);
|
|
static constexpr clock::duration max_delay = std::chrono::seconds(600);
|
|
static constexpr double exponent_base = 2;
|
|
|
|
clock::time_point expire_time;
|
|
clock::duration delay = clock::duration::zero();
|
|
|
|
void next(clock::time_point now)
|
|
{
|
|
if (delay == clock::duration::zero())
|
|
delay = min_delay;
|
|
else
|
|
{
|
|
delay *= exponent_base;
|
|
if (delay > max_delay)
|
|
delay = max_delay;
|
|
}
|
|
|
|
expire_time = now + delay;
|
|
}
|
|
|
|
bool expired(clock::time_point now) const
|
|
{
|
|
return now > expire_time;
|
|
}
|
|
};
|
|
|
|
std::set<Key> true_keys;
|
|
std::map<Key, Expiration> false_keys;
|
|
|
|
template <typename Function, typename ArgsToKey, typename... Args>
|
|
bool get(clock::time_point now, Function && function, ArgsToKey && args_to_key, Args &&... args)
|
|
{
|
|
Key key{args_to_key(std::forward<Args>(args)...)};
|
|
|
|
if (true_keys.count(key))
|
|
return true;
|
|
|
|
auto it = false_keys.find(key);
|
|
if (false_keys.end() != it && !it->second.expired(now))
|
|
return false;
|
|
|
|
bool value = function(std::forward<Args>(args)...);
|
|
|
|
if (value)
|
|
true_keys.insert(key);
|
|
else
|
|
false_keys[key].next(now);
|
|
|
|
return value;
|
|
}
|
|
};
|
|
|
|
template <typename Key> constexpr CachedMergingPredicate<Key>::clock::duration CachedMergingPredicate<Key>::Expiration::min_delay;
|
|
template <typename Key> constexpr CachedMergingPredicate<Key>::clock::duration CachedMergingPredicate<Key>::Expiration::max_delay;
|
|
template <typename Key> constexpr double CachedMergingPredicate<Key>::Expiration::exponent_base;
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::mergeSelectingThread()
|
|
{
|
|
setThreadName("ReplMTMergeSel");
|
|
LOG_DEBUG(log, "Merge selecting thread started");
|
|
|
|
bool deduplicate = false; /// TODO: read deduplicate option from table config
|
|
|
|
auto uncached_merging_predicate = [this](const MergeTreeData::DataPartPtr & left, const MergeTreeData::DataPartPtr & right)
|
|
{
|
|
return canMergePartsAccordingToZooKeeperInfo(left, right, getZooKeeper(), zookeeper_path, data);
|
|
};
|
|
|
|
auto merging_predicate_args_to_key = [](const MergeTreeData::DataPartPtr & left, const MergeTreeData::DataPartPtr & right)
|
|
{
|
|
return std::make_pair(left->name, right->name);
|
|
};
|
|
|
|
CachedMergingPredicate<std::pair<std::string, std::string>> cached_merging_predicate;
|
|
|
|
/// Will be updated below.
|
|
std::chrono::steady_clock::time_point now;
|
|
|
|
auto can_merge = [&] (const MergeTreeData::DataPartPtr & left, const MergeTreeData::DataPartPtr & right, String *)
|
|
{
|
|
return partsWillNotBeMergedOrDisabled(left, right, queue)
|
|
&& cached_merging_predicate.get(now, uncached_merging_predicate, merging_predicate_args_to_key, left, right);
|
|
};
|
|
|
|
while (!shutdown_called && is_leader_node)
|
|
{
|
|
bool success = false;
|
|
|
|
try
|
|
{
|
|
std::lock_guard<std::mutex> merge_selecting_lock(merge_selecting_mutex);
|
|
|
|
/// You need to load new entries into the queue before you select parts to merge.
|
|
/// (so we know which parts are already going to be merged).
|
|
/// We must select parts for merge under the mutex because other threads (OPTIMIZE queries) could push new merges.
|
|
if (merge_selecting_logs_pulling_is_required)
|
|
{
|
|
pullLogsToQueue();
|
|
merge_selecting_logs_pulling_is_required = false;
|
|
}
|
|
|
|
/// If many merges is already queued, then will queue only small enough merges.
|
|
/// Otherwise merge queue could be filled with only large merges,
|
|
/// and in the same time, many small parts could be created and won't be merged.
|
|
size_t merges_queued = queue.countMerges();
|
|
|
|
if (merges_queued >= data.settings.max_replicated_merges_in_queue)
|
|
{
|
|
LOG_TRACE(log, "Number of queued merges (" << merges_queued
|
|
<< ") is greater than max_replicated_merges_in_queue ("
|
|
<< data.settings.max_replicated_merges_in_queue << "), so won't select new parts to merge.");
|
|
}
|
|
else
|
|
{
|
|
MergeTreeDataMerger::FuturePart future_merged_part;
|
|
|
|
size_t max_parts_size_for_merge = merger.getMaxPartsSizeForMerge(data.settings.max_replicated_merges_in_queue, merges_queued);
|
|
|
|
now = std::chrono::steady_clock::now();
|
|
|
|
if (max_parts_size_for_merge > 0
|
|
&& merger.selectPartsToMerge(future_merged_part, false, max_parts_size_for_merge, can_merge))
|
|
{
|
|
merge_selecting_logs_pulling_is_required = true;
|
|
success = createLogEntryToMergeParts(future_merged_part.parts, future_merged_part.name, deduplicate);
|
|
}
|
|
}
|
|
}
|
|
catch (...)
|
|
{
|
|
tryLogCurrentException(__PRETTY_FUNCTION__);
|
|
}
|
|
|
|
if (shutdown_called || !is_leader_node)
|
|
break;
|
|
|
|
if (!success)
|
|
merge_selecting_event.tryWait(MERGE_SELECTING_SLEEP_MS);
|
|
}
|
|
|
|
LOG_DEBUG(log, "Merge selecting thread finished");
|
|
}
|
|
|
|
|
|
bool StorageReplicatedMergeTree::createLogEntryToMergeParts(
|
|
const MergeTreeData::DataPartsVector & parts, const String & merged_name, bool deduplicate, ReplicatedMergeTreeLogEntryData * out_log_entry)
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
bool all_in_zk = true;
|
|
for (const auto & part : parts)
|
|
{
|
|
/// If there is no information about part in ZK, we will not merge it.
|
|
if (!zookeeper->exists(replica_path + "/parts/" + part->name))
|
|
{
|
|
all_in_zk = false;
|
|
|
|
if (part->modification_time + MAX_AGE_OF_LOCAL_PART_THAT_WASNT_ADDED_TO_ZOOKEEPER < time(nullptr))
|
|
{
|
|
LOG_WARNING(log, "Part " << part->name << " (that was selected for merge)"
|
|
<< " with age " << (time(nullptr) - part->modification_time)
|
|
<< " seconds exists locally but not in ZooKeeper."
|
|
<< " Won't do merge with that part and will check it.");
|
|
enqueuePartForCheck(part->name);
|
|
}
|
|
}
|
|
}
|
|
if (!all_in_zk)
|
|
return false;
|
|
|
|
ReplicatedMergeTreeLogEntryData entry;
|
|
entry.type = LogEntry::MERGE_PARTS;
|
|
entry.source_replica = replica_name;
|
|
entry.new_part_name = merged_name;
|
|
entry.deduplicate = deduplicate;
|
|
entry.create_time = time(nullptr);
|
|
|
|
for (const auto & part : parts)
|
|
entry.parts_to_merge.push_back(part->name);
|
|
|
|
String path_created = zookeeper->create(zookeeper_path + "/log/log-", entry.toString(), zkutil::CreateMode::PersistentSequential);
|
|
entry.znode_name = path_created.substr(path_created.find_last_of('/') + 1);
|
|
|
|
const String & partition_id = parts[0]->info.partition_id;
|
|
for (size_t i = 0; i + 1 < parts.size(); ++i)
|
|
{
|
|
/// Remove the unnecessary entries about non-existent blocks.
|
|
for (Int64 number = parts[i]->info.max_block + 1; number <= parts[i + 1]->info.min_block - 1; ++number)
|
|
{
|
|
zookeeper->tryRemove(zookeeper_path + "/block_numbers/" + partition_id + "/block-" + padIndex(number));
|
|
zookeeper->tryRemove(zookeeper_path + "/nonincrement_block_numbers/" + partition_id + "/block-" + padIndex(number));
|
|
}
|
|
}
|
|
|
|
if (out_log_entry)
|
|
*out_log_entry = entry;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::removePartFromZooKeeper(const String & part_name, zkutil::Requests & ops)
|
|
{
|
|
String part_path = replica_path + "/parts/" + part_name;
|
|
|
|
ops.emplace_back(zkutil::makeRemoveRequest(part_path + "/checksums", -1));
|
|
ops.emplace_back(zkutil::makeRemoveRequest(part_path + "/columns", -1));
|
|
ops.emplace_back(zkutil::makeRemoveRequest(part_path, -1));
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::removePartAndEnqueueFetch(const String & part_name)
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
String part_path = replica_path + "/parts/" + part_name;
|
|
|
|
LogEntryPtr log_entry = std::make_shared<LogEntry>();
|
|
log_entry->type = LogEntry::GET_PART;
|
|
log_entry->create_time = tryGetPartCreateTime(zookeeper, replica_path, part_name);
|
|
log_entry->source_replica = "";
|
|
log_entry->new_part_name = part_name;
|
|
|
|
zkutil::Requests ops;
|
|
ops.emplace_back(zkutil::makeCreateRequest(
|
|
replica_path + "/queue/queue-", log_entry->toString(),
|
|
zkutil::CreateMode::PersistentSequential));
|
|
|
|
removePartFromZooKeeper(part_name, ops);
|
|
|
|
auto results = zookeeper->multi(ops);
|
|
|
|
String path_created = dynamic_cast<const zkutil::CreateResponse &>(*results[0]).path_created;
|
|
log_entry->znode_name = path_created.substr(path_created.find_last_of('/') + 1);
|
|
queue.insert(zookeeper, log_entry);
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::becomeLeader()
|
|
{
|
|
std::lock_guard<std::mutex> lock(leader_node_mutex);
|
|
|
|
if (shutdown_called)
|
|
return;
|
|
|
|
if (merge_selecting_thread.joinable())
|
|
{
|
|
LOG_INFO(log, "Deleting old leader");
|
|
is_leader_node = false; /// exit trigger inside thread
|
|
merge_selecting_thread.join();
|
|
}
|
|
|
|
LOG_INFO(log, "Became leader");
|
|
is_leader_node = true;
|
|
merge_selecting_thread = std::thread(&StorageReplicatedMergeTree::mergeSelectingThread, this);
|
|
}
|
|
|
|
|
|
String StorageReplicatedMergeTree::findReplicaHavingPart(const String & part_name, bool active)
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
|
|
|
|
/// Select replicas in uniformly random order.
|
|
std::shuffle(replicas.begin(), replicas.end(), rng);
|
|
|
|
for (const String & replica : replicas)
|
|
{
|
|
/// We don't interested in ourself.
|
|
if (replica == replica_name)
|
|
continue;
|
|
|
|
if (zookeeper->exists(zookeeper_path + "/replicas/" + replica + "/parts/" + part_name) &&
|
|
(!active || zookeeper->exists(zookeeper_path + "/replicas/" + replica + "/is_active")))
|
|
return replica;
|
|
|
|
/// Obviously, replica could become inactive or even vanish after return from this method.
|
|
}
|
|
|
|
return {};
|
|
}
|
|
|
|
|
|
String StorageReplicatedMergeTree::findReplicaHavingCoveringPart(const LogEntry & entry, bool active)
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
|
|
|
|
/// Select replicas in uniformly random order.
|
|
std::shuffle(replicas.begin(), replicas.end(), rng);
|
|
|
|
for (const String & replica : replicas)
|
|
{
|
|
if (replica == replica_name)
|
|
continue;
|
|
|
|
if (active && !zookeeper->exists(zookeeper_path + "/replicas/" + replica + "/is_active"))
|
|
continue;
|
|
|
|
String largest_part_found;
|
|
Strings parts = zookeeper->getChildren(zookeeper_path + "/replicas/" + replica + "/parts");
|
|
for (const String & part_on_replica : parts)
|
|
{
|
|
if (part_on_replica == entry.new_part_name
|
|
|| MergeTreePartInfo::contains(part_on_replica, entry.new_part_name, data.format_version))
|
|
{
|
|
if (largest_part_found.empty()
|
|
|| MergeTreePartInfo::contains(part_on_replica, largest_part_found, data.format_version))
|
|
{
|
|
largest_part_found = part_on_replica;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!largest_part_found.empty())
|
|
{
|
|
bool the_same_part = largest_part_found == entry.new_part_name;
|
|
|
|
/// Make a check in case if selected part differs from source part
|
|
if (!the_same_part)
|
|
{
|
|
String reject_reason;
|
|
if (!queue.addFuturePartIfNotCoveredByThem(largest_part_found, entry, reject_reason))
|
|
{
|
|
LOG_INFO(log, "Will not fetch part " << largest_part_found << " covering " << entry.new_part_name << ". " << reject_reason);
|
|
return {};
|
|
}
|
|
}
|
|
else
|
|
{
|
|
entry.actual_new_part_name = entry.new_part_name;
|
|
}
|
|
|
|
return replica;
|
|
}
|
|
}
|
|
|
|
return {};
|
|
}
|
|
|
|
|
|
/** If a quorum is tracked for a part, update information about it in ZK.
|
|
*/
|
|
void StorageReplicatedMergeTree::updateQuorum(const String & part_name)
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
/// Information on which replicas a part has been added, if the quorum has not yet been reached.
|
|
const String quorum_status_path = zookeeper_path + "/quorum/status";
|
|
/// The name of the previous part for which the quorum was reached.
|
|
const String quorum_last_part_path = zookeeper_path + "/quorum/last_part";
|
|
|
|
String value;
|
|
zkutil::Stat stat;
|
|
|
|
/// If there is no node, then all quorum INSERTs have already reached the quorum, and nothing is needed.
|
|
while (zookeeper->tryGet(quorum_status_path, value, &stat))
|
|
{
|
|
ReplicatedMergeTreeQuorumEntry quorum_entry;
|
|
quorum_entry.fromString(value);
|
|
|
|
if (quorum_entry.part_name != part_name)
|
|
{
|
|
/// The quorum has already been achieved. Moreover, another INSERT with a quorum has already started.
|
|
break;
|
|
}
|
|
|
|
quorum_entry.replicas.insert(replica_name);
|
|
|
|
if (quorum_entry.replicas.size() >= quorum_entry.required_number_of_replicas)
|
|
{
|
|
/// The quorum is reached. Delete the node, and update information about the last part that was successfully written with quorum.
|
|
|
|
zkutil::Requests ops;
|
|
zkutil::Responses responses;
|
|
ops.emplace_back(zkutil::makeRemoveRequest(quorum_status_path, stat.version));
|
|
ops.emplace_back(zkutil::makeSetRequest(quorum_last_part_path, part_name, -1));
|
|
auto code = zookeeper->tryMulti(ops, responses);
|
|
|
|
if (code == ZooKeeperImpl::ZooKeeper::ZOK)
|
|
{
|
|
break;
|
|
}
|
|
else if (code == ZooKeeperImpl::ZooKeeper::ZNONODE)
|
|
{
|
|
/// The quorum has already been achieved.
|
|
break;
|
|
}
|
|
else if (code == ZooKeeperImpl::ZooKeeper::ZBADVERSION)
|
|
{
|
|
/// Node was updated meanwhile. We must re-read it and repeat all the actions.
|
|
continue;
|
|
}
|
|
else
|
|
throw zkutil::KeeperException(code, quorum_status_path);
|
|
}
|
|
else
|
|
{
|
|
/// We update the node, registering there one more replica.
|
|
auto code = zookeeper->trySet(quorum_status_path, quorum_entry.toString(), stat.version);
|
|
|
|
if (code == ZooKeeperImpl::ZooKeeper::ZOK)
|
|
{
|
|
break;
|
|
}
|
|
else if (code == ZooKeeperImpl::ZooKeeper::ZNONODE)
|
|
{
|
|
/// The quorum has already been achieved.
|
|
break;
|
|
}
|
|
else if (code == ZooKeeperImpl::ZooKeeper::ZBADVERSION)
|
|
{
|
|
/// Node was updated meanwhile. We must re-read it and repeat all the actions.
|
|
continue;
|
|
}
|
|
else
|
|
throw zkutil::KeeperException(code, quorum_status_path);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
bool StorageReplicatedMergeTree::fetchPart(const String & part_name, const String & replica_path, bool to_detached, size_t quorum)
|
|
{
|
|
if (auto part = data.getPartIfExists(part_name, {MergeTreeDataPart::State::Outdated, MergeTreeDataPart::State::Deleting}))
|
|
{
|
|
LOG_DEBUG(log, "Part " << part->getNameWithState() << " should be deleted after previous attempt before fetch");
|
|
/// Force immediate parts cleanup to delete the part that was left from the previous fetch attempt.
|
|
cleanup_thread_event.set();
|
|
return false;
|
|
}
|
|
|
|
{
|
|
std::lock_guard<std::mutex> lock(currently_fetching_parts_mutex);
|
|
if (!currently_fetching_parts.insert(part_name).second)
|
|
{
|
|
LOG_DEBUG(log, "Part " << part_name << " is already fetching right now");
|
|
return false;
|
|
}
|
|
}
|
|
|
|
SCOPE_EXIT
|
|
({
|
|
std::lock_guard<std::mutex> lock(currently_fetching_parts_mutex);
|
|
currently_fetching_parts.erase(part_name);
|
|
});
|
|
|
|
LOG_DEBUG(log, "Fetching part " << part_name << " from " << replica_path);
|
|
|
|
TableStructureReadLockPtr table_lock;
|
|
if (!to_detached)
|
|
table_lock = lockStructure(true, __PRETTY_FUNCTION__);
|
|
|
|
ReplicatedMergeTreeAddress address(getZooKeeper()->get(replica_path + "/host"));
|
|
auto timeouts = ConnectionTimeouts::getHTTPTimeouts(context.getSettingsRef());
|
|
|
|
/// Logging
|
|
Stopwatch stopwatch;
|
|
MergeTreeData::MutableDataPartPtr part;
|
|
MergeTreeData::DataPartsVector replaced_parts;
|
|
|
|
auto write_part_log = [&] (const ExecutionStatus & execution_status)
|
|
{
|
|
try
|
|
{
|
|
auto part_log = context.getPartLog(database_name);
|
|
if (!part_log)
|
|
return;
|
|
|
|
PartLogElement part_log_elem;
|
|
|
|
part_log_elem.event_time = time(nullptr);
|
|
part_log_elem.event_type = PartLogElement::DOWNLOAD_PART;
|
|
/// TODO: Stop stopwatch in outer code to exclude ZK timings and so on
|
|
part_log_elem.duration_ms = stopwatch.elapsed() / 10000000;
|
|
|
|
part_log_elem.database_name = database_name;
|
|
part_log_elem.table_name = table_name;
|
|
part_log_elem.part_name = part_name;
|
|
|
|
if (part)
|
|
{
|
|
part_log_elem.bytes_compressed_on_disk = part->bytes_on_disk;
|
|
part_log_elem.rows = part->rows_count; /// Could be approximate (?)
|
|
}
|
|
|
|
part_log_elem.source_part_names.reserve(replaced_parts.size());
|
|
for (const auto & replaced_part : replaced_parts)
|
|
part_log_elem.source_part_names.push_back(replaced_part->name);
|
|
|
|
part_log_elem.error = static_cast<UInt16>(execution_status.code);
|
|
part_log_elem.exception = execution_status.message;
|
|
|
|
part_log->add(part_log_elem);
|
|
}
|
|
catch (...)
|
|
{
|
|
tryLogCurrentException(log, __PRETTY_FUNCTION__);
|
|
}
|
|
};
|
|
|
|
try
|
|
{
|
|
part = fetcher.fetchPart(part_name, replica_path, address.host, address.replication_port, timeouts, to_detached);
|
|
|
|
if (!to_detached)
|
|
{
|
|
MergeTreeData::Transaction transaction;
|
|
data.renameTempPartAndReplace(part, nullptr, &transaction);
|
|
|
|
/** NOTE
|
|
* Here, an error occurs if ALTER occurred with a change in the column type or column deletion,
|
|
* and the part on remote server has not yet been modified.
|
|
* After a while, one of the following attempts to make `fetchPart` succeed.
|
|
*/
|
|
replaced_parts = checkPartChecksumsAndCommit(transaction, part);
|
|
|
|
/** If a quorum is tracked for this part, you must update it.
|
|
* If you do not have time, in case of losing the session, when you restart the server - see the `ReplicatedMergeTreeRestartingThread::updateQuorumIfWeHavePart` method.
|
|
*/
|
|
if (quorum)
|
|
updateQuorum(part_name);
|
|
|
|
merge_selecting_event.set();
|
|
|
|
for (const auto & replaced_part : replaced_parts)
|
|
{
|
|
LOG_DEBUG(log, "Part " << replaced_part->name << " is rendered obsolete by fetching part " << part_name);
|
|
ProfileEvents::increment(ProfileEvents::ObsoleteReplicatedParts);
|
|
}
|
|
|
|
write_part_log({});
|
|
}
|
|
else
|
|
{
|
|
part->renameTo("detached/" + part_name);
|
|
}
|
|
}
|
|
catch (...)
|
|
{
|
|
if (!to_detached)
|
|
write_part_log(ExecutionStatus::fromCurrentException());
|
|
|
|
throw;
|
|
}
|
|
|
|
ProfileEvents::increment(ProfileEvents::ReplicatedPartFetches);
|
|
|
|
LOG_DEBUG(log, "Fetched part " << part_name << " from " << replica_path << (to_detached ? " (to 'detached' directory)" : ""));
|
|
return true;
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::startup()
|
|
{
|
|
if (is_readonly)
|
|
return;
|
|
|
|
queue.initialize(
|
|
zookeeper_path, replica_path,
|
|
database_name + "." + table_name + " (ReplicatedMergeTreeQueue)",
|
|
data.getDataParts(), current_zookeeper);
|
|
|
|
queue.pullLogsToQueue(current_zookeeper, nullptr);
|
|
last_queue_update_finish_time.store(time(nullptr));
|
|
/// NOTE: not updating last_queue_update_start_time because it must contain the time when
|
|
/// the notification of queue change was received. In the beginning it is effectively infinite.
|
|
|
|
StoragePtr ptr = shared_from_this();
|
|
InterserverIOEndpointPtr data_parts_exchange_endpoint = std::make_shared<DataPartsExchange::Service>(data, ptr);
|
|
data_parts_exchange_endpoint_holder = std::make_shared<InterserverIOEndpointHolder>(
|
|
data_parts_exchange_endpoint->getId(replica_path), data_parts_exchange_endpoint, context.getInterserverIOHandler());
|
|
|
|
/// In this thread replica will be activated.
|
|
restarting_thread = std::make_unique<ReplicatedMergeTreeRestartingThread>(*this);
|
|
|
|
/// Wait while restarting_thread initializes LeaderElection (and so on) or makes first attmept to do it
|
|
startup_event.wait();
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::shutdown()
|
|
{
|
|
/** This must be done before waiting for restarting_thread.
|
|
* Because restarting_thread will wait for finishing of tasks in background pool,
|
|
* and parts are fetched in that tasks.
|
|
*/
|
|
fetcher.blocker.cancelForever();
|
|
|
|
if (restarting_thread)
|
|
{
|
|
restarting_thread->stop();
|
|
restarting_thread.reset();
|
|
}
|
|
|
|
if (data_parts_exchange_endpoint_holder)
|
|
{
|
|
data_parts_exchange_endpoint_holder->cancelForever();
|
|
data_parts_exchange_endpoint_holder = nullptr;
|
|
}
|
|
}
|
|
|
|
|
|
StorageReplicatedMergeTree::~StorageReplicatedMergeTree()
|
|
{
|
|
try
|
|
{
|
|
shutdown();
|
|
}
|
|
catch(...)
|
|
{
|
|
tryLogCurrentException(__PRETTY_FUNCTION__);
|
|
}
|
|
}
|
|
|
|
|
|
BlockInputStreams StorageReplicatedMergeTree::read(
|
|
const Names & column_names,
|
|
const SelectQueryInfo & query_info,
|
|
const Context & context,
|
|
QueryProcessingStage::Enum & processed_stage,
|
|
const size_t max_block_size,
|
|
const unsigned num_streams)
|
|
{
|
|
const Settings & settings = context.getSettingsRef();
|
|
|
|
/** The `select_sequential_consistency` setting has two meanings:
|
|
* 1. To throw an exception if on a replica there are not all parts which have been written down on quorum of remaining replicas.
|
|
* 2. Do not read parts that have not yet been written to the quorum of the replicas.
|
|
* For this you have to synchronously go to ZooKeeper.
|
|
*/
|
|
Int64 max_block_number_to_read = 0;
|
|
if (settings.select_sequential_consistency)
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
String last_part;
|
|
zookeeper->tryGet(zookeeper_path + "/quorum/last_part", last_part);
|
|
|
|
if (!last_part.empty() && !data.getActiveContainingPart(last_part)) /// TODO Disable replica for distributed queries.
|
|
throw Exception("Replica doesn't have part " + last_part + " which was successfully written to quorum of other replicas."
|
|
" Send query to another replica or disable 'select_sequential_consistency' setting.", ErrorCodes::REPLICA_IS_NOT_IN_QUORUM);
|
|
|
|
if (last_part.empty()) /// If no part has been written with quorum.
|
|
{
|
|
String quorum_str;
|
|
if (zookeeper->tryGet(zookeeper_path + "/quorum/status", quorum_str))
|
|
{
|
|
ReplicatedMergeTreeQuorumEntry quorum_entry;
|
|
quorum_entry.fromString(quorum_str);
|
|
auto part_info = MergeTreePartInfo::fromPartName(quorum_entry.part_name, data.format_version);
|
|
max_block_number_to_read = part_info.min_block - 1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
auto part_info = MergeTreePartInfo::fromPartName(last_part, data.format_version);
|
|
max_block_number_to_read = part_info.max_block;
|
|
}
|
|
}
|
|
|
|
return reader.read(
|
|
column_names, query_info, context, processed_stage, max_block_size, num_streams, max_block_number_to_read);
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::assertNotReadonly() const
|
|
{
|
|
if (is_readonly)
|
|
throw Exception("Table is in readonly mode", ErrorCodes::TABLE_IS_READ_ONLY);
|
|
}
|
|
|
|
|
|
BlockOutputStreamPtr StorageReplicatedMergeTree::write(const ASTPtr & /*query*/, const Settings & settings)
|
|
{
|
|
assertNotReadonly();
|
|
|
|
bool deduplicate = data.settings.replicated_deduplication_window != 0 && settings.insert_deduplicate;
|
|
|
|
return std::make_shared<ReplicatedMergeTreeBlockOutputStream>(*this,
|
|
settings.insert_quorum, settings.insert_quorum_timeout.totalMilliseconds(), deduplicate);
|
|
}
|
|
|
|
|
|
bool StorageReplicatedMergeTree::optimize(const ASTPtr & query, const ASTPtr & partition, bool final, bool deduplicate, const Context & context)
|
|
{
|
|
assertNotReadonly();
|
|
|
|
if (!is_leader_node)
|
|
{
|
|
sendRequestToLeaderReplica(query, context.getSettingsRef());
|
|
return true;
|
|
}
|
|
|
|
auto can_merge = [this] (const MergeTreeData::DataPartPtr & left, const MergeTreeData::DataPartPtr & right, String * out_reason)
|
|
{
|
|
return partsWillNotBeMergedOrDisabled(left, right, queue, out_reason)
|
|
&& canMergePartsAccordingToZooKeeperInfo(left, right, getZooKeeper(), zookeeper_path, data, out_reason);
|
|
};
|
|
|
|
ReplicatedMergeTreeLogEntryData merge_entry;
|
|
{
|
|
std::lock_guard<std::mutex> merge_selecting_lock(merge_selecting_mutex);
|
|
|
|
/// We must select parts for merge under the mutex because other threads (OPTIMIZE queries) could push new merges.
|
|
pullLogsToQueue();
|
|
|
|
size_t disk_space = DiskSpaceMonitor::getUnreservedFreeSpace(full_path);
|
|
|
|
MergeTreeDataMerger::FuturePart future_merged_part;
|
|
String disable_reason;
|
|
bool selected = false;
|
|
|
|
if (!partition)
|
|
{
|
|
selected = merger.selectPartsToMerge(
|
|
future_merged_part, true, data.settings.max_bytes_to_merge_at_max_space_in_pool, can_merge, &disable_reason);
|
|
}
|
|
else
|
|
{
|
|
String partition_id = data.getPartitionIDFromQuery(partition, context);
|
|
selected = merger.selectAllPartsToMergeWithinPartition(future_merged_part, disk_space, can_merge, partition_id, final, &disable_reason);
|
|
}
|
|
|
|
auto handle_noop = [&] (const String & message)
|
|
{
|
|
if (context.getSettingsRef().optimize_throw_if_noop)
|
|
throw Exception(message, ErrorCodes::CANNOT_ASSIGN_OPTIMIZE);
|
|
return false;
|
|
};
|
|
|
|
if (!selected)
|
|
{
|
|
LOG_INFO(log, "Cannot select parts for optimization" + (disable_reason.empty() ? "" : ": " + disable_reason));
|
|
return handle_noop(disable_reason);
|
|
}
|
|
|
|
/// It is important to pull new logs (even if creation of the entry fails due to network error)
|
|
merge_selecting_logs_pulling_is_required = true;
|
|
|
|
if (!createLogEntryToMergeParts(future_merged_part.parts, future_merged_part.name, deduplicate, &merge_entry))
|
|
return handle_noop("Can't create merge queue node in ZooKeeper");
|
|
}
|
|
|
|
waitForAllReplicasToProcessLogEntry(merge_entry);
|
|
return true;
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::alter(const AlterCommands & params,
|
|
const String & /*database_name*/, const String & /*table_name*/, const Context & context)
|
|
{
|
|
assertNotReadonly();
|
|
|
|
LOG_DEBUG(log, "Doing ALTER");
|
|
|
|
int new_columns_version = -1; /// Initialization is to suppress (useless) false positive warning found by cppcheck.
|
|
String new_columns_str;
|
|
zkutil::Stat stat;
|
|
|
|
{
|
|
/// Just to read current structure. Alter will be done in separate thread.
|
|
auto table_lock = lockStructure(false, __PRETTY_FUNCTION__);
|
|
|
|
if (is_readonly)
|
|
throw Exception("Can't ALTER readonly table", ErrorCodes::TABLE_IS_READ_ONLY);
|
|
|
|
data.checkAlter(params);
|
|
|
|
for (const AlterCommand & param : params)
|
|
if (param.type == AlterCommand::MODIFY_PRIMARY_KEY)
|
|
throw Exception("Modification of primary key is not supported for replicated tables", ErrorCodes::NOT_IMPLEMENTED);
|
|
|
|
ColumnsDescription new_columns = data.getColumns();
|
|
params.apply(new_columns);
|
|
|
|
new_columns_str = new_columns.toString();
|
|
|
|
/// Do ALTER.
|
|
getZooKeeper()->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.");
|
|
|
|
/// Wait until all replicas will apply ALTER.
|
|
|
|
/// Subscribe to change of columns, to finish waiting if someone will do another ALTER.
|
|
if (!getZooKeeper()->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 = getZooKeeper()->getChildren(zookeeper_path + "/replicas");
|
|
|
|
std::set<String> inactive_replicas;
|
|
std::set<String> timed_out_replicas;
|
|
|
|
time_t replication_alter_columns_timeout = context.getSettingsRef().replication_alter_columns_timeout;
|
|
|
|
for (const String & replica : replicas)
|
|
{
|
|
LOG_DEBUG(log, "Waiting for " << replica << " to apply changes");
|
|
|
|
while (!shutdown_called)
|
|
{
|
|
/// Replica could be inactive.
|
|
if (!getZooKeeper()->exists(zookeeper_path + "/replicas/" + replica + "/is_active"))
|
|
{
|
|
LOG_WARNING(log, "Replica " << replica << " is not active during ALTER query."
|
|
" ALTER will be done asynchronously when replica becomes active.");
|
|
|
|
inactive_replicas.emplace(replica);
|
|
break;
|
|
}
|
|
|
|
String replica_columns_str;
|
|
|
|
/// Replica could has been removed.
|
|
if (!getZooKeeper()->tryGet(zookeeper_path + "/replicas/" + replica + "/columns", replica_columns_str, &stat))
|
|
{
|
|
LOG_WARNING(log, replica << " was removed");
|
|
break;
|
|
}
|
|
|
|
int replica_columns_version = stat.version;
|
|
|
|
/// The ALTER has been successfully applied.
|
|
if (replica_columns_str == new_columns_str)
|
|
break;
|
|
|
|
if (!getZooKeeper()->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;
|
|
}
|
|
|
|
if (!getZooKeeper()->exists(zookeeper_path + "/replicas/" + replica + "/columns", &stat, alter_query_event))
|
|
{
|
|
LOG_WARNING(log, replica << " was removed");
|
|
break;
|
|
}
|
|
|
|
if (stat.version != replica_columns_version)
|
|
continue;
|
|
|
|
if (!replication_alter_columns_timeout)
|
|
{
|
|
alter_query_event->wait();
|
|
/// Everything is fine.
|
|
}
|
|
else if (alter_query_event->tryWait(replication_alter_columns_timeout * 1000))
|
|
{
|
|
/// Everything is fine.
|
|
}
|
|
else
|
|
{
|
|
LOG_WARNING(log, "Timeout when waiting for replica " << replica << " to apply ALTER."
|
|
" ALTER will be done asynchronously.");
|
|
|
|
timed_out_replicas.emplace(replica);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (shutdown_called)
|
|
throw Exception("Alter is not finished because table shutdown was called. Alter will be done after table restart.",
|
|
ErrorCodes::UNFINISHED);
|
|
|
|
if (!inactive_replicas.empty() || !timed_out_replicas.empty())
|
|
{
|
|
std::stringstream exception_message;
|
|
exception_message << "Alter is not finished because";
|
|
|
|
if (!inactive_replicas.empty())
|
|
{
|
|
exception_message << " some replicas are inactive right now";
|
|
|
|
for (auto it = inactive_replicas.begin(); it != inactive_replicas.end(); ++it)
|
|
exception_message << (it == inactive_replicas.begin() ? ": " : ", ") << *it;
|
|
}
|
|
|
|
if (!timed_out_replicas.empty() && !inactive_replicas.empty())
|
|
exception_message << " and";
|
|
|
|
if (!timed_out_replicas.empty())
|
|
{
|
|
exception_message << " timeout when waiting for some replicas";
|
|
|
|
for (auto it = timed_out_replicas.begin(); it != timed_out_replicas.end(); ++it)
|
|
exception_message << (it == timed_out_replicas.begin() ? ": " : ", ") << *it;
|
|
|
|
exception_message << " (replication_alter_columns_timeout = " << replication_alter_columns_timeout << ")";
|
|
}
|
|
|
|
exception_message << ". Alter will be done asynchronously.";
|
|
|
|
throw Exception(exception_message.str(), ErrorCodes::UNFINISHED);
|
|
}
|
|
}
|
|
|
|
LOG_DEBUG(log, "ALTER finished");
|
|
}
|
|
|
|
|
|
/// The name of an imaginary part covering all possible parts in the specified partition with numbers in the range from zero to specified right bound.
|
|
static String getFakePartNameCoveringPartRange(
|
|
MergeTreeDataFormatVersion format_version, const String & partition_id, UInt64 left, UInt64 right)
|
|
{
|
|
/// Artificial high level is choosen, to make this part "covering" all parts inside.
|
|
MergeTreePartInfo part_info(partition_id, left, right, 999999999);
|
|
if (format_version < MERGE_TREE_DATA_MIN_FORMAT_VERSION_WITH_CUSTOM_PARTITIONING)
|
|
{
|
|
/// The date range is all month long.
|
|
const auto & lut = DateLUT::instance();
|
|
time_t start_time = lut.YYYYMMDDToDate(parse<UInt32>(partition_id + "01"));
|
|
DayNum_t left_date = lut.toDayNum(start_time);
|
|
DayNum_t right_date = DayNum_t(static_cast<size_t>(left_date) + lut.daysInMonth(start_time) - 1);
|
|
return part_info.getPartNameV0(left_date, right_date);
|
|
}
|
|
else
|
|
return part_info.getPartName();
|
|
}
|
|
|
|
|
|
String StorageReplicatedMergeTree::getFakePartNameCoveringAllPartsInPartition(
|
|
const String & partition_id, Int64 * out_min_block, Int64 * out_max_block)
|
|
{
|
|
/// Even if there is no data in the partition, you still need to mark the range for deletion.
|
|
/// - Because before executing DETACH, tasks for downloading parts to this partition can be executed.
|
|
Int64 left = 0;
|
|
|
|
/** Let's skip one number in `block_numbers` for the partition being deleted, and we will only delete parts until this number.
|
|
* This prohibits merges of deleted parts with the new inserted data.
|
|
* Invariant: merges of deleted parts with other parts do not appear in the log.
|
|
* NOTE: If you need to similarly support a `DROP PART` request, you will have to think of some new mechanism for it,
|
|
* to guarantee this invariant.
|
|
*/
|
|
Int64 right;
|
|
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
AbandonableLockInZooKeeper block_number_lock = allocateBlockNumber(partition_id, zookeeper);
|
|
right = block_number_lock.getNumber();
|
|
block_number_lock.unlock();
|
|
}
|
|
|
|
/// Empty partition.
|
|
if (right == 0)
|
|
return {};
|
|
|
|
--right;
|
|
|
|
if (out_min_block)
|
|
*out_min_block = left;
|
|
if (out_max_block)
|
|
*out_max_block = right;
|
|
return getFakePartNameCoveringPartRange(data.format_version, partition_id, left, right);
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::clearColumnInPartition(
|
|
const ASTPtr & partition, const Field & column_name, const Context & context)
|
|
{
|
|
assertNotReadonly();
|
|
|
|
/// We don't block merges, so anyone can manage this task (not only leader)
|
|
|
|
String partition_id = data.getPartitionIDFromQuery(partition, context);
|
|
String fake_part_name = getFakePartNameCoveringAllPartsInPartition(partition_id);
|
|
|
|
if (fake_part_name.empty())
|
|
{
|
|
LOG_INFO(log, "Will not clear partition " << partition_id << ", it is empty.");
|
|
return;
|
|
}
|
|
|
|
/// We allocated new block number for this part, so new merges can't merge clearing parts with new ones
|
|
|
|
LogEntry entry;
|
|
entry.type = LogEntry::CLEAR_COLUMN;
|
|
entry.new_part_name = fake_part_name;
|
|
entry.column_name = column_name.safeGet<String>();
|
|
entry.create_time = time(nullptr);
|
|
|
|
String log_znode_path = getZooKeeper()->create(zookeeper_path + "/log/log-", entry.toString(), zkutil::CreateMode::PersistentSequential);
|
|
entry.znode_name = log_znode_path.substr(log_znode_path.find_last_of('/') + 1);
|
|
|
|
/// If necessary, wait until the operation is performed on itself or on all replicas.
|
|
if (context.getSettingsRef().replication_alter_partitions_sync != 0)
|
|
{
|
|
if (context.getSettingsRef().replication_alter_partitions_sync == 1)
|
|
waitForReplicaToProcessLogEntry(replica_name, entry);
|
|
else
|
|
waitForAllReplicasToProcessLogEntry(entry);
|
|
}
|
|
}
|
|
|
|
void StorageReplicatedMergeTree::dropPartition(const ASTPtr & query, const ASTPtr & partition, bool detach, const Context & context)
|
|
{
|
|
assertNotReadonly();
|
|
|
|
zkutil::ZooKeeperPtr zookeeper = getZooKeeper();
|
|
|
|
if (!is_leader_node)
|
|
{
|
|
sendRequestToLeaderReplica(query, context.getSettingsRef());
|
|
return;
|
|
}
|
|
|
|
String partition_id = data.getPartitionIDFromQuery(partition, context);
|
|
|
|
Int64 min_block = 0;
|
|
Int64 max_block = 0;
|
|
String fake_part_name = getFakePartNameCoveringAllPartsInPartition(partition_id, &min_block, &max_block);
|
|
|
|
if (fake_part_name.empty())
|
|
{
|
|
LOG_INFO(log, "Will not drop partition " << partition_id << ", it is empty.");
|
|
return;
|
|
}
|
|
|
|
clearBlocksInPartition(*zookeeper, partition_id, min_block, max_block);
|
|
|
|
/** Forbid to choose the parts to be deleted for merging.
|
|
* Invariant: after the `DROP_RANGE` entry appears in the log, merge of deleted parts will not appear in the log.
|
|
*/
|
|
{
|
|
std::lock_guard<std::mutex> merge_selecting_lock(merge_selecting_mutex);
|
|
queue.disableMergesInRange(fake_part_name);
|
|
}
|
|
|
|
LOG_DEBUG(log, "Disabled merges covered by range " << fake_part_name);
|
|
|
|
/// Finally, having achieved the necessary invariants, you can put an entry in the log.
|
|
LogEntry entry;
|
|
entry.type = LogEntry::DROP_RANGE;
|
|
entry.source_replica = replica_name;
|
|
entry.new_part_name = fake_part_name;
|
|
entry.detach = detach;
|
|
entry.create_time = time(nullptr);
|
|
|
|
String log_znode_path = zookeeper->create(zookeeper_path + "/log/log-", entry.toString(), zkutil::CreateMode::PersistentSequential);
|
|
entry.znode_name = log_znode_path.substr(log_znode_path.find_last_of('/') + 1);
|
|
|
|
/// If necessary, wait until the operation is performed on itself or on all replicas.
|
|
if (context.getSettingsRef().replication_alter_partitions_sync != 0)
|
|
{
|
|
if (context.getSettingsRef().replication_alter_partitions_sync == 1)
|
|
waitForReplicaToProcessLogEntry(replica_name, entry);
|
|
else
|
|
waitForAllReplicasToProcessLogEntry(entry);
|
|
}
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::attachPartition(const ASTPtr & partition, bool attach_part, const Context & context)
|
|
{
|
|
assertNotReadonly();
|
|
|
|
String partition_id;
|
|
|
|
if (attach_part)
|
|
partition_id = typeid_cast<const ASTLiteral &>(*partition).value.safeGet<String>();
|
|
else
|
|
partition_id = data.getPartitionIDFromQuery(partition, context);
|
|
|
|
String source_dir = "detached/";
|
|
|
|
/// Let's compose a list of parts that should be added.
|
|
Strings parts;
|
|
if (attach_part)
|
|
{
|
|
parts.push_back(partition_id);
|
|
}
|
|
else
|
|
{
|
|
LOG_DEBUG(log, "Looking for parts for partition " << partition_id << " in " << source_dir);
|
|
ActiveDataPartSet active_parts(data.format_version);
|
|
|
|
std::set<String> part_names;
|
|
for (Poco::DirectoryIterator it = Poco::DirectoryIterator(full_path + source_dir); it != Poco::DirectoryIterator(); ++it)
|
|
{
|
|
String name = it.name();
|
|
MergeTreePartInfo part_info;
|
|
if (!MergeTreePartInfo::tryParsePartName(name, &part_info, data.format_version))
|
|
continue;
|
|
if (part_info.partition_id != partition_id)
|
|
continue;
|
|
LOG_DEBUG(log, "Found part " << name);
|
|
active_parts.add(name);
|
|
part_names.insert(name);
|
|
}
|
|
LOG_DEBUG(log, active_parts.size() << " of them are active");
|
|
parts = active_parts.getParts();
|
|
|
|
/// Inactive parts rename so they can not be attached in case of repeated ATTACH.
|
|
for (const auto & name : part_names)
|
|
{
|
|
String containing_part = active_parts.getContainingPart(name);
|
|
if (!containing_part.empty() && containing_part != name)
|
|
Poco::File(full_path + source_dir + name).renameTo(full_path + source_dir + "inactive_" + name);
|
|
}
|
|
}
|
|
|
|
/// Synchronously check that added parts exist and are not broken. We will write checksums.txt if it does not exist.
|
|
LOG_DEBUG(log, "Checking parts");
|
|
std::vector<MergeTreeData::MutableDataPartPtr> loaded_parts;
|
|
for (const String & part : parts)
|
|
{
|
|
LOG_DEBUG(log, "Checking part " << part);
|
|
loaded_parts.push_back(data.loadPartAndFixMetadata(source_dir + part));
|
|
}
|
|
|
|
ReplicatedMergeTreeBlockOutputStream output(*this, 0, 0, false); /// TODO Allow to use quorum here.
|
|
for (auto & part : loaded_parts)
|
|
{
|
|
String old_name = part->name;
|
|
output.writeExistingPart(part);
|
|
LOG_DEBUG(log, "Attached part " << old_name << " as " << part->name);
|
|
}
|
|
}
|
|
|
|
|
|
bool StorageReplicatedMergeTree::checkTableCanBeDropped() const
|
|
{
|
|
/// Consider only synchronized data
|
|
const_cast<MergeTreeData &>(getData()).recalculateColumnSizes();
|
|
context.checkTableCanBeDropped(database_name, table_name, getData().getTotalActiveSizeInBytes());
|
|
return true;
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::drop()
|
|
{
|
|
{
|
|
auto zookeeper = tryGetZooKeeper();
|
|
|
|
if (is_readonly || !zookeeper)
|
|
throw Exception("Can't drop readonly replicated table (need to drop data in ZooKeeper as well)", ErrorCodes::TABLE_IS_READ_ONLY);
|
|
|
|
// checkTableCanBeDropped(); // uncomment to feel yourself safe
|
|
|
|
shutdown();
|
|
|
|
if (zookeeper->expired())
|
|
throw Exception("Table was not dropped because ZooKeeper session has expired.", ErrorCodes::TABLE_WAS_NOT_DROPPED);
|
|
|
|
LOG_INFO(log, "Removing replica " << replica_path);
|
|
replica_is_active_node = nullptr;
|
|
zookeeper->tryRemoveRecursive(replica_path);
|
|
|
|
/// Check that `zookeeper_path` exists: it could have been deleted by another replica after execution of previous line.
|
|
Strings replicas;
|
|
if (zookeeper->tryGetChildren(zookeeper_path + "/replicas", replicas) == ZooKeeperImpl::ZooKeeper::ZOK && replicas.empty())
|
|
{
|
|
LOG_INFO(log, "Removing table " << zookeeper_path << " (this might take several minutes)");
|
|
zookeeper->tryRemoveRecursive(zookeeper_path);
|
|
}
|
|
}
|
|
|
|
data.dropAllData();
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::rename(const String & new_path_to_db, const String & new_database_name, const String & new_table_name)
|
|
{
|
|
std::string new_full_path = new_path_to_db + escapeForFileName(new_table_name) + '/';
|
|
|
|
data.setPath(new_full_path);
|
|
|
|
database_name = new_database_name;
|
|
table_name = new_table_name;
|
|
full_path = new_full_path;
|
|
|
|
/// TODO: You can update names of loggers.
|
|
}
|
|
|
|
|
|
bool StorageReplicatedMergeTree::existsNodeCached(const std::string & path)
|
|
{
|
|
{
|
|
std::lock_guard<std::mutex> lock(existing_nodes_cache_mutex);
|
|
if (existing_nodes_cache.count(path))
|
|
return true;
|
|
}
|
|
|
|
bool res = getZooKeeper()->exists(path);
|
|
|
|
if (res)
|
|
{
|
|
std::lock_guard<std::mutex> lock(existing_nodes_cache_mutex);
|
|
existing_nodes_cache.insert(path);
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
|
|
AbandonableLockInZooKeeper StorageReplicatedMergeTree::allocateBlockNumber(const String & partition_id, zkutil::ZooKeeperPtr & zookeeper,
|
|
zkutil::Requests * precheck_ops)
|
|
{
|
|
String partition_path = zookeeper_path + "/block_numbers/" + partition_id;
|
|
if (!existsNodeCached(partition_path))
|
|
{
|
|
int code = zookeeper->tryCreate(partition_path, "", zkutil::CreateMode::Persistent);
|
|
if (code && code != ZooKeeperImpl::ZooKeeper::ZNODEEXISTS)
|
|
throw zkutil::KeeperException(code, partition_path);
|
|
}
|
|
|
|
return AbandonableLockInZooKeeper(
|
|
partition_path + "/block-",
|
|
zookeeper_path + "/temp", *zookeeper, precheck_ops);
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::waitForAllReplicasToProcessLogEntry(const ReplicatedMergeTreeLogEntryData & entry)
|
|
{
|
|
LOG_DEBUG(log, "Waiting for all replicas to process " << entry.znode_name);
|
|
|
|
Strings replicas = getZooKeeper()->getChildren(zookeeper_path + "/replicas");
|
|
for (const String & replica : replicas)
|
|
waitForReplicaToProcessLogEntry(replica, entry);
|
|
|
|
LOG_DEBUG(log, "Finished waiting for all replicas to process " << entry.znode_name);
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::waitForReplicaToProcessLogEntry(const String & replica, const ReplicatedMergeTreeLogEntryData & entry)
|
|
{
|
|
String entry_str = entry.toString();
|
|
String log_node_name;
|
|
|
|
/** Two types of entries can be passed to this function
|
|
* 1. (more often) From `log` directory - a common log, from where replicas copy entries to their queue.
|
|
* 2. From the `queue` directory of one of the replicas.
|
|
*
|
|
* The problem is that the numbers (`sequential` node) of the queue elements in `log` and in `queue` do not match.
|
|
* (And the numbers of the same log element for different replicas do not match in the `queue`.)
|
|
*
|
|
* Therefore, you should consider these cases separately.
|
|
*/
|
|
|
|
/** First, you need to wait until replica takes `queue` element from the `log` to its queue,
|
|
* if it has not been done already (see the `pullLogsToQueue` function).
|
|
*
|
|
* To do this, check its node `log_pointer` - the maximum number of the element taken from `log` + 1.
|
|
*/
|
|
|
|
if (startsWith(entry.znode_name, "log-"))
|
|
{
|
|
/** In this case, just take the number from the node name `log-xxxxxxxxxx`.
|
|
*/
|
|
|
|
UInt64 log_index = parse<UInt64>(entry.znode_name.substr(entry.znode_name.size() - 10));
|
|
log_node_name = entry.znode_name;
|
|
|
|
LOG_DEBUG(log, "Waiting for " << replica << " to pull " << log_node_name << " to queue");
|
|
|
|
/// Let's wait until entry gets into the replica queue.
|
|
while (true)
|
|
{
|
|
zkutil::EventPtr event = std::make_shared<Poco::Event>();
|
|
|
|
String log_pointer = getZooKeeper()->get(zookeeper_path + "/replicas/" + replica + "/log_pointer", nullptr, event);
|
|
if (!log_pointer.empty() && parse<UInt64>(log_pointer) > log_index)
|
|
break;
|
|
|
|
event->wait();
|
|
}
|
|
}
|
|
else if (startsWith(entry.znode_name, "queue-"))
|
|
{
|
|
/** In this case, the number of `log` node is unknown. You need look through everything from `log_pointer` to the end,
|
|
* looking for a node with the same content. And if we do not find it - then the replica has already taken this entry in its queue.
|
|
*/
|
|
|
|
String log_pointer = getZooKeeper()->get(zookeeper_path + "/replicas/" + replica + "/log_pointer");
|
|
|
|
Strings log_entries = getZooKeeper()->getChildren(zookeeper_path + "/log");
|
|
UInt64 log_index = 0;
|
|
bool found = false;
|
|
|
|
for (const String & log_entry_name : log_entries)
|
|
{
|
|
log_index = parse<UInt64>(log_entry_name.substr(log_entry_name.size() - 10));
|
|
|
|
if (!log_pointer.empty() && log_index < parse<UInt64>(log_pointer))
|
|
continue;
|
|
|
|
String log_entry_str;
|
|
bool exists = getZooKeeper()->tryGet(zookeeper_path + "/log/" + log_entry_name, log_entry_str);
|
|
if (exists && entry_str == log_entry_str)
|
|
{
|
|
found = true;
|
|
log_node_name = log_entry_name;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (found)
|
|
{
|
|
LOG_DEBUG(log, "Waiting for " << replica << " to pull " << log_node_name << " to queue");
|
|
|
|
/// Let's wait until the entry gets into the replica queue.
|
|
while (true)
|
|
{
|
|
zkutil::EventPtr event = std::make_shared<Poco::Event>();
|
|
|
|
String log_pointer = getZooKeeper()->get(zookeeper_path + "/replicas/" + replica + "/log_pointer", nullptr, event);
|
|
if (!log_pointer.empty() && parse<UInt64>(log_pointer) > log_index)
|
|
break;
|
|
|
|
event->wait();
|
|
}
|
|
}
|
|
}
|
|
else
|
|
throw Exception("Logical error: unexpected name of log node: " + entry.znode_name, ErrorCodes::LOGICAL_ERROR);
|
|
|
|
if (!log_node_name.empty())
|
|
LOG_DEBUG(log, "Looking for node corresponding to " << log_node_name << " in " << replica << " queue");
|
|
else
|
|
LOG_DEBUG(log, "Looking for corresponding node in " << replica << " queue");
|
|
|
|
/** Second - find the corresponding entry in the queue of the specified replica.
|
|
* Its number may match neither the `log` node nor the `queue` node of the current replica (for us).
|
|
* Therefore, we search by comparing the content.
|
|
*/
|
|
|
|
Strings queue_entries = getZooKeeper()->getChildren(zookeeper_path + "/replicas/" + replica + "/queue");
|
|
String queue_entry_to_wait_for;
|
|
|
|
for (const String & entry_name : queue_entries)
|
|
{
|
|
String queue_entry_str;
|
|
bool exists = getZooKeeper()->tryGet(zookeeper_path + "/replicas/" + replica + "/queue/" + entry_name, queue_entry_str);
|
|
if (exists && queue_entry_str == entry_str)
|
|
{
|
|
queue_entry_to_wait_for = entry_name;
|
|
break;
|
|
}
|
|
}
|
|
|
|
/// While looking for the record, it has already been executed and deleted.
|
|
if (queue_entry_to_wait_for.empty())
|
|
{
|
|
LOG_DEBUG(log, "No corresponding node found. Assuming it has been already processed." " Found " << queue_entries.size() << " nodes.");
|
|
return;
|
|
}
|
|
|
|
LOG_DEBUG(log, "Waiting for " << queue_entry_to_wait_for << " to disappear from " << replica << " queue");
|
|
|
|
/// Third - wait until the entry disappears from the replica queue.
|
|
getZooKeeper()->waitForDisappear(zookeeper_path + "/replicas/" + replica + "/queue/" + queue_entry_to_wait_for);
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::getStatus(Status & res, bool with_zk_fields)
|
|
{
|
|
auto zookeeper = tryGetZooKeeper();
|
|
|
|
res.is_leader = is_leader_node;
|
|
res.is_readonly = is_readonly;
|
|
res.is_session_expired = !zookeeper || zookeeper->expired();
|
|
|
|
res.queue = queue.getStatus();
|
|
res.absolute_delay = getAbsoluteDelay(); /// NOTE: may be slightly inconsistent with queue status.
|
|
|
|
res.parts_to_check = part_check_thread.size();
|
|
|
|
res.zookeeper_path = zookeeper_path;
|
|
res.replica_name = replica_name;
|
|
res.replica_path = replica_path;
|
|
res.columns_version = columns_version;
|
|
|
|
if (res.is_session_expired || !with_zk_fields)
|
|
{
|
|
res.log_max_index = 0;
|
|
res.log_pointer = 0;
|
|
res.total_replicas = 0;
|
|
res.active_replicas = 0;
|
|
}
|
|
else
|
|
{
|
|
auto log_entries = zookeeper->getChildren(zookeeper_path + "/log");
|
|
|
|
if (log_entries.empty())
|
|
{
|
|
res.log_max_index = 0;
|
|
}
|
|
else
|
|
{
|
|
const String & last_log_entry = *std::max_element(log_entries.begin(), log_entries.end());
|
|
res.log_max_index = parse<UInt64>(last_log_entry.substr(strlen("log-")));
|
|
}
|
|
|
|
String log_pointer_str = zookeeper->get(replica_path + "/log_pointer");
|
|
res.log_pointer = log_pointer_str.empty() ? 0 : parse<UInt64>(log_pointer_str);
|
|
|
|
auto all_replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
|
|
res.total_replicas = all_replicas.size();
|
|
|
|
res.active_replicas = 0;
|
|
for (const String & replica : all_replicas)
|
|
if (zookeeper->exists(zookeeper_path + "/replicas/" + replica + "/is_active"))
|
|
++res.active_replicas;
|
|
}
|
|
}
|
|
|
|
|
|
/// TODO: Probably it is better to have queue in ZK with tasks for leader (like DDL)
|
|
void StorageReplicatedMergeTree::sendRequestToLeaderReplica(const ASTPtr & query, const Settings & settings)
|
|
{
|
|
auto live_replicas = getZooKeeper()->getChildren(zookeeper_path + "/leader_election");
|
|
if (live_replicas.empty())
|
|
throw Exception("No active replicas", ErrorCodes::NO_ACTIVE_REPLICAS);
|
|
|
|
std::sort(live_replicas.begin(), live_replicas.end());
|
|
const auto leader = getZooKeeper()->get(zookeeper_path + "/leader_election/" + live_replicas.front());
|
|
|
|
if (leader == replica_name)
|
|
throw Exception("Leader was suddenly changed or logical error.", ErrorCodes::LEADERSHIP_CHANGED);
|
|
|
|
ReplicatedMergeTreeAddress leader_address(getZooKeeper()->get(zookeeper_path + "/replicas/" + leader + "/host"));
|
|
|
|
/// TODO: add setters and getters interface for database and table fields of AST
|
|
auto new_query = query->clone();
|
|
if (auto * alter = typeid_cast<ASTAlterQuery *>(new_query.get()))
|
|
{
|
|
alter->database = leader_address.database;
|
|
alter->table = leader_address.table;
|
|
}
|
|
else if (auto * optimize = typeid_cast<ASTOptimizeQuery *>(new_query.get()))
|
|
{
|
|
optimize->database = leader_address.database;
|
|
optimize->table = leader_address.table;
|
|
}
|
|
else
|
|
throw Exception("Can't proxy this query. Unsupported query type", ErrorCodes::NOT_IMPLEMENTED);
|
|
|
|
/// NOTE Works only if there is access from the default user without a password. You can fix it by adding a parameter to the server config.
|
|
|
|
auto timeouts = ConnectionTimeouts::getTCPTimeoutsWithoutFailover(context.getSettingsRef());
|
|
Connection connection(
|
|
leader_address.host,
|
|
leader_address.queries_port,
|
|
leader_address.database,
|
|
"", "", timeouts, "ClickHouse replica");
|
|
|
|
RemoteBlockInputStream stream(connection, formattedAST(new_query), {}, context, &settings);
|
|
NullBlockOutputStream output({});
|
|
|
|
copyData(stream, output);
|
|
return;
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::getQueue(LogEntriesData & res, String & replica_name_)
|
|
{
|
|
replica_name_ = replica_name;
|
|
queue.getEntries(res);
|
|
}
|
|
|
|
time_t StorageReplicatedMergeTree::getAbsoluteDelay() const
|
|
{
|
|
time_t min_unprocessed_insert_time = 0;
|
|
time_t max_processed_insert_time = 0;
|
|
queue.getInsertTimes(min_unprocessed_insert_time, max_processed_insert_time);
|
|
|
|
/// Load start time, then finish time to avoid reporting false delay when start time is updated
|
|
/// between loading of two variables.
|
|
time_t queue_update_start_time = last_queue_update_start_time.load();
|
|
time_t queue_update_finish_time = last_queue_update_finish_time.load();
|
|
|
|
time_t current_time = time(nullptr);
|
|
|
|
if (!queue_update_finish_time)
|
|
{
|
|
/// We have not updated queue even once yet (perhaps replica is readonly).
|
|
/// As we have no info about the current state of replication log, return effectively infinite delay.
|
|
return current_time;
|
|
}
|
|
else if (min_unprocessed_insert_time)
|
|
{
|
|
/// There are some unprocessed insert entries in queue.
|
|
return (current_time > min_unprocessed_insert_time) ? (current_time - min_unprocessed_insert_time) : 0;
|
|
}
|
|
else if (queue_update_start_time > queue_update_finish_time)
|
|
{
|
|
/// Queue is empty, but there are some in-flight or failed queue update attempts
|
|
/// (likely because of problems with connecting to ZooKeeper).
|
|
/// Return the time passed since last attempt.
|
|
return (current_time > queue_update_start_time) ? (current_time - queue_update_start_time) : 0;
|
|
}
|
|
else
|
|
{
|
|
/// Everything is up-to-date.
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
void StorageReplicatedMergeTree::getReplicaDelays(time_t & out_absolute_delay, time_t & out_relative_delay)
|
|
{
|
|
assertNotReadonly();
|
|
|
|
time_t current_time = time(nullptr);
|
|
|
|
out_absolute_delay = getAbsoluteDelay();
|
|
out_relative_delay = 0;
|
|
|
|
/** Relative delay is the maximum difference of absolute delay from any other replica,
|
|
* (if this replica lags behind any other live replica, or zero, otherwise).
|
|
* Calculated only if the absolute delay is large enough.
|
|
*/
|
|
|
|
if (out_absolute_delay < static_cast<time_t>(data.settings.min_relative_delay_to_yield_leadership))
|
|
return;
|
|
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
time_t max_replicas_unprocessed_insert_time = 0;
|
|
bool have_replica_with_nothing_unprocessed = false;
|
|
|
|
Strings replicas = zookeeper->getChildren(zookeeper_path + "/replicas");
|
|
|
|
for (const auto & replica : replicas)
|
|
{
|
|
if (replica == replica_name)
|
|
continue;
|
|
|
|
/// Skip dead replicas.
|
|
if (!zookeeper->exists(zookeeper_path + "/replicas/" + replica + "/is_active"))
|
|
continue;
|
|
|
|
String value;
|
|
if (!zookeeper->tryGet(zookeeper_path + "/replicas/" + replica + "/min_unprocessed_insert_time", value))
|
|
continue;
|
|
|
|
time_t replica_time = value.empty() ? 0 : parse<time_t>(value);
|
|
|
|
if (replica_time == 0)
|
|
{
|
|
/** Note
|
|
* The conclusion that the replica does not lag may be incorrect,
|
|
* because the information about `min_unprocessed_insert_time` is taken
|
|
* only from that part of the log that has been moved to the queue.
|
|
* If the replica for some reason has stalled `queueUpdatingThread`,
|
|
* then `min_unprocessed_insert_time` will be incorrect.
|
|
*/
|
|
|
|
have_replica_with_nothing_unprocessed = true;
|
|
break;
|
|
}
|
|
|
|
if (replica_time > max_replicas_unprocessed_insert_time)
|
|
max_replicas_unprocessed_insert_time = replica_time;
|
|
}
|
|
|
|
if (have_replica_with_nothing_unprocessed)
|
|
out_relative_delay = out_absolute_delay;
|
|
else
|
|
{
|
|
max_replicas_unprocessed_insert_time = std::min(current_time, max_replicas_unprocessed_insert_time);
|
|
time_t min_replicas_delay = current_time - max_replicas_unprocessed_insert_time;
|
|
if (out_absolute_delay > min_replicas_delay)
|
|
out_relative_delay = out_absolute_delay - min_replicas_delay;
|
|
}
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::fetchPartition(const ASTPtr & partition, const String & from_, const Context & context)
|
|
{
|
|
String partition_id = data.getPartitionIDFromQuery(partition, context);
|
|
|
|
String from = from_;
|
|
if (from.back() == '/')
|
|
from.resize(from.size() - 1);
|
|
|
|
LOG_INFO(log, "Will fetch partition " << partition_id << " from shard " << from_);
|
|
|
|
/** Let's check that there is no such partition in the `detached` directory (where we will write the downloaded parts).
|
|
* Unreliable (there is a race condition) - such a partition may appear a little later.
|
|
*/
|
|
Poco::DirectoryIterator dir_end;
|
|
for (Poco::DirectoryIterator dir_it{data.getFullPath() + "detached/"}; dir_it != dir_end; ++dir_it)
|
|
{
|
|
MergeTreePartInfo part_info;
|
|
if (MergeTreePartInfo::tryParsePartName(dir_it.name(), &part_info, data.format_version)
|
|
&& part_info.partition_id == partition_id)
|
|
throw Exception("Detached partition " + partition_id + " already exists.", ErrorCodes::PARTITION_ALREADY_EXISTS);
|
|
}
|
|
|
|
zkutil::Strings replicas;
|
|
zkutil::Strings active_replicas;
|
|
String best_replica;
|
|
|
|
{
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
/// List of replicas of source shard.
|
|
replicas = zookeeper->getChildren(from + "/replicas");
|
|
|
|
/// Leave only active replicas.
|
|
active_replicas.reserve(replicas.size());
|
|
|
|
for (const String & replica : replicas)
|
|
if (zookeeper->exists(from + "/replicas/" + replica + "/is_active"))
|
|
active_replicas.push_back(replica);
|
|
|
|
if (active_replicas.empty())
|
|
throw Exception("No active replicas for shard " + from, ErrorCodes::NO_ACTIVE_REPLICAS);
|
|
|
|
/** You must select the best (most relevant) replica.
|
|
* This is a replica with the maximum `log_pointer`, then with the minimum `queue` size.
|
|
* NOTE This is not exactly the best criteria. It does not make sense to download old partitions,
|
|
* and it would be nice to be able to choose the replica closest by network.
|
|
* NOTE Of course, there are data races here. You can solve it by retrying.
|
|
*/
|
|
Int64 max_log_pointer = -1;
|
|
UInt64 min_queue_size = std::numeric_limits<UInt64>::max();
|
|
|
|
for (const String & replica : active_replicas)
|
|
{
|
|
String current_replica_path = from + "/replicas/" + replica;
|
|
|
|
String log_pointer_str = zookeeper->get(current_replica_path + "/log_pointer");
|
|
Int64 log_pointer = log_pointer_str.empty() ? 0 : parse<UInt64>(log_pointer_str);
|
|
|
|
zkutil::Stat stat;
|
|
zookeeper->get(current_replica_path + "/queue", &stat);
|
|
size_t queue_size = stat.numChildren;
|
|
|
|
if (log_pointer > max_log_pointer
|
|
|| (log_pointer == max_log_pointer && queue_size < min_queue_size))
|
|
{
|
|
max_log_pointer = log_pointer;
|
|
min_queue_size = queue_size;
|
|
best_replica = replica;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (best_replica.empty())
|
|
throw Exception("Logical error: cannot choose best replica.", ErrorCodes::LOGICAL_ERROR);
|
|
|
|
LOG_INFO(log, "Found " << replicas.size() << " replicas, " << active_replicas.size() << " of them are active."
|
|
<< " Selected " << best_replica << " to fetch from.");
|
|
|
|
String best_replica_path = from + "/replicas/" + best_replica;
|
|
|
|
/// Let's find out which parts are on the best replica.
|
|
|
|
/** Trying to download these parts.
|
|
* Some of them could be deleted due to the merge.
|
|
* In this case, update the information about the available parts and try again.
|
|
*/
|
|
|
|
unsigned try_no = 0;
|
|
Strings missing_parts;
|
|
do
|
|
{
|
|
if (try_no)
|
|
LOG_INFO(log, "Some of parts (" << missing_parts.size() << ") are missing. Will try to fetch covering parts.");
|
|
|
|
if (try_no >= 5)
|
|
throw Exception("Too many retries to fetch parts from " + best_replica_path, ErrorCodes::TOO_MANY_RETRIES_TO_FETCH_PARTS);
|
|
|
|
Strings parts = getZooKeeper()->getChildren(best_replica_path + "/parts");
|
|
ActiveDataPartSet active_parts_set(data.format_version, parts);
|
|
Strings parts_to_fetch;
|
|
|
|
if (missing_parts.empty())
|
|
{
|
|
parts_to_fetch = active_parts_set.getParts();
|
|
|
|
/// Leaving only the parts of the desired partition.
|
|
Strings parts_to_fetch_partition;
|
|
for (const String & part : parts_to_fetch)
|
|
{
|
|
if (MergeTreePartInfo::fromPartName(part, data.format_version).partition_id == partition_id)
|
|
parts_to_fetch_partition.push_back(part);
|
|
}
|
|
|
|
parts_to_fetch = std::move(parts_to_fetch_partition);
|
|
|
|
if (parts_to_fetch.empty())
|
|
throw Exception("Partition " + partition_id + " on " + best_replica_path + " doesn't exist", ErrorCodes::PARTITION_DOESNT_EXIST);
|
|
}
|
|
else
|
|
{
|
|
for (const String & missing_part : missing_parts)
|
|
{
|
|
String containing_part = active_parts_set.getContainingPart(missing_part);
|
|
if (!containing_part.empty())
|
|
parts_to_fetch.push_back(containing_part);
|
|
else
|
|
LOG_WARNING(log, "Part " << missing_part << " on replica " << best_replica_path << " has been vanished.");
|
|
}
|
|
}
|
|
|
|
LOG_INFO(log, "Parts to fetch: " << parts_to_fetch.size());
|
|
|
|
missing_parts.clear();
|
|
for (const String & part : parts_to_fetch)
|
|
{
|
|
try
|
|
{
|
|
fetchPart(part, best_replica_path, true, 0);
|
|
}
|
|
catch (const DB::Exception & e)
|
|
{
|
|
if (e.code() != ErrorCodes::RECEIVED_ERROR_FROM_REMOTE_IO_SERVER && e.code() != ErrorCodes::RECEIVED_ERROR_TOO_MANY_REQUESTS)
|
|
throw;
|
|
|
|
LOG_INFO(log, e.displayText());
|
|
missing_parts.push_back(part);
|
|
}
|
|
}
|
|
|
|
++try_no;
|
|
} while (!missing_parts.empty());
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::freezePartition(const ASTPtr & partition, const String & with_name, const Context & context)
|
|
{
|
|
data.freezePartition(partition, with_name, context);
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::clearOldPartsAndRemoveFromZK()
|
|
{
|
|
/// Critical section is not required (since grabOldParts() returns unique part set on each call)
|
|
|
|
auto table_lock = lockStructure(false, __PRETTY_FUNCTION__);
|
|
auto zookeeper = getZooKeeper();
|
|
|
|
MergeTreeData::DataPartsVector parts = data.grabOldParts();
|
|
if (parts.empty())
|
|
return;
|
|
|
|
MergeTreeData::DataPartsVector parts_to_delete_only_from_filesystem; // Only duplicates
|
|
MergeTreeData::DataPartsVector parts_to_delete_completely; // All parts except duplicates
|
|
MergeTreeData::DataPartsVector parts_to_retry_deletion; // Parts that should be retried due to network problems
|
|
MergeTreeData::DataPartsVector parts_to_remove_from_filesystem; // Parts removed from ZK
|
|
|
|
for (const auto & part : parts)
|
|
{
|
|
if (!part->is_duplicate)
|
|
parts_to_delete_completely.emplace_back(part);
|
|
else
|
|
parts_to_delete_only_from_filesystem.emplace_back(part);
|
|
}
|
|
parts.clear();
|
|
|
|
auto remove_parts_from_filesystem = [log=log] (const MergeTreeData::DataPartsVector & parts_to_remove)
|
|
{
|
|
for (auto & part : parts_to_remove)
|
|
{
|
|
try
|
|
{
|
|
part->remove();
|
|
}
|
|
catch (...)
|
|
{
|
|
tryLogCurrentException(log, "There is a problem with deleting part " + part->name + " from filesystem");
|
|
}
|
|
}
|
|
};
|
|
|
|
/// Delete duplicate parts from filesystem
|
|
if (!parts_to_delete_only_from_filesystem.empty())
|
|
{
|
|
remove_parts_from_filesystem(parts_to_delete_only_from_filesystem);
|
|
data.removePartsFinally(parts_to_delete_only_from_filesystem);
|
|
|
|
LOG_DEBUG(log, "Removed " << parts_to_delete_only_from_filesystem.size() << " old duplicate parts");
|
|
}
|
|
|
|
/// Delete normal parts from ZooKeeper
|
|
NameSet part_names_to_retry_deletion;
|
|
try
|
|
{
|
|
Strings part_names_to_delete_completely;
|
|
for (const auto & part : parts_to_delete_completely)
|
|
part_names_to_delete_completely.emplace_back(part->name);
|
|
|
|
LOG_DEBUG(log, "Removing " << parts_to_delete_completely.size() << " old parts from ZooKeeper");
|
|
removePartsFromZooKeeper(zookeeper, part_names_to_delete_completely, &part_names_to_retry_deletion);
|
|
}
|
|
catch (...)
|
|
{
|
|
LOG_ERROR(log, "There is a problem with deleting parts from ZooKeeper: " << getCurrentExceptionMessage(false));
|
|
}
|
|
|
|
/// Part names that were reliably deleted from ZooKeeper should be deleted from filesystem
|
|
auto num_reliably_deleted_parts = parts_to_delete_completely.size() - part_names_to_retry_deletion.size();
|
|
LOG_DEBUG(log, "Removed " << num_reliably_deleted_parts << " old parts from ZooKeeper. Removing them from filesystem.");
|
|
|
|
/// Delete normal parts on two sets
|
|
for (auto & part : parts_to_delete_completely)
|
|
{
|
|
if (part_names_to_retry_deletion.count(part->name) == 0)
|
|
parts_to_remove_from_filesystem.emplace_back(part);
|
|
else
|
|
parts_to_retry_deletion.emplace_back(part);
|
|
}
|
|
|
|
/// Will retry deletion
|
|
if (!parts_to_retry_deletion.empty())
|
|
{
|
|
data.rollbackDeletingParts(parts_to_retry_deletion);
|
|
LOG_DEBUG(log, "Will retry deletion of " << parts_to_retry_deletion.size() << " parts in the next time");
|
|
}
|
|
|
|
/// Remove parts from filesystem and finally from data_parts
|
|
if (!parts_to_remove_from_filesystem.empty())
|
|
{
|
|
remove_parts_from_filesystem(parts_to_remove_from_filesystem);
|
|
data.removePartsFinally(parts_to_remove_from_filesystem);
|
|
|
|
LOG_DEBUG(log, "Removed " << parts_to_remove_from_filesystem.size() << " old parts");
|
|
}
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::removePartsFromZooKeeper(zkutil::ZooKeeperPtr & zookeeper, const Strings & part_names,
|
|
NameSet * parts_should_be_retied)
|
|
{
|
|
zkutil::Requests ops;
|
|
auto it_first_node_in_batch = part_names.cbegin();
|
|
|
|
for (auto it = part_names.cbegin(); it != part_names.cend(); ++it)
|
|
{
|
|
removePartFromZooKeeper(*it, ops);
|
|
|
|
auto it_next = std::next(it);
|
|
if (ops.size() >= zkutil::MULTI_BATCH_SIZE || it_next == part_names.cend())
|
|
{
|
|
zkutil::Responses unused_responses;
|
|
auto code = zookeeper->tryMultiNoThrow(ops, unused_responses);
|
|
ops.clear();
|
|
|
|
if (code == ZooKeeperImpl::ZooKeeper::ZNONODE)
|
|
{
|
|
/// Fallback
|
|
LOG_DEBUG(log, "There are no some part nodes in ZooKeeper, will remove part nodes sequentially");
|
|
|
|
for (auto it_in_batch = it_first_node_in_batch; it_in_batch != it_next; ++it_in_batch)
|
|
{
|
|
zkutil::Requests cur_ops;
|
|
removePartFromZooKeeper(*it_in_batch, cur_ops);
|
|
auto cur_code = zookeeper->tryMultiNoThrow(cur_ops, unused_responses);
|
|
|
|
if (cur_code == ZooKeeperImpl::ZooKeeper::ZNONODE)
|
|
{
|
|
LOG_DEBUG(log, "There is no part " << *it_in_batch << " in ZooKeeper, it was only in filesystem");
|
|
}
|
|
else if (parts_should_be_retied && zkutil::isHardwareError(cur_code))
|
|
{
|
|
parts_should_be_retied->emplace(*it_in_batch);
|
|
}
|
|
else if (cur_code)
|
|
{
|
|
LOG_WARNING(log, "Cannot remove part " << *it_in_batch << " from ZooKeeper: " << zkutil::ZooKeeper::error2string(cur_code));
|
|
}
|
|
}
|
|
}
|
|
else if (parts_should_be_retied && zkutil::isHardwareError(code))
|
|
{
|
|
for (auto it_in_batch = it_first_node_in_batch; it_in_batch != it_next; ++it_in_batch)
|
|
parts_should_be_retied->emplace(*it_in_batch);
|
|
}
|
|
else if (code)
|
|
{
|
|
LOG_WARNING(log, "There was a problem with deleting " << (it_next - it_first_node_in_batch)
|
|
<< " nodes from ZooKeeper: " << ::zkutil::ZooKeeper::error2string(code));
|
|
}
|
|
|
|
it_first_node_in_batch = it_next;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void StorageReplicatedMergeTree::clearBlocksInPartition(
|
|
zkutil::ZooKeeper & zookeeper, const String & partition_id, Int64 min_block_num, Int64 max_block_num)
|
|
{
|
|
Strings blocks;
|
|
if (zookeeper.tryGetChildren(zookeeper_path + "/blocks", blocks))
|
|
throw Exception(zookeeper_path + "/blocks doesn't exist", ErrorCodes::NOT_FOUND_NODE);
|
|
|
|
String partition_prefix = partition_id + "_";
|
|
std::vector<std::pair<String, std::future<zkutil::GetResponse>>> get_futures;
|
|
for (const String & block_id : blocks)
|
|
{
|
|
if (startsWith(block_id, partition_prefix))
|
|
{
|
|
String path = zookeeper_path + "/blocks/" + block_id;
|
|
get_futures.emplace_back(path, zookeeper.asyncTryGet(path));
|
|
}
|
|
}
|
|
|
|
std::vector<std::pair<String, std::future<zkutil::RemoveResponse>>> to_delete_futures;
|
|
for (auto & pair : get_futures)
|
|
{
|
|
const String & path = pair.first;
|
|
auto result = pair.second.get();
|
|
|
|
if (result.error == ZooKeeperImpl::ZooKeeper::ZNONODE)
|
|
continue;
|
|
|
|
ReadBufferFromString buf(result.data);
|
|
Int64 block_num = 0;
|
|
bool parsed = tryReadIntText(block_num, buf) && buf.eof();
|
|
if (!parsed || (min_block_num <= block_num && block_num <= max_block_num))
|
|
to_delete_futures.emplace_back(path, zookeeper.asyncTryRemove(path));
|
|
}
|
|
|
|
for (auto & pair : to_delete_futures)
|
|
{
|
|
const String & path = pair.first;
|
|
int32_t rc = pair.second.get().error;
|
|
if (rc == ZooKeeperImpl::ZooKeeper::ZNOTEMPTY)
|
|
{
|
|
/// Can happen if there are leftover block nodes with children created by previous server versions.
|
|
zookeeper.removeRecursive(path);
|
|
}
|
|
else if (rc)
|
|
LOG_WARNING(log,
|
|
"Error while deleting ZooKeeper path `" << path << "`: " + zkutil::ZooKeeper::error2string(rc) << ", ignoring.");
|
|
}
|
|
|
|
LOG_TRACE(log, "Deleted " << to_delete_futures.size() << " deduplication block IDs in partition ID " << partition_id);
|
|
}
|
|
|
|
}
|