mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 00:22:29 +00:00
106 lines
3.8 KiB
C++
106 lines
3.8 KiB
C++
#pragma once
|
|
|
|
#include <Core/UUID.h>
|
|
#include <Parsers/ASTQueryParameter.h>
|
|
#include <Parsers/ASTWithAlias.h>
|
|
|
|
#include <optional>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
struct IdentifierSemantic;
|
|
struct IdentifierSemanticImpl;
|
|
struct StorageID;
|
|
|
|
class ASTTableIdentifier;
|
|
|
|
/// FIXME: rewrite code about params - they should be substituted at the parsing stage,
|
|
/// or parsed as a separate AST entity.
|
|
|
|
/// Generic identifier. ASTTableIdentifier - for table identifier.
|
|
class ASTIdentifier : public ASTWithAlias
|
|
{
|
|
public:
|
|
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
|
|
std::shared_ptr<ASTTableIdentifier> createTable() const; // returns |nullptr| if identifier is not table.
|
|
|
|
protected:
|
|
String full_name;
|
|
std::vector<String> name_parts;
|
|
std::shared_ptr<IdentifierSemanticImpl> semantic; /// pimpl
|
|
|
|
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
|
|
|
|
friend class ReplaceQueryParameterVisitor;
|
|
friend struct IdentifierSemantic;
|
|
friend void setIdentifierSpecial(ASTPtr & ast);
|
|
|
|
void resetFullName();
|
|
};
|
|
|
|
class ASTTableIdentifier : public ASTIdentifier
|
|
{
|
|
public:
|
|
explicit ASTTableIdentifier(const String & table_name, std::vector<ASTPtr> && name_params = {});
|
|
explicit ASTTableIdentifier(const StorageID & table_id, std::vector<ASTPtr> && name_params = {});
|
|
ASTTableIdentifier(const String & database_name, const String & table_name, std::vector<ASTPtr> && name_params = {});
|
|
|
|
String getID(char delim) const override { return "TableIdentifier" + (delim + name()); }
|
|
ASTPtr clone() const override;
|
|
|
|
UUID uuid = UUIDHelpers::Nil; // FIXME(ilezhankin): make private
|
|
|
|
StorageID getTableId() const;
|
|
String getDatabaseName() const;
|
|
|
|
// 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;
|
|
};
|
|
|
|
|
|
/// ASTIdentifier Helpers: hide casts and semantic.
|
|
|
|
void setIdentifierSpecial(ASTPtr & ast);
|
|
|
|
String getIdentifierName(const IAST * ast);
|
|
std::optional<String> tryGetIdentifierName(const IAST * ast);
|
|
bool tryGetIdentifierNameInto(const IAST * ast, String & name);
|
|
|
|
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); }
|
|
|
|
}
|