mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-19 06:01:57 +00:00
86 lines
2.8 KiB
C++
86 lines
2.8 KiB
C++
#include <Interpreters/InterpreterAlterQuery.h>
|
|
#include <Interpreters/DDLWorker.h>
|
|
#include <Interpreters/MutationsInterpreter.h>
|
|
#include <Interpreters/AddDefaultDatabaseVisitor.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Parsers/ASTAlterQuery.h>
|
|
#include <Storages/IStorage.h>
|
|
#include <Storages/AlterCommands.h>
|
|
#include <Storages/MutationCommands.h>
|
|
#include <Storages/PartitionCommands.h>
|
|
#include <Common/typeid_cast.h>
|
|
|
|
#include <algorithm>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int LOGICAL_ERROR;
|
|
extern const int ILLEGAL_COLUMN;
|
|
}
|
|
|
|
|
|
InterpreterAlterQuery::InterpreterAlterQuery(const ASTPtr & query_ptr_, const Context & context_)
|
|
: query_ptr(query_ptr_), context(context_)
|
|
{
|
|
}
|
|
|
|
BlockIO InterpreterAlterQuery::execute()
|
|
{
|
|
const auto * alter = query_ptr->as<ASTAlterQuery>();
|
|
|
|
if (!alter->cluster.empty())
|
|
return executeDDLQueryOnCluster(query_ptr, context, {alter->database});
|
|
|
|
const String & table_name = alter->table;
|
|
String database_name = alter->database.empty() ? context.getCurrentDatabase() : alter->database;
|
|
StoragePtr table = context.getTable(database_name, table_name);
|
|
|
|
/// Add default database to table identifiers that we can encounter in e.g. default expressions,
|
|
/// mutation expression, etc.
|
|
AddDefaultDatabaseVisitor visitor(database_name);
|
|
ASTPtr command_list_ptr = alter->command_list->ptr();
|
|
visitor.visit(command_list_ptr);
|
|
|
|
AlterCommands alter_commands;
|
|
PartitionCommands partition_commands;
|
|
MutationCommands mutation_commands;
|
|
for (ASTAlterCommand * command_ast : alter->command_list->commands)
|
|
{
|
|
if (auto alter_command = AlterCommand::parse(command_ast))
|
|
alter_commands.emplace_back(std::move(*alter_command));
|
|
else if (auto partition_command = PartitionCommand::parse(command_ast))
|
|
partition_commands.emplace_back(std::move(*partition_command));
|
|
else if (auto mut_command = MutationCommand::parse(command_ast))
|
|
mutation_commands.emplace_back(std::move(*mut_command));
|
|
else
|
|
throw Exception("Wrong parameter type in ALTER query", ErrorCodes::LOGICAL_ERROR);
|
|
}
|
|
|
|
if (!mutation_commands.empty())
|
|
{
|
|
MutationsInterpreter(table, mutation_commands, context).validate();
|
|
table->mutate(mutation_commands, context);
|
|
}
|
|
|
|
if (!partition_commands.empty())
|
|
{
|
|
partition_commands.validate(*table);
|
|
table->alterPartition(query_ptr, partition_commands, context);
|
|
}
|
|
|
|
if (!alter_commands.empty())
|
|
{
|
|
auto table_lock_holder = table->lockAlterIntention(context.getCurrentQueryId());
|
|
alter_commands.validate(*table, context);
|
|
table->alter(alter_commands, database_name, table_name, context, table_lock_holder);
|
|
}
|
|
|
|
return {};
|
|
}
|
|
|
|
}
|