mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-01 14:10:50 +00:00
66 lines
1.6 KiB
C++
66 lines
1.6 KiB
C++
|
#include <memory>
|
||
|
#include <vector>
|
||
|
|
||
|
#include <Common/typeid_cast.h>
|
||
|
#include <Interpreters/evaluateQualified.h>
|
||
|
|
||
|
namespace DB
|
||
|
{
|
||
|
|
||
|
class ASTIdentifier;
|
||
|
class ASTQualifiedAsterisk;
|
||
|
class ASTSelectQuery;
|
||
|
struct ASTTableJoin;
|
||
|
|
||
|
class NamesAndTypesList;
|
||
|
|
||
|
|
||
|
class TranslateQualifiedNamesVisitor
|
||
|
{
|
||
|
public:
|
||
|
static constexpr const char * visitor_label = "translateQualifiedNames";
|
||
|
|
||
|
TranslateQualifiedNamesVisitor(const NamesAndTypesList & source_columns_, const std::vector<DatabaseAndTableWithAlias> & tables_,
|
||
|
std::ostream * ostr_ = nullptr)
|
||
|
: source_columns(source_columns_),
|
||
|
tables(tables_),
|
||
|
visit_depth(0),
|
||
|
ostr(ostr_)
|
||
|
{}
|
||
|
|
||
|
void visit(ASTPtr & ast) const
|
||
|
{
|
||
|
if (!tryVisit<ASTIdentifier>(ast) &&
|
||
|
!tryVisit<ASTQualifiedAsterisk>(ast) &&
|
||
|
!tryVisit<ASTTableJoin>(ast) &&
|
||
|
!tryVisit<ASTSelectQuery>(ast))
|
||
|
visitChildren(ast); /// default: do nothing, visit children
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
const NamesAndTypesList & source_columns;
|
||
|
const std::vector<DatabaseAndTableWithAlias> & tables;
|
||
|
mutable size_t visit_depth;
|
||
|
std::ostream * ostr;
|
||
|
|
||
|
void visit(ASTIdentifier * node, ASTPtr & ast) const;
|
||
|
void visit(ASTQualifiedAsterisk * node, ASTPtr & ast) const;
|
||
|
void visit(ASTTableJoin * node, ASTPtr & ast) const;
|
||
|
void visit(ASTSelectQuery * ast, ASTPtr &) const;
|
||
|
|
||
|
void visitChildren(ASTPtr &) const;
|
||
|
|
||
|
template <typename T>
|
||
|
bool tryVisit(ASTPtr & ast) const
|
||
|
{
|
||
|
if (T * t = typeid_cast<T *>(ast.get()))
|
||
|
{
|
||
|
visit(t, ast);
|
||
|
return true;
|
||
|
}
|
||
|
return false;
|
||
|
}
|
||
|
};
|
||
|
|
||
|
}
|