Log UDF name

Author: nibblerenush

src/Functions/UserDefined/UserDefinedExecutableFunctionFactory.cpp

@@ -2,9 +2,11 @@
 
 #include <filesystem>
 
+#include <Common/CurrentThread.h>
 #include <Common/filesystemHelpers.h>
 #include <Common/FieldVisitorToString.h>
 #include <Common/quoteString.h>
+#include <Core/Settings.h>
 #include <DataTypes/FieldToDataType.h>
 
 #include <Processors/Sources/ShellCommandSource.h>
@@ -28,6 +30,11 @@ namespace ErrorCodes
     extern const int BAD_ARGUMENTS;
 }
 
+namespace Setting
+{
+    extern const SettingsBool log_queries;
+}
+
 namespace
 {
 
@@ -245,6 +252,14 @@ FunctionOverloadResolverPtr UserDefinedExecutableFunctionFactory::get(const Stri
     const auto & loader = context->getExternalUserDefinedExecutableFunctionsLoader();
     auto executable_function = std::static_pointer_cast<const UserDefinedExecutableFunction>(loader.load(function_name));
     auto function = std::make_shared<UserDefinedFunction>(std::move(executable_function), std::move(context), std::move(parameters));
+
+    if (CurrentThread::isInitialized())
+    {
+        auto query_context = CurrentThread::get().getQueryContext();
+        if (query_context && query_context->getSettingsRef()[Setting::log_queries])
+            query_context->addQueryFactoriesInfo(Context::QueryLogFactories::Function, function_name);
+    }
+
     return std::make_unique<FunctionToOverloadResolverAdaptor>(std::move(function));
 }
 
@@ -257,6 +272,14 @@ FunctionOverloadResolverPtr UserDefinedExecutableFunctionFactory::tryGet(const S
     {
         auto executable_function = std::static_pointer_cast<const UserDefinedExecutableFunction>(load_result.object);
         auto function = std::make_shared<UserDefinedFunction>(std::move(executable_function), std::move(context), std::move(parameters));
+
+        if (CurrentThread::isInitialized())
+        {
+            auto query_context = CurrentThread::get().getQueryContext();
+            if (query_context && query_context->getSettingsRef()[Setting::log_queries])
+                query_context->addQueryFactoriesInfo(Context::QueryLogFactories::Function, function_name);
+        }
+
         return std::make_unique<FunctionToOverloadResolverAdaptor>(std::move(function));
     }