mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-18 05:32:52 +00:00
84 lines
2.8 KiB
C++
84 lines
2.8 KiB
C++
#pragma once
|
|
|
|
#include <Interpreters/IInterpreter.h>
|
|
#include <Storages/ColumnsDescription.h>
|
|
#include <Storages/IStorage_fwd.h>
|
|
#include <Storages/IndicesDescription.h>
|
|
#include <Storages/ConstraintsDescription.h>
|
|
#include <Common/ThreadPool.h>
|
|
#include <Access/AccessRightsElement.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class Context;
|
|
class ASTCreateQuery;
|
|
class ASTExpressionList;
|
|
class ASTConstraintDeclaration;
|
|
|
|
|
|
/** Allows to create new table or database,
|
|
* or create an object for existing table or database.
|
|
*/
|
|
class InterpreterCreateQuery : public IInterpreter
|
|
{
|
|
public:
|
|
InterpreterCreateQuery(const ASTPtr & query_ptr_, Context & context_);
|
|
|
|
BlockIO execute() override;
|
|
|
|
/// List of columns and their types in AST.
|
|
static ASTPtr formatColumns(const NamesAndTypesList & columns);
|
|
static ASTPtr formatColumns(const ColumnsDescription & columns);
|
|
|
|
static ASTPtr formatIndices(const IndicesDescription & indices);
|
|
static ASTPtr formatConstraints(const ConstraintsDescription & constraints);
|
|
|
|
void setForceRestoreData(bool has_force_restore_data_flag_)
|
|
{
|
|
has_force_restore_data_flag = has_force_restore_data_flag_;
|
|
}
|
|
|
|
void setInternal(bool internal_)
|
|
{
|
|
internal = internal_;
|
|
}
|
|
|
|
/// Obtain information about columns, their types, default values and column comments, for case when columns in CREATE query is specified explicitly.
|
|
static ColumnsDescription getColumnsDescription(const ASTExpressionList & columns, const Context & context);
|
|
static ConstraintsDescription getConstraintsDescription(const ASTExpressionList * constraints);
|
|
|
|
private:
|
|
struct TableProperties
|
|
{
|
|
ColumnsDescription columns;
|
|
IndicesDescription indices;
|
|
ConstraintsDescription constraints;
|
|
};
|
|
|
|
BlockIO createDatabase(ASTCreateQuery & create);
|
|
BlockIO createTable(ASTCreateQuery & create);
|
|
BlockIO createDictionary(ASTCreateQuery & create);
|
|
|
|
/// Calculate list of columns, constraints, indices, etc... of table. Rewrite query in canonical way.
|
|
TableProperties setProperties(ASTCreateQuery & create) const;
|
|
void validateTableStructure(const ASTCreateQuery & create, const TableProperties & properties) const;
|
|
void setEngine(ASTCreateQuery & create) const;
|
|
AccessRightsElements getRequiredAccess() const;
|
|
|
|
/// Create IStorage and add it to database. If table already exists and IF NOT EXISTS specified, do nothing and return false.
|
|
bool doCreateTable(const ASTCreateQuery & create, const TableProperties & properties);
|
|
/// Inserts data in created table if it's CREATE ... SELECT
|
|
BlockIO fillTableIfNeeded(const ASTCreateQuery & create);
|
|
|
|
ASTPtr query_ptr;
|
|
Context & context;
|
|
|
|
/// Skip safety threshold when loading tables.
|
|
bool has_force_restore_data_flag = false;
|
|
/// Is this an internal query - not from the user.
|
|
bool internal = false;
|
|
};
|
|
}
|