[feature](agg) support aggregate function group_array_intersect

be/src/vec/aggregate_functions/aggregate_function_group_array_intersect.cpp

@@ -0,0 +1,90 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+// This file is copied from
+// https://github.com/ClickHouse/ClickHouse/blob/master/src/AggregateFunctions/AggregateFunctionGroupArrayIntersect.cpp
+// and modified by Doris
+
+#include "vec/aggregate_functions/aggregate_function_group_array_intersect.h"
+
+namespace doris::vectorized {
+
+IAggregateFunction* create_with_extra_types(const DataTypePtr& nested_type,
+                                            const DataTypes& argument_types) {
+    WhichDataType which(nested_type);
+    if (which.idx == TypeIndex::Date || which.idx == TypeIndex::DateTime) {
+        throw Exception(ErrorCode::INVALID_ARGUMENT,
+                        "We don't support array<date> or array<datetime> for "
+                        "group_array_intersect(), please use array<datev2> or array<datetimev2>.");
+    } else if (which.idx == TypeIndex::DateV2) {
+        return new AggregateFunctionGroupArrayIntersect<DateV2>(argument_types);
+    } else if (which.idx == TypeIndex::DateTimeV2) {
+        return new AggregateFunctionGroupArrayIntersect<DateTimeV2>(argument_types);
+    } else {
+        /// Check that we can use plain version of AggregateFunctionGroupArrayIntersectGeneric
+        if (nested_type->is_value_unambiguously_represented_in_contiguous_memory_region())
+            return new AggregateFunctionGroupArrayIntersectGeneric<true>(argument_types);
+        else
+            return new AggregateFunctionGroupArrayIntersectGeneric<false>(argument_types);
+    }
+}
+
+inline AggregateFunctionPtr create_aggregate_function_group_array_intersect_impl(
+        const std::string& name, const DataTypes& argument_types, const bool result_is_nullable) {
+    const auto& nested_type = remove_nullable(
+            dynamic_cast<const DataTypeArray&>(*(argument_types[0])).get_nested_type());
+    AggregateFunctionPtr res = nullptr;
+
+    WhichDataType which(nested_type);
+#define DISPATCH(TYPE)                                                                  \
+    if (which.idx == TypeIndex::TYPE)                                                   \
+        res = creator_without_type::create<AggregateFunctionGroupArrayIntersect<TYPE>>( \
+                argument_types, result_is_nullable);
+    FOR_NUMERIC_TYPES(DISPATCH)
+#undef DISPATCH
+
+    if (!res) {
+        res = AggregateFunctionPtr(create_with_extra_types(nested_type, argument_types));
+    }
+
+    if (!res) {
+        throw Exception(ErrorCode::INVALID_ARGUMENT,
+                        "Illegal type {} of argument for aggregate function {}",
+                        argument_types[0]->get_name(), name);
+    }
+
+    return res;
+}
+
+AggregateFunctionPtr create_aggregate_function_group_array_intersect(
+        const std::string& name, const DataTypes& argument_types, const bool result_is_nullable) {
+    assert_unary(name, argument_types);
+    const DataTypePtr& argument_type = remove_nullable(argument_types[0]);
+
+    if (!WhichDataType(argument_type).is_array())
+        throw Exception(ErrorCode::INVALID_ARGUMENT,
+                        "Aggregate function groupArrayIntersect accepts only array type argument. "
+                        "Provided argument type: " +
+                                argument_type->get_name());
+    return create_aggregate_function_group_array_intersect_impl(name, {argument_type},
+                                                                result_is_nullable);
+}
+
+void register_aggregate_function_group_array_intersect(AggregateFunctionSimpleFactory& factory) {
+    factory.register_function_both("group_array_intersect",
+                                   create_aggregate_function_group_array_intersect);
+}
+} // namespace doris::vectorized