// 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.

#pragma once

#include "core/column/column_array.h"

#include "core/data_type/data_type_array.h"

#include "core/data_type/data_type_number.h"

#include "exprs/function/array/function_array_utils.h"

#include "exprs/function/function.h"

#include "exprs/function/function_helpers.h"

namespace doris {

// Functions with arguments is two arrays of the same element type.

template <typename Impl, typename Name>

class FunctionArrayBinary : public IFunction {

public:

static constexpr auto name = Name::name;

static FunctionPtr create() { return std::make_shared<FunctionArrayBinary>(); }

String get_name() const override { return name; }

bool is_variadic() const override { return false; }

size_t get_number_of_arguments() const override { return 2; }

DataTypePtr get_return_type_impl(const DataTypes& arguments) const override {

DCHECK(arguments[0]->get_primitive_type() == TYPE_ARRAY) << arguments[0]->get_name();

DCHECK(arguments[1]->get_primitive_type() == TYPE_ARRAY) << arguments[1]->get_name();

auto left_nested_type = remove_nullable(

assert_cast<const DataTypeArray&>(*(arguments[0])).get_nested_type());

auto right_nested_type = remove_nullable(

assert_cast<const DataTypeArray&>(*(arguments[1])).get_nested_type());

DCHECK(left_nested_type->equals_ignore_precision(*right_nested_type))

<< "data type " << arguments[0]->get_name() << " not equal with "

<< arguments[1]->get_name();

return Impl::get_return_type(arguments);

}

Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,

uint32_t result, size_t input_rows_count) const override {

const auto& [left_column, left_const] =

unpack_if_const(block.get_by_position(arguments[0]).column);

const auto& [right_column, right_const] =

unpack_if_const(block.get_by_position(arguments[1]).column);

// extract array column

ColumnArrayExecutionData left_data;

ColumnArrayExecutionData right_data;

ColumnPtr res_ptr = nullptr;

if (extract_column_array_info(*left_column, left_data) &&

extract_column_array_info(*right_column, right_data) &&

Impl::execute(res_ptr, left_data, right_data, left_const, right_const) ==

Status::OK()) {

block.replace_by_position(result, std::move(res_ptr));

return Status::OK();

}

return Status::RuntimeError(

fmt::format("execute failed, unsupported types for function {}({}, {})", get_name(),

block.get_by_position(arguments[0]).type->get_name(),

block.get_by_position(arguments[1]).type->get_name()));

}

};

} // namespace doris