mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-11 17:02:25 +00:00
48 lines
1.4 KiB
C++
48 lines
1.4 KiB
C++
#include <Poco/String.h>
|
|
#include <Interpreters/misc.h>
|
|
#include <Interpreters/MarkTableIdentifiersVisitor.h>
|
|
#include <Interpreters/IdentifierSemantic.h>
|
|
#include <Parsers/ASTFunction.h>
|
|
#include <Parsers/ASTSelectQuery.h>
|
|
#include <Parsers/ASTTablesInSelectQuery.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
bool MarkTableIdentifiersMatcher::needChildVisit(ASTPtr & node, const ASTPtr & child)
|
|
{
|
|
if (child->as<ASTSelectQuery>())
|
|
return false;
|
|
if (node->as<ASTTableExpression>())
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
void MarkTableIdentifiersMatcher::visit(ASTPtr & ast, Data & data)
|
|
{
|
|
if (auto * node_func = ast->as<ASTFunction>())
|
|
visit(*node_func, ast, data);
|
|
else if (auto * node_table = ast->as<ASTTableExpression>())
|
|
visit(*node_table, ast, data);
|
|
}
|
|
|
|
void MarkTableIdentifiersMatcher::visit(ASTTableExpression & table, ASTPtr &, Data &)
|
|
{
|
|
if (table.database_and_table_name)
|
|
setIdentifierSpecial(table.database_and_table_name);
|
|
}
|
|
|
|
void MarkTableIdentifiersMatcher::visit(const ASTFunction & func, ASTPtr &, Data & data)
|
|
{
|
|
/// `IN t` can be specified, where t is a table, which is equivalent to `IN (SELECT * FROM t)`.
|
|
if (functionIsInOrGlobalInOperator(func.name))
|
|
{
|
|
auto & ast = func.arguments->children.at(1);
|
|
if (auto opt_name = tryGetIdentifierName(ast))
|
|
if (!data.aliases.count(*opt_name))
|
|
setIdentifierSpecial(ast);
|
|
}
|
|
}
|
|
|
|
}
|