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/Functions/array/arrayExists.cpp b/contrib/clickhouse/src/Functions/array/arrayExists.cpp
new file mode 100644
index 00000000000..d59425338a5
--- /dev/null
+++ b/contrib/clickhouse/src/Functions/array/arrayExists.cpp
@@ -0,0 +1,71 @@
+#include <Functions/array/arrayExists.h>
+#include <Functions/FunctionFactory.h>
+
+namespace DB
+{
+
+namespace ErrorCodes
+{
+    extern const int ILLEGAL_COLUMN;
+}
+
+ColumnPtr ArrayExistsImpl::execute(const ColumnArray & array, ColumnPtr mapped)
+{
+    const ColumnUInt8 * column_filter = typeid_cast<const ColumnUInt8 *>(&*mapped);
+
+    if (!column_filter)
+    {
+        const auto * column_filter_const = checkAndGetColumnConst<ColumnUInt8>(&*mapped);
+
+        if (!column_filter_const)
+            throw Exception(ErrorCodes::ILLEGAL_COLUMN, "Unexpected type of filter column");
+
+        if (column_filter_const->getValue<UInt8>())
+        {
+            const IColumn::Offsets & offsets = array.getOffsets();
+            auto out_column = ColumnUInt8::create(offsets.size());
+            ColumnUInt8::Container & out_exists = out_column->getData();
+
+            size_t pos = 0;
+            for (size_t i = 0; i < offsets.size(); ++i)
+            {
+                out_exists[i] = offsets[i] - pos > 0;
+                pos = offsets[i];
+            }
+
+            return out_column;
+        }
+        else
+            return DataTypeUInt8().createColumnConst(array.size(), 0u);
+    }
+
+    const IColumn::Filter & filter = column_filter->getData();
+    const IColumn::Offsets & offsets = array.getOffsets();
+    auto out_column = ColumnUInt8::create(offsets.size());
+    ColumnUInt8::Container & out_exists = out_column->getData();
+
+    size_t pos = 0;
+    for (size_t i = 0; i < offsets.size(); ++i)
+    {
+        UInt8 exists = 0;
+        for (; pos < offsets[i]; ++pos)
+        {
+            if (filter[pos])
+            {
+                exists = 1;
+                pos = offsets[i];
+                break;
+            }
+        }
+        out_exists[i] = exists;
+    }
+
+    return out_column;
+}
+
+REGISTER_FUNCTION(ArrayExists)
+{
+    factory.registerFunction<FunctionArrayExists>();
+}
+
+}