ClickHouse/src/Storages/PartitionCommands.cpp

101 lines
3.3 KiB
C++
Raw Normal View History

#include <Storages/PartitionCommands.h>
#include <Storages/IStorage.h>
#include <Storages/MergeTree/PartDestinationType.h>
#include <Parsers/ASTAlterQuery.h>
#include <Parsers/ASTIdentifier.h>
namespace DB
{
std::optional<PartitionCommand> PartitionCommand::parse(const ASTAlterCommand * command_ast)
{
if (command_ast->type == ASTAlterCommand::DROP_PARTITION)
{
PartitionCommand res;
res.type = DROP_PARTITION;
res.partition = command_ast->partition;
res.detach = command_ast->detach;
return res;
}
2019-07-22 11:23:11 +00:00
else if (command_ast->type == ASTAlterCommand::DROP_DETACHED_PARTITION)
{
PartitionCommand res;
res.type = DROP_DETACHED_PARTITION;
res.partition = command_ast->partition;
res.part = command_ast->part;
return res;
}
else if (command_ast->type == ASTAlterCommand::ATTACH_PARTITION)
{
PartitionCommand res;
res.type = ATTACH_PARTITION;
res.partition = command_ast->partition;
res.part = command_ast->part;
return res;
}
2019-07-18 15:19:03 +00:00
else if (command_ast->type == ASTAlterCommand::MOVE_PARTITION)
{
PartitionCommand res;
2019-07-18 15:19:03 +00:00
res.type = MOVE_PARTITION;
res.partition = command_ast->partition;
2019-07-18 15:19:03 +00:00
res.part = command_ast->part;
2019-07-23 09:38:26 +00:00
switch (command_ast->move_destination_type)
2019-07-18 15:19:03 +00:00
{
case PartDestinationType::DISK:
res.move_destination_type = PartitionCommand::MoveDestinationType::DISK;
2019-07-18 15:19:03 +00:00
break;
case PartDestinationType::VOLUME:
res.move_destination_type = PartitionCommand::MoveDestinationType::VOLUME;
2019-07-18 15:19:03 +00:00
break;
2019-12-17 07:06:41 +00:00
case PartDestinationType::TABLE:
res.move_destination_type = PartitionCommand::MoveDestinationType::TABLE;
2019-09-16 07:27:38 +00:00
res.to_database = command_ast->to_database;
res.to_table = command_ast->to_table;
2019-12-17 07:06:41 +00:00
break;
2019-10-09 13:02:05 +00:00
default:
2019-09-16 07:27:38 +00:00
break;
2019-07-18 15:19:03 +00:00
}
if (res.move_destination_type != PartitionCommand::MoveDestinationType::TABLE)
2019-09-16 07:27:38 +00:00
res.move_destination_name = command_ast->move_destination_name;
return res;
}
else if (command_ast->type == ASTAlterCommand::REPLACE_PARTITION)
{
PartitionCommand res;
res.type = REPLACE_PARTITION;
res.partition = command_ast->partition;
res.replace = command_ast->replace;
res.from_database = command_ast->from_database;
res.from_table = command_ast->from_table;
return res;
}
else if (command_ast->type == ASTAlterCommand::FETCH_PARTITION)
{
PartitionCommand res;
res.type = FETCH_PARTITION;
res.partition = command_ast->partition;
res.from_zookeeper_path = command_ast->from;
return res;
}
else if (command_ast->type == ASTAlterCommand::FREEZE_PARTITION)
{
PartitionCommand res;
res.type = FREEZE_PARTITION;
res.partition = command_ast->partition;
res.with_name = command_ast->with_name;
return res;
}
else if (command_ast->type == ASTAlterCommand::FREEZE_ALL)
{
PartitionCommand command;
command.type = PartitionCommand::FREEZE_ALL_PARTITIONS;
command.with_name = command_ast->with_name;
return command;
}
else
return {};
}
2020-03-17 17:22:41 +00:00
}