mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-01 06:00:49 +00:00
78 lines
2.4 KiB
C++
78 lines
2.4 KiB
C++
#include <DB/Interpreters/InterpreterSelectQuery.h>
|
||
#include <DB/Parsers/ASTIdentifier.h>
|
||
#include <DB/Parsers/ASTCreateQuery.h>
|
||
#include <DB/Parsers/ASTSelectQuery.h>
|
||
|
||
#include <DB/Storages/StorageView.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
|
||
StoragePtr StorageView::create(const String & table_name_, const String & database_name_,
|
||
Context & context_, ASTPtr & query_, NamesAndTypesListPtr columns_)
|
||
{
|
||
return (new StorageView(table_name_, database_name_, context_, query_, columns_))->thisPtr();
|
||
}
|
||
|
||
|
||
StorageView::StorageView(const String & table_name_, const String & database_name_,
|
||
Context & context_, ASTPtr & query_, NamesAndTypesListPtr columns_):
|
||
table_name(table_name_), database_name(database_name_), context(context_), columns(columns_)
|
||
{
|
||
ASTCreateQuery & create = dynamic_cast<ASTCreateQuery &>(*query_);
|
||
ASTSelectQuery & select = dynamic_cast<ASTSelectQuery &>(*create.select);
|
||
|
||
/// Если во внутреннем запросе не указана база данных, получить ее из контекста и записать в запрос.
|
||
if (!select.database)
|
||
{
|
||
select.database = new ASTIdentifier(StringRange(), database_name_, ASTIdentifier::Database);
|
||
select.children.push_back(select.database);
|
||
}
|
||
|
||
inner_query = select;
|
||
|
||
if (inner_query.database)
|
||
select_database_name = dynamic_cast<const ASTIdentifier &>(*inner_query.database).name;
|
||
else
|
||
throw Exception("Logical error while creating StorageView."
|
||
" Could not retrieve database name from select query.",
|
||
DB::ErrorCodes::LOGICAL_ERROR);
|
||
|
||
if (inner_query.table)
|
||
select_table_name = dynamic_cast<const ASTIdentifier &>(*inner_query.table).name;
|
||
else
|
||
throw Exception("Logical error while creating StorageView."
|
||
" Could not retrieve table name from select query.",
|
||
DB::ErrorCodes::LOGICAL_ERROR);
|
||
|
||
context.getGlobalContext().addDependency(
|
||
DatabaseAndTableName(select_database_name, select_table_name),
|
||
DatabaseAndTableName(database_name, table_name));
|
||
}
|
||
|
||
|
||
BlockInputStreams StorageView::read(
|
||
const Names & column_names,
|
||
ASTPtr query,
|
||
const Settings & settings,
|
||
QueryProcessingStage::Enum & processed_stage,
|
||
size_t max_block_size,
|
||
unsigned threads)
|
||
{
|
||
return BlockInputStreams(1,
|
||
InterpreterSelectQuery(getInnerQuery(), context, column_names, false).execute());
|
||
}
|
||
|
||
|
||
void StorageView::drop()
|
||
{
|
||
context.getGlobalContext().removeDependency(
|
||
DatabaseAndTableName(select_database_name, select_table_name),
|
||
DatabaseAndTableName(database_name, table_name));
|
||
}
|
||
|
||
|
||
}
|