[feature-wip](array-type) support the array type in reverse function (#11213)

Co-authored-by: hucheng01 <hucheng01@baidu.com>
2022-08-09 20:49:09 +08:00
parent 169996d8e4
commit df47b6941d
9 changed files with 285 additions and 39 deletions
--- a/be/src/vec/functions/array/function_array_reverse.h
+++ b/be/src/vec/functions/array/function_array_reverse.h
@ -0,0 +1,92 @@
+// 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/Functions/array/arrayReverse.cpp
+// and modified by Doris
+#pragma once
+
+#include "vec/columns/column_array.h"
+#include "vec/columns/column_const.h"
+#include "vec/data_types/data_type_array.h"
+#include "vec/data_types/data_type_number.h"
+#include "vec/functions/array/function_array_utils.h"
+
+namespace doris::vectorized {
+
+struct ArrayReverseImpl {
+    static Status _execute(Block& block, const ColumnNumbers& arguments, size_t result,
+                           size_t input_rows_count) {
+        ColumnPtr src_column =
+                block.get_by_position(arguments[0]).column->convert_to_full_column_if_const();
+        ColumnArrayExecutionData src;
+        if (!extract_column_array_info(*src_column, src)) {
+            return Status::RuntimeError(
+                    fmt::format("execute failed, unsupported types for function {}({})", "reverse",
+                                block.get_by_position(arguments[0]).type->get_name()));
+        }
+
+        bool is_nullable = src.nested_nullmap_data ? true : false;
+        ColumnArrayMutableData dst = create_mutable_data(src.nested_col, is_nullable);
+        dst.offsets_ptr->reserve(input_rows_count);
+
+        auto res_val = _execute_internal(*src.nested_col, *src.offsets_ptr, *dst.nested_col,
+                                         *dst.offsets_ptr, src.nested_nullmap_data,
+                                         dst.nested_nullmap_data);
+        if (!res_val) {
+            return Status::RuntimeError(
+                    fmt::format("execute failed or unsupported types for function {}({})",
+                                "reverse", block.get_by_position(arguments[0]).type->get_name()));
+        }
+
+        ColumnPtr res_column = assemble_column_array(dst);
+        block.replace_by_position(result, std::move(res_column));
+        return Status::OK();
+    }
+
+    static bool _execute_internal(const IColumn& src_column,
+                                  const ColumnArray::Offsets& src_offsets, IColumn& dest_column,
+                                  ColumnArray::Offsets& dest_offsets, const UInt8* src_null_map,
+                                  ColumnUInt8::Container* dest_null_map) {
+        ColumnArray::Offset prev_src_offset = 0;
+
+        for (auto curr_src_offset : src_offsets) {
+            size_t array_size = curr_src_offset - prev_src_offset;
+            for (size_t j = 0; j < array_size; ++j) {
+                size_t j_reverse = curr_src_offset - j - 1;
+                if (src_null_map && src_null_map[j_reverse]) {
+                    DCHECK(dest_null_map != nullptr);
+                    // Note: here we need to insert default value
+                    dest_column.insert_default();
+                    (*dest_null_map).push_back(true);
+                    continue;
+                }
+
+                dest_column.insert_from(src_column, j_reverse);
+                if (dest_null_map) {
+                    (*dest_null_map).push_back(false);
+                }
+            }
+
+            dest_offsets.push_back(curr_src_offset);
+            prev_src_offset = curr_src_offset;
+        }
+
+        return true;
+    }
+};
+
+} // namespace doris::vectorized