#include #include #include #include #include #include #include #include #include #include #include #include namespace DB { namespace ErrorCodes { extern const int CANNOT_CREATE_RECURSIVE_FUNCTION; extern const int UNSUPPORTED_METHOD; } BlockIO InterpreterCreateFunctionQuery::execute() { FunctionNameNormalizer().visit(query_ptr.get()); ASTCreateFunctionQuery & create_function_query = query_ptr->as(); AccessRightsElements access_rights_elements; access_rights_elements.emplace_back(AccessType::CREATE_FUNCTION); if (create_function_query.or_replace) access_rights_elements.emplace_back(AccessType::DROP_FUNCTION); if (!create_function_query.cluster.empty()) return executeDDLQueryOnCluster(query_ptr, getContext(), access_rights_elements); auto current_context = getContext(); current_context->checkAccess(access_rights_elements); auto & user_defined_function_factory = UserDefinedSQLFunctionFactory::instance(); auto & function_name = create_function_query.function_name; bool if_not_exists = create_function_query.if_not_exists; bool replace = create_function_query.or_replace; create_function_query.if_not_exists = false; create_function_query.or_replace = false; validateFunction(create_function_query.function_core, function_name); user_defined_function_factory.registerFunction(current_context, function_name, query_ptr, replace, if_not_exists, persist_function); return {}; } void InterpreterCreateFunctionQuery::validateFunction(ASTPtr function, const String & name) { const auto * args_tuple = function->as()->arguments->children.at(0)->as(); std::unordered_set arguments; for (const auto & argument : args_tuple->arguments->children) { const auto & argument_name = argument->as()->name(); auto [_, inserted] = arguments.insert(argument_name); if (!inserted) throw Exception(ErrorCodes::UNSUPPORTED_METHOD, "Identifier {} already used as function parameter", argument_name); } ASTPtr function_body = function->as()->children.at(0)->children.at(1); validateFunctionRecursiveness(function_body, name); } 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(ErrorCodes::CANNOT_CREATE_RECURSIVE_FUNCTION, "You cannot create recursive function"); validateFunctionRecursiveness(child, function_to_create); } } }