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

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

1 files changed, 54 insertions, 0 deletions

diff --git a/contrib/clickhouse/src/AggregateFunctions/AggregateFunctionForEach.cpp b/contrib/clickhouse/src/AggregateFunctions/AggregateFunctionForEach.cpp
new file mode 100644
index 00000000000..2e973b69fde
--- /dev/null
+++ b/contrib/clickhouse/src/AggregateFunctions/AggregateFunctionForEach.cpp
@@ -0,0 +1,54 @@
+#include <AggregateFunctions/AggregateFunctionForEach.h>
+#include <AggregateFunctions/AggregateFunctionCombinatorFactory.h>
+#include <Common/typeid_cast.h>
+
+
+namespace DB
+{
+
+namespace ErrorCodes
+{
+    extern const int ILLEGAL_TYPE_OF_ARGUMENT;
+}
+
+namespace
+{
+
+class AggregateFunctionCombinatorForEach final : public IAggregateFunctionCombinator
+{
+public:
+    String getName() const override { return "ForEach"; }
+
+    DataTypes transformArguments(const DataTypes & arguments) const override
+    {
+        DataTypes nested_arguments;
+        for (const auto & type : arguments)
+        {
+            if (const DataTypeArray * array = typeid_cast<const DataTypeArray *>(type.get()))
+                nested_arguments.push_back(array->getNestedType());
+            else
+                throw Exception(ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT, "Illegal type {} of argument for "
+                                "aggregate function with {} suffix. Must be array.", type->getName(), getName());
+        }
+
+        return nested_arguments;
+    }
+
+    AggregateFunctionPtr transformAggregateFunction(
+        const AggregateFunctionPtr & nested_function,
+        const AggregateFunctionProperties &,
+        const DataTypes & arguments,
+        const Array & params) const override
+    {
+        return std::make_shared<AggregateFunctionForEach>(nested_function, arguments, params);
+    }
+};
+
+}
+
+void registerAggregateFunctionCombinatorForEach(AggregateFunctionCombinatorFactory & factory)
+{
+    factory.registerCombinator(std::make_shared<AggregateFunctionCombinatorForEach>());
+}
+
+}