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

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

1 files changed, 71 insertions, 0 deletions

diff --git a/contrib/clickhouse/src/TableFunctions/TableFunctionPostgreSQL.cpp b/contrib/clickhouse/src/TableFunctions/TableFunctionPostgreSQL.cpp
new file mode 100644
index 00000000000..6ed5883c3cc
--- /dev/null
+++ b/contrib/clickhouse/src/TableFunctions/TableFunctionPostgreSQL.cpp
@@ -0,0 +1,71 @@
+#include <TableFunctions/TableFunctionPostgreSQL.h>
+
+#if USE_LIBPQXX
+#include <Interpreters/evaluateConstantExpression.h>
+#include <Parsers/ASTFunction.h>
+#include <TableFunctions/ITableFunction.h>
+#include <TableFunctions/TableFunctionFactory.h>
+#include <Common/Exception.h>
+#include "registerTableFunctions.h"
+#include <Common/parseRemoteDescription.h>
+
+
+namespace DB
+{
+
+namespace ErrorCodes
+{
+    extern const int BAD_ARGUMENTS;
+}
+
+
+StoragePtr TableFunctionPostgreSQL::executeImpl(const ASTPtr & /*ast_function*/,
+        ContextPtr context, const std::string & table_name, ColumnsDescription cached_columns, bool /*is_insert_query*/) const
+{
+    auto result = std::make_shared<StoragePostgreSQL>(
+        StorageID(getDatabaseName(), table_name),
+        connection_pool,
+        configuration->table,
+        cached_columns,
+        ConstraintsDescription{},
+        String{},
+        context,
+        configuration->schema,
+        configuration->on_conflict);
+
+    result->startup();
+    return result;
+}
+
+
+ColumnsDescription TableFunctionPostgreSQL::getActualTableStructure(ContextPtr context, bool /*is_insert_query*/) const
+{
+    return StoragePostgreSQL::getTableStructureFromData(connection_pool, configuration->table, configuration->schema, context);
+}
+
+
+void TableFunctionPostgreSQL::parseArguments(const ASTPtr & ast_function, ContextPtr context)
+{
+    const auto & func_args = ast_function->as<ASTFunction &>();
+    if (!func_args.arguments)
+        throw Exception(ErrorCodes::BAD_ARGUMENTS, "Table function 'PostgreSQL' must have arguments.");
+
+    configuration.emplace(StoragePostgreSQL::getConfiguration(func_args.arguments->children, context));
+    const auto & settings = context->getSettingsRef();
+    connection_pool = std::make_shared<postgres::PoolWithFailover>(
+        *configuration,
+        settings.postgresql_connection_pool_size,
+        settings.postgresql_connection_pool_wait_timeout,
+        POSTGRESQL_POOL_WITH_FAILOVER_DEFAULT_MAX_TRIES,
+        settings.postgresql_connection_pool_auto_close_connection);
+}
+
+
+void registerTableFunctionPostgreSQL(TableFunctionFactory & factory)
+{
+    factory.registerFunction<TableFunctionPostgreSQL>();
+}
+
+}
+
+#endif