mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-30 03:22:14 +00:00
40 lines
1.2 KiB
C++
40 lines
1.2 KiB
C++
#pragma once
|
||
|
||
#include <DB/IO/WriteBufferFromOStream.h>
|
||
#include <DB/Parsers/IAST.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
|
||
/** Базовый класс для AST, которые могут содержать алиас (идентификаторы, литералы, функции).
|
||
*/
|
||
class ASTWithAlias : public IAST
|
||
{
|
||
public:
|
||
/// Алиас, если есть, или пустая строка.
|
||
String alias;
|
||
|
||
using IAST::IAST;
|
||
|
||
String getAliasOrColumnName() const override { return alias.empty() ? getColumnName() : alias; }
|
||
String tryGetAlias() const override { return alias; }
|
||
void setAlias(const String & to) override { alias = to; }
|
||
|
||
/// Вызывает formatImplWithoutAlias, а также выводит алиас. Если надо - заключает всё выражение в скобки.
|
||
void formatImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const override final;
|
||
|
||
virtual void formatImplWithoutAlias(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const = 0;
|
||
};
|
||
|
||
/// helper for setting aliases and chaining result to other functions
|
||
inline ASTPtr setAlias(ASTPtr ast, const String & alias)
|
||
{
|
||
ast->setAlias(alias);
|
||
return ast;
|
||
};
|
||
|
||
|
||
}
|