mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-19 06:01:57 +00:00
81 lines
2.6 KiB
C++
81 lines
2.6 KiB
C++
#include <sstream>
|
|
|
|
#include <Parsers/parseQuery.h>
|
|
#include <Parsers/ParserCreateQuery.h>
|
|
#include <Parsers/ASTCreateQuery.h>
|
|
#include <Parsers/formatAST.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Interpreters/InterpreterCreateQuery.h>
|
|
#include <Storages/StorageFactory.h>
|
|
#include <Databases/DatabasesCommon.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int EMPTY_LIST_OF_COLUMNS_PASSED;
|
|
}
|
|
|
|
|
|
String getTableDefinitionFromCreateQuery(const ASTPtr & query)
|
|
{
|
|
ASTPtr query_clone = query->clone();
|
|
ASTCreateQuery & create = typeid_cast<ASTCreateQuery &>(*query_clone.get());
|
|
|
|
/// We remove everything that is not needed for ATTACH from the query.
|
|
create.attach = true;
|
|
create.database.clear();
|
|
create.as_database.clear();
|
|
create.as_table.clear();
|
|
create.if_not_exists = false;
|
|
create.is_populate = false;
|
|
|
|
/// For views it is necessary to save the SELECT query itself, for the rest - on the contrary
|
|
if (!create.is_view && !create.is_materialized_view)
|
|
create.select = nullptr;
|
|
|
|
std::ostringstream statement_stream;
|
|
formatAST(create, statement_stream, 0, false);
|
|
statement_stream << '\n';
|
|
return statement_stream.str();
|
|
}
|
|
|
|
|
|
std::pair<String, StoragePtr> createTableFromDefinition(
|
|
const String & definition,
|
|
const String & database_name,
|
|
const String & database_data_path,
|
|
Context & context,
|
|
bool has_force_restore_data_flag,
|
|
const String & description_for_error_message)
|
|
{
|
|
ParserCreateQuery parser;
|
|
ASTPtr ast = parseQuery(parser, definition.data(), definition.data() + definition.size(), description_for_error_message);
|
|
|
|
ASTCreateQuery & ast_create_query = typeid_cast<ASTCreateQuery &>(*ast);
|
|
ast_create_query.attach = true;
|
|
ast_create_query.database = database_name;
|
|
|
|
/// We do not directly use `InterpreterCreateQuery::execute`, because
|
|
/// - the database has not been created yet;
|
|
/// - the code is simpler, since the query is already brought to a suitable form.
|
|
if (!ast_create_query.columns)
|
|
throw Exception("Missing definition of columns.", ErrorCodes::EMPTY_LIST_OF_COLUMNS_PASSED);
|
|
|
|
InterpreterCreateQuery::ColumnsInfo columns_info = InterpreterCreateQuery::getColumnsInfo(*ast_create_query.columns, context);
|
|
|
|
return
|
|
{
|
|
ast_create_query.table,
|
|
StorageFactory::instance().get(
|
|
ast_create_query,
|
|
database_data_path, ast_create_query.table, database_name, context, context.getGlobalContext(),
|
|
columns_info.columns, columns_info.materialized_columns, columns_info.alias_columns, columns_info.column_defaults,
|
|
true, has_force_restore_data_flag)
|
|
};
|
|
}
|
|
|
|
}
|