ClickHouse/dbms/src/Storages/StorageView.cpp

78 lines
2.4 KiB
C++
Raw Normal View History

#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();
}
2014-06-12 23:21:38 +00:00
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);
2014-06-12 23:21:38 +00:00
context.getGlobalContext().addDependency(
DatabaseAndTableName(select_database_name, select_table_name),
DatabaseAndTableName(database_name, table_name));
}
2014-06-12 23:21:38 +00:00
BlockInputStreams StorageView::read(
const Names & column_names,
ASTPtr query,
const Settings & settings,
QueryProcessingStage::Enum & processed_stage,
size_t max_block_size,
unsigned threads)
{
2014-06-12 23:21:38 +00:00
return BlockInputStreams(1,
InterpreterSelectQuery(getInnerQuery(), context, column_names, false).execute());
}
2014-06-12 23:21:38 +00:00
void StorageView::drop()
{
context.getGlobalContext().removeDependency(
DatabaseAndTableName(select_database_name, select_table_name),
DatabaseAndTableName(database_name, table_name));
}
}