mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-29 21:20:49 +00:00
4c391f8e99
* initial commit: add setting and stub * typo * added test stub * fix * wip merging new integration test and code proto * adding steps interpreters * adding firstly proposed solution (moving parts etc) * added checking zookeeper path existence * fixing the include * fixing and sorting includes * fixing outdated struct * fix the name * added ast ptr as level of indirection * fix ref * updating the changes * working on test stub * fix iterator -> reference * revert rocksdb submodule update * fixed show privileges test * updated the test stub * replaced rand() with thread_local_rng(), updated the tests updated the test fixed test config path test fix removed error messages fixed the test updated the test fixed string literal fixed literal typo: = * fixed the empty replica error message * updated the test and the code with logs * updated the possible test cases, updated * added the code/test milestone comments * updated the test (added more testcases) * replaced native assert with CH one * individual replicas recursive delete fix * updated the AS db.name AST * two small logging fixes * manually generated AST fixes * Updated the test, added the possible algo change * Some thoughts about optimizing the solution: ALTER MOVE PARTITION .. TO TABLE -> move to detached/ + ALTER ... ATTACH * fix * Removed the replica sync in test as it's invalid * Some test tweaks * tmp * Rewrote the algo by using the executeQuery instead of hand-crafting the ASTPtr. Two questions still active. * tr: logging active parts * Extracted the parts moving algo into a separate helper function * Fixed the test data and the queries slightly * Replaced query to system.parts to direct invocation, started building the test that breaks on various parts. * Added the case for tables when at least one replica is alive * Updated the test to test replicas restoration by detaching/attaching * Altered the test to check restoration without replica restart * Added the tables swap in the start if the server failed last time * Hotfix when only /replicas/replica... path was deleted * Restore ZK paths while creating a replicated MergeTree table * Updated the docs, fixed the algo for individual replicas restoration case * Initial parts table storage fix, tests sync fix * Reverted individual replica restoration to general algo * Slightly optimised getDataParts * Trying another solution with parts detaching * Rewrote algo without any steps, added ON CLUSTER support * Attaching parts from other replica on restoration * Getting part checksums from ZK * Removed ON CLUSTER, finished working solution * Multiple small changes after review * Fixing parallel test * Supporting rewritten form on cluster * Test fix * Moar logging * Using source replica as checksum provider * improve test, remove some code from parser * Trying solution with move to detached + forget * Moving all parts (not only Committed) to detached * Edited docs for RESTORE REPLICA * Re-merging * minor fixes Co-authored-by: Alexander Tokmakov <avtokmakov@yandex-team.ru>
132 lines
3.5 KiB
C++
132 lines
3.5 KiB
C++
#include <Parsers/ASTIdentifier.h>
|
|
#include <Parsers/ASTRenameQuery.h>
|
|
|
|
#include <Parsers/CommonParsers.h>
|
|
#include <Parsers/ParserRenameQuery.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
|
|
/// Parse database.table or table.
|
|
static bool parseDatabaseAndTable(
|
|
ASTRenameQuery::Table & db_and_table, IParser::Pos & pos, Expected & expected)
|
|
{
|
|
ParserIdentifier name_p;
|
|
ParserToken s_dot(TokenType::Dot);
|
|
|
|
ASTPtr database;
|
|
ASTPtr table;
|
|
|
|
if (!name_p.parse(pos, table, expected))
|
|
return false;
|
|
|
|
if (s_dot.ignore(pos, expected))
|
|
{
|
|
database = table;
|
|
if (!name_p.parse(pos, table, expected))
|
|
return false;
|
|
}
|
|
|
|
db_and_table.database.clear();
|
|
tryGetIdentifierNameInto(database, db_and_table.database);
|
|
tryGetIdentifierNameInto(table, db_and_table.table);
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool ParserRenameQuery::parseImpl(Pos & pos, ASTPtr & node, Expected & expected)
|
|
{
|
|
ParserKeyword s_rename_table("RENAME TABLE");
|
|
ParserKeyword s_exchange_tables("EXCHANGE TABLES");
|
|
ParserKeyword s_rename_dictionary("RENAME DICTIONARY");
|
|
ParserKeyword s_exchange_dictionaries("EXCHANGE DICTIONARIES");
|
|
ParserKeyword s_rename_database("RENAME DATABASE");
|
|
ParserKeyword s_to("TO");
|
|
ParserKeyword s_and("AND");
|
|
ParserToken s_comma(TokenType::Comma);
|
|
|
|
bool exchange = false;
|
|
bool dictionary = false;
|
|
|
|
if (s_rename_table.ignore(pos, expected))
|
|
;
|
|
else if (s_exchange_tables.ignore(pos, expected))
|
|
exchange = true;
|
|
else if (s_rename_dictionary.ignore(pos, expected))
|
|
dictionary = true;
|
|
else if (s_exchange_dictionaries.ignore(pos, expected))
|
|
{
|
|
exchange = true;
|
|
dictionary = true;
|
|
}
|
|
else if (s_rename_database.ignore(pos, expected))
|
|
{
|
|
ASTPtr from_db;
|
|
ASTPtr to_db;
|
|
ParserIdentifier db_name_p;
|
|
if (!db_name_p.parse(pos, from_db, expected))
|
|
return false;
|
|
if (!s_to.ignore(pos, expected))
|
|
return false;
|
|
if (!db_name_p.parse(pos, to_db, expected))
|
|
return false;
|
|
|
|
String cluster_str;
|
|
if (ParserKeyword{"ON"}.ignore(pos, expected))
|
|
{
|
|
if (!ASTQueryWithOnCluster::parse(pos, cluster_str, expected))
|
|
return false;
|
|
}
|
|
|
|
auto query = std::make_shared<ASTRenameQuery>();
|
|
query->database = true;
|
|
query->elements.emplace({});
|
|
tryGetIdentifierNameInto(from_db, query->elements.front().from.database);
|
|
tryGetIdentifierNameInto(to_db, query->elements.front().to.database);
|
|
query->cluster = cluster_str;
|
|
node = query;
|
|
return true;
|
|
}
|
|
else
|
|
return false;
|
|
|
|
ASTRenameQuery::Elements elements;
|
|
|
|
const auto ignore_delim = [&] { return exchange ? s_and.ignore(pos) : s_to.ignore(pos); };
|
|
|
|
while (true)
|
|
{
|
|
if (!elements.empty() && !s_comma.ignore(pos))
|
|
break;
|
|
|
|
ASTRenameQuery::Element& ref = elements.emplace_back();
|
|
|
|
if (!parseDatabaseAndTable(ref.from, pos, expected)
|
|
|| !ignore_delim()
|
|
|| !parseDatabaseAndTable(ref.to, pos, expected))
|
|
return false;
|
|
}
|
|
|
|
String cluster_str;
|
|
if (ParserKeyword{"ON"}.ignore(pos, expected))
|
|
{
|
|
if (!ASTQueryWithOnCluster::parse(pos, cluster_str, expected))
|
|
return false;
|
|
}
|
|
|
|
auto query = std::make_shared<ASTRenameQuery>();
|
|
query->cluster = cluster_str;
|
|
node = query;
|
|
|
|
query->elements = elements;
|
|
query->exchange = exchange;
|
|
query->dictionary = dictionary;
|
|
return true;
|
|
}
|
|
|
|
|
|
}
|