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

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

1 files changed, 74 insertions, 0 deletions

diff --git a/contrib/clickhouse/src/TableFunctions/TableFunctionS3Cluster.cpp b/contrib/clickhouse/src/TableFunctions/TableFunctionS3Cluster.cpp
new file mode 100644
index 00000000000..4eb18851bb5
--- /dev/null
+++ b/contrib/clickhouse/src/TableFunctions/TableFunctionS3Cluster.cpp
@@ -0,0 +1,74 @@
+#include "clickhouse_config.h"
+
+#if USE_AWS_S3
+
+#include <TableFunctions/TableFunctionS3Cluster.h>
+#include <TableFunctions/TableFunctionFactory.h>
+#include <Interpreters/parseColumnsListForTableFunction.h>
+#include <Storages/StorageS3.h>
+
+#include "registerTableFunctions.h"
+
+#include <memory>
+
+
+namespace DB
+{
+
+StoragePtr TableFunctionS3Cluster::executeImpl(
+    const ASTPtr & /*function*/, ContextPtr context,
+    const std::string & table_name, ColumnsDescription /*cached_columns*/, bool /*is_insert_query*/) const
+{
+    StoragePtr storage;
+    ColumnsDescription columns;
+    bool structure_argument_was_provided = configuration.structure != "auto";
+
+    if (structure_argument_was_provided)
+    {
+        columns = parseColumnsListFromString(configuration.structure, context);
+    }
+    else if (!structure_hint.empty())
+    {
+        columns = structure_hint;
+    }
+
+    if (context->getClientInfo().query_kind == ClientInfo::QueryKind::SECONDARY_QUERY)
+    {
+        /// On worker node this filename won't contains globs
+        storage = std::make_shared<StorageS3>(
+            configuration,
+            context,
+            StorageID(getDatabaseName(), table_name),
+            columns,
+            ConstraintsDescription{},
+            /* comment */String{},
+            /* format_settings */std::nullopt, /// No format_settings for S3Cluster
+            /*distributed_processing=*/true);
+    }
+    else
+    {
+        storage = std::make_shared<StorageS3Cluster>(
+            cluster_name,
+            configuration,
+            StorageID(getDatabaseName(), table_name),
+            columns,
+            ConstraintsDescription{},
+            context,
+            structure_argument_was_provided);
+    }
+
+    storage->startup();
+
+    return storage;
+}
+
+
+void registerTableFunctionS3Cluster(TableFunctionFactory & factory)
+{
+    factory.registerFunction<TableFunctionS3Cluster>();
+}
+
+
+}
+
+#endif