[feature-wip](array-type) add function array_union/array_except/array_intersect (#10781)

Add array_union/array_except/array_intersect function.
This commit is contained in:
xy720
2022-07-22 13:50:13 +08:00
committed by GitHub
parent 9d21b2154d
commit 3744321f01
18 changed files with 1160 additions and 2 deletions

View File

@ -0,0 +1,79 @@
// 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.
#include "vec/functions/array/function_array_binary.h"
#include "vec/functions/array/function_array_set.h"
#include "vec/functions/simple_function_factory.h"
namespace doris::vectorized {
struct NameArrayIntersect {
static constexpr auto name = "array_intersect";
};
template <typename Set, typename Element>
struct IntersectAction {
// True if set has null element
bool null_flag = false;
// True if result_set has null element
bool result_null_flag = false;
// True if it should execute the left array first.
static constexpr auto execute_left_column_first = false;
// Handle Null element.
// Return true means this null element should put into result column.
template <bool is_left>
bool apply_null() {
if constexpr (is_left) {
if (!result_null_flag) {
result_null_flag = true;
return null_flag;
}
} else {
if (!null_flag) {
null_flag = true;
}
}
return false;
}
// Handle Non-Null element.
// Return ture means this Non-Null element should put into result column.
template <bool is_left>
bool apply(Set& set, Set& result_set, const Element& elem) {
if constexpr (is_left) {
if (set.find(elem) && !result_set.find(elem)) {
result_set.insert(elem);
return true;
}
} else {
if (!set.find(elem)) {
set.insert(elem);
}
}
return false;
}
};
using FunctionArrayIntersect =
FunctionArrayBinary<ArraySetImpl<SetOperation::INTERSECT>, NameArrayIntersect>;
void register_function_array_intersect(SimpleFunctionFactory& factory) {
factory.register_function<FunctionArrayIntersect>();
}
} // namespace doris::vectorized