YQ Connector: move tests from yql to ydb (OSS)

Перенос папки с тестами на Коннектор из папки yql в папку ydb (синхронизируется с github).

1 files changed, 53 insertions, 0 deletions

diff --git a/contrib/clickhouse/src/TableFunctions/ITableFunction.cpp b/contrib/clickhouse/src/TableFunctions/ITableFunction.cpp
new file mode 100644
index 00000000000..137e1dc27fe
--- /dev/null
+++ b/contrib/clickhouse/src/TableFunctions/ITableFunction.cpp
@@ -0,0 +1,53 @@
+#include <TableFunctions/ITableFunction.h>
+#include <Interpreters/Context.h>
+#include <Storages/StorageFactory.h>
+#include <Storages/StorageTableFunction.h>
+#include <Access/Common/AccessFlags.h>
+#include <Common/ProfileEvents.h>
+#include <TableFunctions/TableFunctionFactory.h>
+
+
+namespace ProfileEvents
+{
+    extern const Event TableFunctionExecute;
+}
+
+namespace DB
+{
+
+AccessType ITableFunction::getSourceAccessType() const
+{
+    return StorageFactory::instance().getSourceAccessType(getStorageTypeName());
+}
+
+StoragePtr ITableFunction::execute(const ASTPtr & ast_function, ContextPtr context, const std::string & table_name,
+                                   ColumnsDescription cached_columns, bool use_global_context, bool is_insert_query) const
+{
+    ProfileEvents::increment(ProfileEvents::TableFunctionExecute);
+
+    AccessFlags required_access = getSourceAccessType();
+    auto table_function_properties = TableFunctionFactory::instance().tryGetProperties(getName());
+    if (is_insert_query || !(table_function_properties && table_function_properties->allow_readonly))
+        required_access |= AccessType::CREATE_TEMPORARY_TABLE;
+    context->checkAccess(required_access);
+
+    auto context_to_use = use_global_context ? context->getGlobalContext() : context;
+
+    if (cached_columns.empty())
+        return executeImpl(ast_function, context, table_name, std::move(cached_columns), is_insert_query);
+
+    if (hasStaticStructure() && cached_columns == getActualTableStructure(context,is_insert_query))
+        return executeImpl(ast_function, context_to_use, table_name, std::move(cached_columns), is_insert_query);
+
+    auto this_table_function = shared_from_this();
+    auto get_storage = [=]() -> StoragePtr
+    {
+        return this_table_function->executeImpl(ast_function, context_to_use, table_name, cached_columns, is_insert_query);
+    };
+
+    /// It will request actual table structure and create underlying storage lazily
+    return std::make_shared<StorageTableFunctionProxy>(StorageID(getDatabaseName(), table_name), std::move(get_storage),
+                                                       std::move(cached_columns), needStructureConversion());
+}
+
+}