ClickHouse/dbms/include/DB/Parsers/ASTFunction.h

84 lines
2.1 KiB
C++

#pragma once
#include <DB/Parsers/IAST.h>
#include <DB/Functions/IFunction.h>
#include <DB/AggregateFunctions/IAggregateFunction.h>
namespace DB
{
/** Применение функции или оператора
*/
class ASTFunction : public IAST
{
public:
/// имя функции
String name;
/// аргументы
ASTPtr arguments;
/// параметры - для параметрических агрегатных функций. Пример: quantile(0.9)(x) - то, что в первых скобках - параметры.
ASTPtr parameters;
/// алиас, если есть
String alias;
/// сама функция
FunctionPtr function;
/// или агрегатная функция
AggregateFunctionPtr aggregate_function;
/// тип возвращаемого значения
DataTypePtr return_type;
/// номер столбца возвращаемого значения
size_t return_column_number;
ASTFunction() {}
ASTFunction(StringRange range_) : IAST(range_) {}
String getColumnName()
{
std::stringstream s;
s << name;
if (parameters)
{
s << "(";
for (ASTs::iterator it = parameters->children.begin(); it != parameters->children.end(); ++it)
{
if (it != parameters->children.begin())
s << ", ";
s << (*it)->getColumnName();
}
s << ")";
}
s << "(";
for (ASTs::iterator it = arguments->children.begin(); it != arguments->children.end(); ++it)
{
if (it != arguments->children.begin())
s << ", ";
s << (*it)->getColumnName();
}
s << ")";
return s.str();
}
String getAlias() { return alias.empty() ? getColumnName() : alias; }
/** Получить текст, который идентифицирует этот элемент. */
String getID() { 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;
}
};
}