mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 19:02:04 +00:00
76 lines
2.5 KiB
C++
76 lines
2.5 KiB
C++
#include <Interpreters/ExpressionActions.h>
|
|
#include <Interpreters/ExpressionAnalyzer.h>
|
|
#include <Interpreters/InterpreterCreateFunctionQuery.h>
|
|
#include <Interpreters/FunctionNameNormalizer.h>
|
|
#include <Functions/FunctionFactory.h>
|
|
#include <Parsers/ASTIdentifier.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int UNKNOWN_IDENTIFIER;
|
|
extern const int CANNOT_CREATE_RECURSIVE_FUNCTION;
|
|
}
|
|
|
|
BlockIO InterpreterCreateFunctionQuery::execute()
|
|
{
|
|
FunctionNameNormalizer().visit(query_ptr.get());
|
|
auto & create_function_query = query_ptr->as<ASTCreateFunctionQuery &>();
|
|
validateFunction(create_function_query.function_core, create_function_query.function_name);
|
|
FunctionFactory::instance().registerUserDefinedFunction(create_function_query);
|
|
return {};
|
|
}
|
|
|
|
void InterpreterCreateFunctionQuery::validateFunction(ASTPtr function, const String & name)
|
|
{
|
|
const auto * args_tuple = function->as<ASTFunction>()->arguments->children.at(0)->as<ASTFunction>();
|
|
std::unordered_set<String> arguments;
|
|
for (const auto & argument : args_tuple->arguments->children)
|
|
arguments.insert(argument->as<ASTIdentifier>()->name());
|
|
|
|
std::set<String> identifiers_in_body;
|
|
ASTPtr function_body = function->as<ASTFunction>()->children.at(0)->children.at(1);
|
|
getIdentifiers(function_body, identifiers_in_body);
|
|
|
|
for (const auto & identifier : identifiers_in_body)
|
|
{
|
|
if (!arguments.contains(identifier))
|
|
{
|
|
WriteBufferFromOwnString s;
|
|
s << "Identifier '" << identifier << "' does not exist in arguments";
|
|
throw Exception(s.str(), ErrorCodes::UNKNOWN_IDENTIFIER);
|
|
}
|
|
}
|
|
|
|
validateFunctionRecursiveness(function_body, name);
|
|
}
|
|
|
|
void InterpreterCreateFunctionQuery::getIdentifiers(ASTPtr node, std::set<String> & identifiers)
|
|
{
|
|
for (const auto & child : node->children)
|
|
{
|
|
auto identifier_name_opt = tryGetIdentifierName(child);
|
|
if (identifier_name_opt)
|
|
identifiers.insert(identifier_name_opt.value());
|
|
|
|
getIdentifiers(child, identifiers);
|
|
}
|
|
}
|
|
|
|
void InterpreterCreateFunctionQuery::validateFunctionRecursiveness(ASTPtr node, const String & function_to_create)
|
|
{
|
|
for (const auto & child : node->children)
|
|
{
|
|
auto function_name_opt = tryGetFunctionName(child);
|
|
if (function_name_opt && function_name_opt.value() == function_to_create)
|
|
throw Exception("You cannot create recursive function", ErrorCodes::CANNOT_CREATE_RECURSIVE_FUNCTION);
|
|
|
|
validateFunctionRecursiveness(child, function_to_create);
|
|
}
|
|
}
|
|
|
|
|
|
}
|