thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-23 16:12:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

SQLUserDefinedFunctionsLoader allow symlinks in user_defined directory

Browse Source
This commit is contained in:
Maksim Kita 2022-06-17 17:43:42 +02:00
parent 53f5f355d9
commit 965b706350
1 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/Interpreters/UserDefinedSQLObjectsLoader.cpp
View File

@ -86,28 +86,33 @@ void UserDefinedSQLObjectsLoader::loadObjects(ContextPtr context)
if (unlikely(!enable_persistence)) if (unlikely(!enable_persistence))
return; return;
LOG_DEBUG(log, "loading user defined objects"); LOG_DEBUG(log, "Loading user defined objects");
String dir_path = context->getUserDefinedPath(); String dir_path = context->getUserDefinedPath();
Poco::DirectoryIterator dir_end; Poco::DirectoryIterator dir_end;
for (Poco::DirectoryIterator it(dir_path); it != dir_end; ++it) for (Poco::DirectoryIterator it(dir_path); it != dir_end; ++it)
{ {
if (it->isLink()) if (it->isDirectory())
continue; continue;
const auto & file_name = it.name(); const std::string & file_name = it.name();
/// For '.svn', '.gitignore' directory and similar. /// For '.svn', '.gitignore' directory and similar.
if (file_name.at(0) == '.') if (file_name.at(0) == '.')
continue; continue;
if (!it->isDirectory() && endsWith(file_name, ".sql")) if (!startsWith(file_name, "function_") || !endsWith(file_name, ".sql"))
{ continue;
std::string_view object_name = file_name;
object_name.remove_suffix(strlen(".sql")); std::string_view object_name = file_name;
object_name.remove_prefix(strlen("function_"));
loadUserDefinedObject(context, UserDefinedSQLObjectType::Function, object_name, dir_path + it.name()); object_name.remove_prefix(strlen("function_"));
} object_name.remove_suffix(strlen(".sql"));
if (object_name.empty())
continue;
loadUserDefinedObject(context, UserDefinedSQLObjectType::Function, object_name, dir_path + it.name());
} }
} }