mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-17 21:24:28 +00:00
106 lines
3.5 KiB
C++
106 lines
3.5 KiB
C++
#include <Interpreters/ApplyWithSubqueryVisitor.h>
|
|
#include <Interpreters/IdentifierSemantic.h>
|
|
#include <Interpreters/StorageID.h>
|
|
#include <Interpreters/misc.h>
|
|
#include <Parsers/ASTFunction.h>
|
|
#include <Parsers/ASTSelectQuery.h>
|
|
#include <Parsers/ASTSubquery.h>
|
|
#include <Parsers/ASTTablesInSelectQuery.h>
|
|
#include <Parsers/ASTWithElement.h>
|
|
|
|
namespace DB
|
|
{
|
|
void ApplyWithSubqueryVisitor::visit(ASTPtr & ast, const Data & data)
|
|
{
|
|
if (auto * node_select = ast->as<ASTSelectQuery>())
|
|
visit(*node_select, data);
|
|
else
|
|
{
|
|
for (auto & child : ast->children)
|
|
visit(child, data);
|
|
if (auto * node_func = ast->as<ASTFunction>())
|
|
visit(*node_func, data);
|
|
else if (auto * node_table = ast->as<ASTTableExpression>())
|
|
visit(*node_table, data);
|
|
}
|
|
}
|
|
|
|
void ApplyWithSubqueryVisitor::visit(ASTSelectQuery & ast, const Data & data)
|
|
{
|
|
std::optional<Data> new_data;
|
|
if (auto with = ast.with())
|
|
{
|
|
for (auto & child : with->children)
|
|
{
|
|
visit(child, new_data ? *new_data : data);
|
|
if (auto * ast_with_elem = child->as<ASTWithElement>())
|
|
{
|
|
if (!new_data)
|
|
new_data = data;
|
|
new_data->subqueries[ast_with_elem->name] = ast_with_elem->subquery;
|
|
}
|
|
}
|
|
}
|
|
|
|
for (auto & child : ast.children)
|
|
{
|
|
if (child != ast.with())
|
|
visit(child, new_data ? *new_data : data);
|
|
}
|
|
}
|
|
|
|
void ApplyWithSubqueryVisitor::visit(ASTSelectWithUnionQuery & ast, const Data & data)
|
|
{
|
|
for (auto & child : ast.children)
|
|
visit(child, data);
|
|
}
|
|
|
|
void ApplyWithSubqueryVisitor::visit(ASTTableExpression & table, const Data & data)
|
|
{
|
|
if (table.database_and_table_name)
|
|
{
|
|
auto table_id = IdentifierSemantic::extractDatabaseAndTable(table.database_and_table_name->as<ASTIdentifier &>());
|
|
if (table_id.database_name.empty())
|
|
{
|
|
auto subquery_it = data.subqueries.find(table_id.table_name);
|
|
if (subquery_it != data.subqueries.end())
|
|
{
|
|
auto old_alias = table.database_and_table_name->tryGetAlias();
|
|
table.children.clear();
|
|
table.database_and_table_name.reset();
|
|
table.subquery = subquery_it->second->clone();
|
|
table.subquery->as<ASTSubquery &>().cte_name = table_id.table_name;
|
|
if (!old_alias.empty())
|
|
table.subquery->setAlias(old_alias);
|
|
table.children.emplace_back(table.subquery);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void ApplyWithSubqueryVisitor::visit(ASTFunction & func, const Data & data)
|
|
{
|
|
if (checkFunctionIsInOrGlobalInOperator(func))
|
|
{
|
|
auto & ast = func.arguments->children.at(1);
|
|
if (const auto * ident = ast->as<ASTIdentifier>())
|
|
{
|
|
auto table_id = IdentifierSemantic::extractDatabaseAndTable(*ident);
|
|
if (table_id.database_name.empty())
|
|
{
|
|
auto subquery_it = data.subqueries.find(table_id.table_name);
|
|
if (subquery_it != data.subqueries.end())
|
|
{
|
|
auto old_alias = func.arguments->children[1]->tryGetAlias();
|
|
func.arguments->children[1] = subquery_it->second->clone();
|
|
func.arguments->children[1]->as<ASTSubquery &>().cte_name = table_id.table_name;
|
|
if (!old_alias.empty())
|
|
func.arguments->children[1]->setAlias(old_alias);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|