mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-18 05:32:52 +00:00
92 lines
3.2 KiB
C++
92 lines
3.2 KiB
C++
#pragma once
|
|
|
|
#include <optional>
|
|
|
|
#include <Parsers/ASTQueryParameter.h>
|
|
#include <Parsers/ASTWithAlias.h>
|
|
#include <Core/UUID.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
struct IdentifierSemantic;
|
|
struct IdentifierSemanticImpl;
|
|
struct StorageID;
|
|
|
|
|
|
/// Identifier (column, table or alias)
|
|
class ASTIdentifier : public ASTWithAlias
|
|
{
|
|
friend class ReplaceQueryParameterVisitor;
|
|
public:
|
|
UUID uuid = UUIDHelpers::Nil;
|
|
|
|
explicit ASTIdentifier(const String & short_name, ASTPtr && name_param = {});
|
|
explicit ASTIdentifier(std::vector<String> && name_parts, bool special = false, std::vector<ASTPtr> && name_params = {});
|
|
|
|
/** Get the text that identifies this element. */
|
|
String getID(char delim) const override { return "Identifier" + (delim + name()); }
|
|
|
|
/** Get the query param out of a non-compound identifier. */
|
|
ASTPtr getParam() const;
|
|
|
|
ASTPtr clone() const override;
|
|
|
|
void collectIdentifierNames(IdentifierNameSet & set) const override { set.insert(name()); }
|
|
|
|
bool compound() const { return name_parts.size() > 1; }
|
|
bool isShort() const { return name_parts.size() == 1; }
|
|
bool supposedToBeCompound() const; // TODO(ilezhankin): get rid of this
|
|
|
|
void setShortName(const String & new_name);
|
|
|
|
/// The composite identifier will have a concatenated name (of the form a.b.c),
|
|
/// and individual components will be available inside the name_parts.
|
|
const String & shortName() const { return name_parts.back(); }
|
|
const String & name() const;
|
|
|
|
void restoreTable(); // TODO(ilezhankin): get rid of this
|
|
|
|
// FIXME: used only when it's needed to rewrite distributed table name to real remote table name.
|
|
void resetTable(const String & database_name, const String & table_name); // TODO(ilezhankin): get rid of this
|
|
|
|
void updateTreeHashImpl(SipHash & hash_state) const override;
|
|
|
|
protected:
|
|
String full_name;
|
|
std::vector<String> name_parts;
|
|
|
|
void formatImplWithoutAlias(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const override;
|
|
void appendColumnNameImpl(WriteBuffer & ostr) const override;
|
|
|
|
private:
|
|
using ASTWithAlias::children; /// ASTIdentifier is child free
|
|
|
|
std::shared_ptr<IdentifierSemanticImpl> semantic; /// pimpl
|
|
|
|
friend struct IdentifierSemantic;
|
|
friend ASTPtr createTableIdentifier(const StorageID & table_id);
|
|
friend void setIdentifierSpecial(ASTPtr & ast);
|
|
friend StorageID getTableIdentifier(const ASTPtr & ast);
|
|
|
|
void resetFullName();
|
|
};
|
|
|
|
/// ASTIdentifier Helpers: hide casts and semantic.
|
|
|
|
ASTPtr createTableIdentifier(const String & database_name, const String & table_name);
|
|
ASTPtr createTableIdentifier(const StorageID & table_id);
|
|
void setIdentifierSpecial(ASTPtr & ast);
|
|
|
|
String getIdentifierName(const IAST * ast);
|
|
std::optional<String> tryGetIdentifierName(const IAST * ast);
|
|
bool tryGetIdentifierNameInto(const IAST * ast, String & name);
|
|
StorageID getTableIdentifier(const ASTPtr & ast);
|
|
|
|
inline String getIdentifierName(const ASTPtr & ast) { return getIdentifierName(ast.get()); }
|
|
inline std::optional<String> tryGetIdentifierName(const ASTPtr & ast) { return tryGetIdentifierName(ast.get()); }
|
|
inline bool tryGetIdentifierNameInto(const ASTPtr & ast, String & name) { return tryGetIdentifierNameInto(ast.get(), name); }
|
|
|
|
}
|