mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-18 13:42:02 +00:00
96 lines
5.1 KiB
C++
96 lines
5.1 KiB
C++
#pragma once
|
|
|
|
#include <Poco/SharedPtr.h>
|
|
|
|
#include <DB/Core/Names.h>
|
|
#include <DB/Core/Block.h>
|
|
#include <DB/Core/ColumnNumbers.h>
|
|
#include <DB/Core/ColumnsWithNameAndType.h>
|
|
#include <DB/DataTypes/IDataType.h>
|
|
#include <DB/Interpreters/ExpressionActions.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/** Интерфейс для обычных функций.
|
|
* Обычные функции - это функции, которые не меняют количество строк в таблице,
|
|
* и результат работы которых для каждой строчки не зависит от других строк.
|
|
*
|
|
* Функция может принимать произвольное количество аргументов и возвращать произвольное количество значений.
|
|
* Типы и количество значений результата зависят от типов и количества аргументов.
|
|
*
|
|
* Функция диспетчеризуется для целого блока. Это позволяет производить всевозможные проверки редко,
|
|
* и делать основную работу в виде эффективного цикла.
|
|
*
|
|
* Функция применяется к одному или нескольким столбцам блока, и записывает свой результат,
|
|
* добавляя новые столбцы к блоку. Функция не модифицирует свои агрументы.
|
|
*/
|
|
class IFunction
|
|
{
|
|
public:
|
|
/** Наследник IFunction должен реализовать:
|
|
* - getName
|
|
* - либо getReturnType, либо getReturnTypeAndPrerequisites
|
|
* - одну из перегрузок execute.
|
|
*/
|
|
|
|
/// Получить основное имя функции.
|
|
virtual String getName() const = 0;
|
|
|
|
/// Получить тип результата по типам аргументов. Если функция неприменима для данных аргументов - кинуть исключение.
|
|
/// Перегрузка для тех, кому не нужны prerequisites и значения константных аргументов. Снаружи не вызывается.
|
|
virtual DataTypePtr getReturnType(const DataTypes & arguments) const
|
|
{
|
|
throw Exception("getReturnType is not implemented for " + getName(), ErrorCodes::NOT_IMPLEMENTED);
|
|
}
|
|
|
|
/** Получить тип результата по типам аргументов и значениям константных аргументов.
|
|
* Если функция неприменима для данных аргументов - кинуть исключение.
|
|
* Еще можно вернуть описание дополнительных столбцов, которые требуются для выполнения функции.
|
|
* Для неконстантных столбцов arguments[i].column = nullptr.
|
|
* Осмысленные типы элементов в out_prerequisites: APPLY_FUNCTION, ADD_COLUMN.
|
|
*/
|
|
virtual void getReturnTypeAndPrerequisites(
|
|
const ColumnsWithNameAndType & arguments,
|
|
DataTypePtr & out_return_type,
|
|
ExpressionActions::Actions & out_prerequisites)
|
|
{
|
|
DataTypes types(arguments.size());
|
|
for (size_t i = 0; i < arguments.size(); ++i)
|
|
types[i] = arguments[i].type;
|
|
out_return_type = getReturnType(types);
|
|
}
|
|
|
|
/// Вызывается, если хоть один агрумент функции - лямбда-выражение.
|
|
/// Для аргументов-лямбда-выражений определяет типы аргументов этих выражений и кладет результат в arguments.
|
|
virtual void getLambdaArgumentTypes(DataTypes & arguments) const
|
|
{
|
|
throw Exception("Function " + getName() + " can't have lambda-expressions as arguments", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT);
|
|
}
|
|
|
|
/// Выполнить функцию над блоком. Замечание: может вызываться одновременно из нескольких потоков, для одного объекта.
|
|
/// Перегрузка для тех, кому не нужны prerequisites. Снаружи не вызывается.
|
|
virtual void execute(Block & block, const ColumnNumbers & arguments, size_t result)
|
|
{
|
|
throw Exception("execute is not implemented for " + getName(), ErrorCodes::NOT_IMPLEMENTED);
|
|
}
|
|
|
|
/// Выполнить функцию над блоком. Замечание: может вызываться одновременно из нескольких потоков, для одного объекта.
|
|
/// prerequisites идут в том же порядке, что и out_prerequisites, полученные из getReturnTypeAndPrerequisites.
|
|
virtual void execute(Block & block, const ColumnNumbers & arguments, const ColumnNumbers & prerequisites, size_t result)
|
|
{
|
|
execute(block, arguments, result);
|
|
}
|
|
|
|
virtual ~IFunction() {}
|
|
};
|
|
|
|
|
|
using Poco::SharedPtr;
|
|
|
|
typedef SharedPtr<IFunction> FunctionPtr;
|
|
|
|
|
|
}
|