mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 18:32:29 +00:00
139 lines
3.4 KiB
C++
139 lines
3.4 KiB
C++
#include <Common/typeid_cast.h>
|
|
#include <Parsers/ASTIdentifier.h>
|
|
#include <Parsers/queryToString.h>
|
|
#include <IO/WriteBufferFromOStream.h>
|
|
#include <IO/WriteHelpers.h>
|
|
#include <Interpreters/IdentifierSemantic.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int UNEXPECTED_AST_STRUCTURE;
|
|
}
|
|
|
|
|
|
ASTPtr ASTIdentifier::clone() const
|
|
{
|
|
auto ret = std::make_shared<ASTIdentifier>(*this);
|
|
ret->semantic = std::make_shared<IdentifierSemanticImpl>(*ret->semantic);
|
|
return ret;
|
|
}
|
|
|
|
std::shared_ptr<ASTIdentifier> ASTIdentifier::createSpecial(const String & name, std::vector<String> && name_parts)
|
|
{
|
|
auto ret = std::make_shared<ASTIdentifier>(name, std::move(name_parts));
|
|
ret->semantic->special = true;
|
|
return ret;
|
|
}
|
|
|
|
ASTIdentifier::ASTIdentifier(const String & name_, std::vector<String> && name_parts_)
|
|
: name(name_)
|
|
, name_parts(name_parts_)
|
|
, semantic(std::make_shared<IdentifierSemanticImpl>())
|
|
{
|
|
}
|
|
|
|
ASTIdentifier::ASTIdentifier(std::vector<String> && name_parts_)
|
|
: ASTIdentifier(name_parts_.at(0) + '.' + name_parts_.at(1), std::move(name_parts_))
|
|
{}
|
|
|
|
void ASTIdentifier::setShortName(const String & new_name)
|
|
{
|
|
name = new_name;
|
|
name_parts.clear();
|
|
|
|
bool special = semantic->special;
|
|
*semantic = IdentifierSemanticImpl();
|
|
semantic->special = special;
|
|
}
|
|
|
|
void ASTIdentifier::restoreCompoundName()
|
|
{
|
|
if (name_parts.empty())
|
|
return;
|
|
name = name_parts[0];
|
|
for (size_t i = 1; i < name_parts.size(); ++i)
|
|
name += '.' + name_parts[i];
|
|
}
|
|
|
|
void ASTIdentifier::formatImplWithoutAlias(const FormatSettings & settings, FormatState &, FormatStateStacked) const
|
|
{
|
|
auto format_element = [&](const String & elem_name)
|
|
{
|
|
settings.ostr << (settings.hilite ? hilite_identifier : "");
|
|
settings.writeIdentifier(elem_name);
|
|
settings.ostr << (settings.hilite ? hilite_none : "");
|
|
};
|
|
|
|
/// It could be compound but short
|
|
if (!isShort())
|
|
{
|
|
for (size_t i = 0, size = name_parts.size(); i < size; ++i)
|
|
{
|
|
if (i != 0)
|
|
settings.ostr << '.';
|
|
|
|
format_element(name_parts[i]);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
format_element(name);
|
|
}
|
|
}
|
|
|
|
void ASTIdentifier::appendColumnNameImpl(WriteBuffer & ostr) const
|
|
{
|
|
writeString(name, ostr);
|
|
}
|
|
|
|
ASTPtr createTableIdentifier(const String & database_name, const String & table_name)
|
|
{
|
|
if (database_name.empty())
|
|
return ASTIdentifier::createSpecial(table_name);
|
|
|
|
ASTPtr database_and_table = ASTIdentifier::createSpecial(database_name + "." + table_name, {database_name, table_name});
|
|
return database_and_table;
|
|
}
|
|
|
|
String getIdentifierName(const IAST * ast)
|
|
{
|
|
String res;
|
|
if (tryGetIdentifierNameInto(ast, res))
|
|
return res;
|
|
throw Exception(ast ? queryToString(*ast) + " is not an identifier" : "AST node is nullptr", ErrorCodes::UNEXPECTED_AST_STRUCTURE);
|
|
}
|
|
|
|
std::optional<String> tryGetIdentifierName(const IAST * ast)
|
|
{
|
|
String res;
|
|
if (tryGetIdentifierNameInto(ast, res))
|
|
return res;
|
|
return {};
|
|
}
|
|
|
|
bool tryGetIdentifierNameInto(const IAST * ast, String & name)
|
|
{
|
|
if (ast)
|
|
{
|
|
if (const auto * node = ast->as<ASTIdentifier>())
|
|
{
|
|
name = node->name;
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void setIdentifierSpecial(ASTPtr & ast)
|
|
{
|
|
if (ast)
|
|
if (auto * id = ast->as<ASTIdentifier>())
|
|
id->semantic->special = true;
|
|
}
|
|
|
|
}
|