mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-23 16:12:01 +00:00
89 lines
2.4 KiB
C++
89 lines
2.4 KiB
C++
#pragma once
|
|
|
|
#include <DB/Parsers/IAST.h>
|
|
#include <DB/Functions/IFunction.h>
|
|
#include <DB/AggregateFunctions/IAggregateFunction.h>
|
|
#include <DB/IO/WriteBufferFromString.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/** Применение функции или оператора
|
|
*/
|
|
class ASTFunction : public IAST
|
|
{
|
|
public:
|
|
/// имя функции
|
|
String name;
|
|
/// аргументы
|
|
ASTPtr arguments;
|
|
/// параметры - для параметрических агрегатных функций. Пример: quantile(0.9)(x) - то, что в первых скобках - параметры.
|
|
ASTPtr parameters;
|
|
/// алиас, если есть
|
|
String alias;
|
|
|
|
bool is_aggregate_function;
|
|
bool is_lambda_expression;
|
|
|
|
/// сама функция
|
|
FunctionPtr function;
|
|
/// или агрегатная функция
|
|
AggregateFunctionPtr aggregate_function;
|
|
/// или лямбда-выражение (ColumnExpression)
|
|
ColumnPtr column_expression;
|
|
/// тип возвращаемого значения
|
|
DataTypePtr return_type;
|
|
|
|
ASTFunction() : is_aggregate_function(false), is_lambda_expression(false) {}
|
|
ASTFunction(StringRange range_) : IAST(range_), is_aggregate_function(false), is_lambda_expression(false) {}
|
|
|
|
String getColumnName() const
|
|
{
|
|
String res;
|
|
WriteBufferFromString wb(res);
|
|
writeString(name, wb);
|
|
|
|
if (parameters)
|
|
{
|
|
writeChar('(', wb);
|
|
for (ASTs::const_iterator it = parameters->children.begin(); it != parameters->children.end(); ++it)
|
|
{
|
|
if (it != parameters->children.begin())
|
|
writeCString(", ", wb);
|
|
writeString((*it)->getColumnName(), wb);
|
|
}
|
|
writeChar(')', wb);
|
|
}
|
|
|
|
writeChar('(', wb);
|
|
for (ASTs::const_iterator it = arguments->children.begin(); it != arguments->children.end(); ++it)
|
|
{
|
|
if (it != arguments->children.begin())
|
|
writeCString(", ", wb);
|
|
writeString((*it)->getColumnName(), wb);
|
|
}
|
|
writeChar(')', wb);
|
|
|
|
return res;
|
|
}
|
|
|
|
String getAlias() const { return alias.empty() ? getColumnName() : alias; }
|
|
|
|
/** Получить текст, который идентифицирует этот элемент. */
|
|
String getID() const { return "Function_" + name; }
|
|
|
|
ASTPtr clone() const
|
|
{
|
|
ASTFunction * res = new ASTFunction(*this);
|
|
res->children.clear();
|
|
|
|
if (arguments) { res->arguments = arguments->clone(); res->children.push_back(res->arguments); }
|
|
if (parameters) { res->parameters = parameters->clone(); res->children.push_back(res->parameters); }
|
|
|
|
return res;
|
|
}
|
|
};
|
|
|
|
}
|